blob: 53e80b00847e836abca1d70f4c46877fbdd62fd2 [file] [log] [blame]
Guido van Rossum394a47b2001-08-02 14:14:20 +00001import os, sys, string, random, tempfile, unittest
2
3from test_support import run_unittest
4
5class TestImport(unittest.TestCase):
6
7 def __init__(self, *args, **kw):
8 self.package_name = 'PACKAGE_'
9 while sys.modules.has_key(self.package_name):
10 self.package_name += random.choose(string.letters)
11 self.module_name = self.package_name + '.foo'
12 unittest.TestCase.__init__(self, *args, **kw)
13
14 def remove_modules(self):
15 for module_name in (self.package_name, self.module_name):
16 if sys.modules.has_key(module_name):
17 del sys.modules[module_name]
18
19 def setUp(self):
20 self.test_dir = tempfile.mktemp()
21 os.mkdir(self.test_dir)
22 sys.path.append(self.test_dir)
23 self.package_dir = os.path.join(self.test_dir,
24 self.package_name)
25 os.mkdir(self.package_dir)
26 open(os.path.join(self.package_dir, '__init__.py'), 'w')
27 self.module_path = os.path.join(self.package_dir, 'foo.py')
28
29 def tearDown(self):
30 for file in os.listdir(self.package_dir):
31 os.remove(os.path.join(self.package_dir, file))
32 os.rmdir(self.package_dir)
33 os.rmdir(self.test_dir)
34 self.assertNotEqual(sys.path.count(self.test_dir), 0)
35 sys.path.remove(self.test_dir)
36 self.remove_modules()
37
38 def rewrite_file(self, contents):
39 compiled_path = self.module_path + 'c'
40 if os.path.exists(compiled_path): os.remove(compiled_path)
41 open(self.module_path, 'w').write(contents)
42
43 def test_package_import__semantics(self):
44
45 # Generate a couple of broken modules to try importing.
46
47 # ...try loading the module when there's a SyntaxError
48 self.rewrite_file('for')
49 try: __import__(self.module_name)
50 except SyntaxError: pass
51 else: raise RuntimeError, 'Failed to induce SyntaxError'
52 self.assert_(not sys.modules.has_key(self.module_name) and
53 not hasattr(sys.modules[self.package_name], 'foo'))
54
55 # ...make up a variable name that isn't bound in __builtins__
56 var = 'a'
57 while var in dir(__builtins__):
58 var += random.choose(string.letters)
59
60 # ...make a module that just contains that
61 self.rewrite_file(var)
62
63 try: __import__(self.module_name)
64 except NameError: pass
65 else: raise RuntimeError, 'Failed to induce NameError.'
66 module = __import__(self.module_name).foo
67
68 # ...now change the module so that the NameError doesn't
69 # happen
70 self.rewrite_file('%s = 1' % var)
71 reload(module)
72 self.assertEqual(getattr(module, var), 1)
73
74run_unittest(TestImport)