test_pkgimport.py 2.71 KB
Newer Older
Barry Warsaw's avatar
Barry Warsaw committed
1 2 3 4 5 6 7 8 9
import os
import sys
import shutil
import string
import random
import tempfile
import unittest

from imp import cache_from_source
10
from test.support import run_unittest, create_empty_file
11 12 13 14 15

class TestImport(unittest.TestCase):

    def __init__(self, *args, **kw):
        self.package_name = 'PACKAGE_'
16
        while self.package_name in sys.modules:
17
            self.package_name += random.choose(string.ascii_letters)
18 19 20 21 22
        self.module_name = self.package_name + '.foo'
        unittest.TestCase.__init__(self, *args, **kw)

    def remove_modules(self):
        for module_name in (self.package_name, self.module_name):
23
            if module_name in sys.modules:
24 25 26
                del sys.modules[module_name]

    def setUp(self):
27
        self.test_dir = tempfile.mkdtemp()
28 29 30 31
        sys.path.append(self.test_dir)
        self.package_dir = os.path.join(self.test_dir,
                                        self.package_name)
        os.mkdir(self.package_dir)
32
        create_empty_file(os.path.join(self.package_dir, '__init__.py'))
Skip Montanaro's avatar
Skip Montanaro committed
33
        self.module_path = os.path.join(self.package_dir, 'foo.py')
34 35

    def tearDown(self):
Barry Warsaw's avatar
Barry Warsaw committed
36
        shutil.rmtree(self.test_dir)
37 38 39 40 41
        self.assertNotEqual(sys.path.count(self.test_dir), 0)
        sys.path.remove(self.test_dir)
        self.remove_modules()

    def rewrite_file(self, contents):
Barry Warsaw's avatar
Barry Warsaw committed
42 43 44 45 46
        compiled_path = cache_from_source(self.module_path)
        if os.path.exists(compiled_path):
            os.remove(compiled_path)
        with open(self.module_path, 'w') as f:
            f.write(contents)
Tim Peters's avatar
Tim Peters committed
47

48 49 50 51 52 53 54 55
    def test_package_import__semantics(self):

        # Generate a couple of broken modules to try importing.

        # ...try loading the module when there's a SyntaxError
        self.rewrite_file('for')
        try: __import__(self.module_name)
        except SyntaxError: pass
56 57 58
        else: raise RuntimeError('Failed to induce SyntaxError') # self.fail()?
        self.assertNotIn(self.module_name, sys.modules)
        self.assertFalse(hasattr(sys.modules[self.package_name], 'foo'))
59 60 61 62

        # ...make up a variable name that isn't bound in __builtins__
        var = 'a'
        while var in dir(__builtins__):
63
            var += random.choose(string.ascii_letters)
64 65 66 67 68 69

        # ...make a module that just contains that
        self.rewrite_file(var)

        try: __import__(self.module_name)
        except NameError: pass
70
        else: raise RuntimeError('Failed to induce NameError.')
71 72 73 74

        # ...now  change  the module  so  that  the NameError  doesn't
        # happen
        self.rewrite_file('%s = 1' % var)
75
        module = __import__(self.module_name).foo
76 77
        self.assertEqual(getattr(module, var), 1)

78 79 80 81 82 83 84

def test_main():
    run_unittest(TestImport)


if __name__ == "__main__":
    test_main()