blob: 3fd593c63455fc8795b2ddd3a7002d043d09676e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
import os
import sys
import shutil
import unittest
import py3kcompat as py3k
orig_path = os.path.join(os.path.dirname(__file__))
workdir = os.getcwd()
src = os.path.join(orig_path, 'test_module_template.py')
dst = os.path.join(workdir, 'test_module.py')
shutil.copyfile(src, dst)
sys.path.append(workdir)
def reload_module(moduleName):
if py3k.IS_PY3K:
import imp
imp.reload(moduleName)
else:
reload(moduleName)
def increment_module_value():
modfile = open(dst, 'a')
modfile.write('Sentinel.value += 1' + os.linesep)
modfile.flush()
modfile.close()
if py3k.IS_PY3K:
import imp
cacheFile = imp.cache_from_source(dst)
else:
cacheFile = dst + 'c'
os.remove(cacheFile)
class TestModuleReloading(unittest.TestCase):
def testModuleReloading(self):
'''Test module reloading with on-the-fly modifications.'''
import test_module
self.assertEqual(test_module.Sentinel.value, 10)
increment_module_value()
reload_module(sys.modules['test_module'])
self.assertEqual(test_module.Sentinel.value, 11)
reload_module(sys.modules['test_module'])
self.assertEqual(test_module.Sentinel.value, 11)
increment_module_value()
reload_module(sys.modules['test_module'])
self.assertEqual(test_module.Sentinel.value, 12)
if __name__ == "__main__":
unittest.main()
|