| 1 | import os, sys, string, random, tempfile, unittest
|
|---|
| 2 |
|
|---|
| 3 | from test.test_support import run_unittest
|
|---|
| 4 |
|
|---|
| 5 | class TestImport(unittest.TestCase):
|
|---|
| 6 |
|
|---|
| 7 | def __init__(self, *args, **kw):
|
|---|
| 8 | self.package_name = 'PACKAGE_'
|
|---|
| 9 | while self.package_name in sys.modules:
|
|---|
| 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 module_name in sys.modules:
|
|---|
| 17 | del sys.modules[module_name]
|
|---|
| 18 |
|
|---|
| 19 | def setUp(self):
|
|---|
| 20 | self.test_dir = tempfile.mkdtemp()
|
|---|
| 21 | sys.path.append(self.test_dir)
|
|---|
| 22 | self.package_dir = os.path.join(self.test_dir,
|
|---|
| 23 | self.package_name)
|
|---|
| 24 | os.mkdir(self.package_dir)
|
|---|
| 25 | open(os.path.join(
|
|---|
| 26 | self.package_dir, '__init__'+os.extsep+'py'), 'w').close()
|
|---|
| 27 | self.module_path = os.path.join(self.package_dir, 'foo'+os.extsep+'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 | for extension in "co":
|
|---|
| 40 | compiled_path = self.module_path + extension
|
|---|
| 41 | if os.path.exists(compiled_path):
|
|---|
| 42 | os.remove(compiled_path)
|
|---|
| 43 | f = open(self.module_path, 'w')
|
|---|
| 44 | f.write(contents)
|
|---|
| 45 | f.close()
|
|---|
| 46 |
|
|---|
| 47 | def test_package_import__semantics(self):
|
|---|
| 48 |
|
|---|
| 49 | # Generate a couple of broken modules to try importing.
|
|---|
| 50 |
|
|---|
| 51 | # ...try loading the module when there's a SyntaxError
|
|---|
| 52 | self.rewrite_file('for')
|
|---|
| 53 | try: __import__(self.module_name)
|
|---|
| 54 | except SyntaxError: pass
|
|---|
| 55 | else: raise RuntimeError, 'Failed to induce SyntaxError'
|
|---|
| 56 | self.assertNotIn(self.module_name, sys.modules)
|
|---|
| 57 | self.assertFalse(hasattr(sys.modules[self.package_name], 'foo'))
|
|---|
| 58 |
|
|---|
| 59 | # ...make up a variable name that isn't bound in __builtins__
|
|---|
| 60 | var = 'a'
|
|---|
| 61 | while var in dir(__builtins__):
|
|---|
| 62 | var += random.choose(string.letters)
|
|---|
| 63 |
|
|---|
| 64 | # ...make a module that just contains that
|
|---|
| 65 | self.rewrite_file(var)
|
|---|
| 66 |
|
|---|
| 67 | try: __import__(self.module_name)
|
|---|
| 68 | except NameError: pass
|
|---|
| 69 | else: raise RuntimeError, 'Failed to induce NameError.'
|
|---|
| 70 |
|
|---|
| 71 | # ...now change the module so that the NameError doesn't
|
|---|
| 72 | # happen
|
|---|
| 73 | self.rewrite_file('%s = 1' % var)
|
|---|
| 74 | module = __import__(self.module_name).foo
|
|---|
| 75 | self.assertEqual(getattr(module, var), 1)
|
|---|
| 76 |
|
|---|
| 77 |
|
|---|
| 78 | def test_main():
|
|---|
| 79 | run_unittest(TestImport)
|
|---|
| 80 |
|
|---|
| 81 |
|
|---|
| 82 | if __name__ == "__main__":
|
|---|
| 83 | test_main()
|
|---|
