diff options
| author | Preston Timmons <prestontimmons@gmail.com> | 2013-04-06 13:59:39 -0500 |
|---|---|---|
| committer | Carl Meyer <carl@oddbird.net> | 2013-04-12 15:31:58 -0600 |
| commit | 612ef3e5c9ba172868d22d93be64a340b7742fc6 (patch) | |
| tree | 5ecffc45e32660be34464b89de093f238e3ec7fc /tests/utils_tests/module_loading.py | |
| parent | 3810dc3070468abd3a3d9beec406a5aa3edd6b52 (diff) | |
Modified utils_tests for unittest2 discovery.
Diffstat (limited to 'tests/utils_tests/module_loading.py')
| -rw-r--r-- | tests/utils_tests/module_loading.py | 194 |
1 files changed, 0 insertions, 194 deletions
diff --git a/tests/utils_tests/module_loading.py b/tests/utils_tests/module_loading.py deleted file mode 100644 index 0a905ef2f1..0000000000 --- a/tests/utils_tests/module_loading.py +++ /dev/null @@ -1,194 +0,0 @@ -import os -import sys -import imp -from zipimport import zipimporter - -from django.core.exceptions import ImproperlyConfigured -from django.utils import unittest -from django.utils.importlib import import_module -from django.utils.module_loading import import_by_path, module_has_submodule -from django.utils._os import upath - - -class DefaultLoader(unittest.TestCase): - def setUp(self): - sys.meta_path.insert(0, ProxyFinder()) - - def tearDown(self): - sys.meta_path.pop(0) - - def test_loader(self): - "Normal module existence can be tested" - test_module = import_module('utils_tests.test_module') - test_no_submodule = import_module( - 'utils_tests.test_no_submodule') - - # An importable child - self.assertTrue(module_has_submodule(test_module, 'good_module')) - mod = import_module('utils_tests.test_module.good_module') - self.assertEqual(mod.content, 'Good Module') - - # A child that exists, but will generate an import error if loaded - self.assertTrue(module_has_submodule(test_module, 'bad_module')) - self.assertRaises(ImportError, import_module, 'utils_tests.test_module.bad_module') - - # A child that doesn't exist - self.assertFalse(module_has_submodule(test_module, 'no_such_module')) - self.assertRaises(ImportError, import_module, 'utils_tests.test_module.no_such_module') - - # A child that doesn't exist, but is the name of a package on the path - self.assertFalse(module_has_submodule(test_module, 'django')) - self.assertRaises(ImportError, import_module, 'utils_tests.test_module.django') - - # Don't be confused by caching of import misses - import types # causes attempted import of utils_tests.types - self.assertFalse(module_has_submodule(sys.modules['utils_tests'], 'types')) - - # A module which doesn't have a __path__ (so no submodules) - self.assertFalse(module_has_submodule(test_no_submodule, 'anything')) - self.assertRaises(ImportError, import_module, - 'utils_tests.test_no_submodule.anything') - -class EggLoader(unittest.TestCase): - def setUp(self): - self.old_path = sys.path[:] - self.egg_dir = '%s/eggs' % os.path.dirname(upath(__file__)) - - def tearDown(self): - sys.path = self.old_path - sys.path_importer_cache.clear() - - sys.modules.pop('egg_module.sub1.sub2.bad_module', None) - sys.modules.pop('egg_module.sub1.sub2.good_module', None) - sys.modules.pop('egg_module.sub1.sub2', None) - sys.modules.pop('egg_module.sub1', None) - sys.modules.pop('egg_module.bad_module', None) - sys.modules.pop('egg_module.good_module', None) - sys.modules.pop('egg_module', None) - - def test_shallow_loader(self): - "Module existence can be tested inside eggs" - egg_name = '%s/test_egg.egg' % self.egg_dir - sys.path.append(egg_name) - egg_module = import_module('egg_module') - - # An importable child - self.assertTrue(module_has_submodule(egg_module, 'good_module')) - mod = import_module('egg_module.good_module') - self.assertEqual(mod.content, 'Good Module') - - # A child that exists, but will generate an import error if loaded - self.assertTrue(module_has_submodule(egg_module, 'bad_module')) - self.assertRaises(ImportError, import_module, 'egg_module.bad_module') - - # A child that doesn't exist - self.assertFalse(module_has_submodule(egg_module, 'no_such_module')) - self.assertRaises(ImportError, import_module, 'egg_module.no_such_module') - - def test_deep_loader(self): - "Modules deep inside an egg can still be tested for existence" - egg_name = '%s/test_egg.egg' % self.egg_dir - sys.path.append(egg_name) - egg_module = import_module('egg_module.sub1.sub2') - - # An importable child - self.assertTrue(module_has_submodule(egg_module, 'good_module')) - mod = import_module('egg_module.sub1.sub2.good_module') - self.assertEqual(mod.content, 'Deep Good Module') - - # A child that exists, but will generate an import error if loaded - self.assertTrue(module_has_submodule(egg_module, 'bad_module')) - self.assertRaises(ImportError, import_module, 'egg_module.sub1.sub2.bad_module') - - # A child that doesn't exist - self.assertFalse(module_has_submodule(egg_module, 'no_such_module')) - self.assertRaises(ImportError, import_module, 'egg_module.sub1.sub2.no_such_module') - - -class ModuleImportTestCase(unittest.TestCase): - def test_import_by_path(self): - cls = import_by_path( - 'django.utils.module_loading.import_by_path') - self.assertEqual(cls, import_by_path) - - # Test exceptions raised - for path in ('no_dots_in_path', 'unexistent.path', - 'utils_tests.unexistent'): - self.assertRaises(ImproperlyConfigured, import_by_path, path) - - with self.assertRaises(ImproperlyConfigured) as cm: - import_by_path('unexistent.module.path', error_prefix="Foo") - self.assertTrue(str(cm.exception).startswith('Foo')) - - def test_import_error_traceback(self): - """Test preserving the original traceback on an ImportError.""" - try: - import_by_path('test_module.bad_module.content') - except ImproperlyConfigured: - traceback = sys.exc_info()[2] - - self.assertIsNotNone(traceback.tb_next.tb_next, - 'Should have more than the calling frame in the traceback.') - - -class ProxyFinder(object): - def __init__(self): - self._cache = {} - - def find_module(self, fullname, path=None): - tail = fullname.rsplit('.', 1)[-1] - try: - fd, fn, info = imp.find_module(tail, path) - if fullname in self._cache: - old_fd = self._cache[fullname][0] - if old_fd: - old_fd.close() - self._cache[fullname] = (fd, fn, info) - except ImportError: - return None - else: - return self # this is a loader as well - - def load_module(self, fullname): - if fullname in sys.modules: - return sys.modules[fullname] - fd, fn, info = self._cache[fullname] - try: - return imp.load_module(fullname, fd, fn, info) - finally: - if fd: - fd.close() - -class TestFinder(object): - def __init__(self, *args, **kwargs): - self.importer = zipimporter(*args, **kwargs) - - def find_module(self, path): - importer = self.importer.find_module(path) - if importer is None: - return - return TestLoader(importer) - -class TestLoader(object): - def __init__(self, importer): - self.importer = importer - - def load_module(self, name): - mod = self.importer.load_module(name) - mod.__loader__ = self - return mod - -class CustomLoader(EggLoader): - """The Custom Loader test is exactly the same as the EggLoader, but - it uses a custom defined Loader and Finder that is intentionally - split into two classes. Although the EggLoader combines both functions - into one class, this isn't required. - """ - def setUp(self): - super(CustomLoader, self).setUp() - sys.path_hooks.insert(0, TestFinder) - sys.path_importer_cache.clear() - - def tearDown(self): - super(CustomLoader, self).tearDown() - sys.path_hooks.pop(0) |
