home *** CD-ROM | disk | FTP | other *** search
/ Freelog 33 / Freelog033.iso / Progr / Python-2.2.1.exe / TEST_PKGIMPORT.PY < prev    next >
Encoding:
Python Source  |  2001-10-24  |  2.9 KB  |  85 lines
  1. import os, sys, string, random, tempfile, unittest
  2.  
  3. from 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 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__'+os.extsep+'py'), 'w')
  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.assert_(not sys.modules.has_key(self.module_name) and
  57.                      not 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.         module = __import__(self.module_name).foo
  71.  
  72.         # ...now  change  the module  so  that  the NameError  doesn't
  73.         # happen
  74.         self.rewrite_file('%s = 1' % var)
  75.         reload(module)
  76.         self.assertEqual(getattr(module, var), 1)
  77.  
  78.  
  79. def test_main():
  80.     run_unittest(TestImport)
  81.  
  82.  
  83. if __name__ == "__main__":
  84.     test_main()
  85.