@@ -305,9 +305,9 @@ def __init__(self, additional_skip_names=None, patch_path=True,
305
305
self ._skipNames .discard ('path' )
306
306
self ._skipNames .discard ('genericpath' )
307
307
308
- self ._modules_to_reload = [tempfile ]
308
+ self .modules_to_reload = [tempfile ]
309
309
if modules_to_reload is not None :
310
- self ._modules_to_reload .extend (modules_to_reload )
310
+ self .modules_to_reload .extend (modules_to_reload )
311
311
self ._use_dynamic_patch = use_dynamic_patch
312
312
313
313
# Attributes set by _findModules()
@@ -352,7 +352,7 @@ def __init__(self, additional_skip_names=None, patch_path=True,
352
352
self ._stubs = None
353
353
self .fs = None
354
354
self .fake_open = None
355
- self ._fake_modules = {}
355
+ self .fake_modules = {}
356
356
self ._dyn_patcher = None
357
357
358
358
# _isStale is set by tearDown(), reset by _refresh()
@@ -396,8 +396,8 @@ def _refresh(self):
396
396
397
397
self .fs = fake_filesystem .FakeFilesystem ()
398
398
for name in self ._fake_module_classes :
399
- self ._fake_modules [name ] = self ._fake_module_classes [name ](self .fs )
400
- self ._fake_modules ['path' ] = self ._fake_modules ['os' ].path
399
+ self .fake_modules [name ] = self ._fake_module_classes [name ](self .fs )
400
+ self .fake_modules ['path' ] = self .fake_modules ['os' ].path
401
401
self .fake_open = fake_filesystem .FakeFileOpen (self .fs )
402
402
403
403
self ._isStale = False
@@ -419,12 +419,12 @@ def setUp(self, doctester=None):
419
419
self ._stubs .smart_set (builtins , 'open' , self .fake_open )
420
420
for name in self ._modules :
421
421
for module , attr in self ._modules [name ]:
422
- self ._stubs .smart_set (module , attr , self ._fake_modules [name ])
422
+ self ._stubs .smart_set (module , attr , self .fake_modules [name ])
423
423
424
424
self ._dyn_patcher = DynamicPatcher (self )
425
425
sys .meta_path .insert (0 , self ._dyn_patcher )
426
426
427
- for module in self ._modules_to_reload :
427
+ for module in self .modules_to_reload :
428
428
if module .__name__ in sys .modules :
429
429
reload (module )
430
430
@@ -465,7 +465,7 @@ class DynamicPatcher(object):
465
465
def __init__ (self , patcher ):
466
466
self ._patcher = patcher
467
467
self .sysmodules = {}
468
- self .modules = self ._patcher ._fake_modules
468
+ self .modules = self ._patcher .fake_modules
469
469
if 'path' in self .modules :
470
470
self .modules ['os.path' ] = self .modules ['path' ]
471
471
del self .modules ['path' ]
@@ -483,6 +483,9 @@ def __init__(self, patcher):
483
483
def cleanup (self ):
484
484
for module in self .sysmodules :
485
485
sys .modules [module ] = self .sysmodules [module ]
486
+ for module in self ._patcher .modules_to_reload :
487
+ if module .__name__ in sys .modules :
488
+ reload (module )
486
489
487
490
def needs_patch (self , name ):
488
491
"""Check if the module with the given name shall be replaced."""
0 commit comments