diff options
| author | django-bot <ops@djangoproject.com> | 2022-02-08 12:09:55 +0100 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-08 12:15:38 +0100 |
| commit | 6a682b38e75d4c975b4c4493565a59f1bc14397c (patch) | |
| tree | 0bd9cda550bea26238656d9f120d769e8b41bb9e /tests/utils_tests/test_module_loading.py | |
| parent | e73ce08888e6f34d3f050377cfd2fbb733be94a9 (diff) | |
[4.0.x] Refs #33476 -- Reformatted code with Black.
Backport of 9c19aff7c7561e3a82978a272ecdaad40dda5c00 from main.
Diffstat (limited to 'tests/utils_tests/test_module_loading.py')
| -rw-r--r-- | tests/utils_tests/test_module_loading.py | 168 |
1 files changed, 96 insertions, 72 deletions
diff --git a/tests/utils_tests/test_module_loading.py b/tests/utils_tests/test_module_loading.py index 773c0a2735..736d456d2d 100644 --- a/tests/utils_tests/test_module_loading.py +++ b/tests/utils_tests/test_module_loading.py @@ -7,192 +7,215 @@ from zipimport import zipimporter from django.test import SimpleTestCase, modify_settings from django.test.utils import extend_sys_path from django.utils.module_loading import ( - autodiscover_modules, import_string, module_has_submodule, + autodiscover_modules, + import_string, + module_has_submodule, ) from django.utils.version import PY310 class DefaultLoader(unittest.TestCase): - 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') + 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') + 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.assertTrue(module_has_submodule(test_module, "bad_module")) with self.assertRaises(ImportError): - import_module('utils_tests.test_module.bad_module') + 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.assertFalse(module_has_submodule(test_module, "no_such_module")) with self.assertRaises(ImportError): - import_module('utils_tests.test_module.no_such_module') + 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.assertFalse(module_has_submodule(test_module, "django")) with self.assertRaises(ImportError): - import_module('utils_tests.test_module.django') + import_module("utils_tests.test_module.django") # Don't be confused by caching of import misses import types # NOQA: causes attempted import of utils_tests.types - self.assertFalse(module_has_submodule(sys.modules['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.assertFalse(module_has_submodule(test_no_submodule, "anything")) with self.assertRaises(ImportError): - import_module('utils_tests.test_no_submodule.anything') + import_module("utils_tests.test_no_submodule.anything") def test_has_sumbodule_with_dotted_path(self): """Nested module existence can be tested.""" - test_module = import_module('utils_tests.test_module') + test_module = import_module("utils_tests.test_module") # A grandchild that exists. - self.assertIs(module_has_submodule(test_module, 'child_module.grandchild_module'), True) + self.assertIs( + module_has_submodule(test_module, "child_module.grandchild_module"), True + ) # A grandchild that doesn't exist. - self.assertIs(module_has_submodule(test_module, 'child_module.no_such_module'), False) + self.assertIs( + module_has_submodule(test_module, "child_module.no_such_module"), False + ) # A grandchild whose parent doesn't exist. - self.assertIs(module_has_submodule(test_module, 'no_such_module.grandchild_module'), False) + self.assertIs( + module_has_submodule(test_module, "no_such_module.grandchild_module"), False + ) # A grandchild whose parent is not a package. - self.assertIs(module_has_submodule(test_module, 'good_module.no_such_module'), False) + self.assertIs( + module_has_submodule(test_module, "good_module.no_such_module"), False + ) class EggLoader(unittest.TestCase): def setUp(self): - self.egg_dir = '%s/eggs' % os.path.dirname(__file__) + self.egg_dir = "%s/eggs" % os.path.dirname(__file__) def tearDown(self): 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) + 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 + egg_name = "%s/test_egg.egg" % self.egg_dir with extend_sys_path(egg_name): - egg_module = import_module('egg_module') + 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') + 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.assertTrue(module_has_submodule(egg_module, "bad_module")) with self.assertRaises(ImportError): - import_module('egg_module.bad_module') + import_module("egg_module.bad_module") # A child that doesn't exist - self.assertFalse(module_has_submodule(egg_module, 'no_such_module')) + self.assertFalse(module_has_submodule(egg_module, "no_such_module")) with self.assertRaises(ImportError): - import_module('egg_module.no_such_module') + 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 + egg_name = "%s/test_egg.egg" % self.egg_dir with extend_sys_path(egg_name): - egg_module = import_module('egg_module.sub1.sub2') + 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') + 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.assertTrue(module_has_submodule(egg_module, "bad_module")) with self.assertRaises(ImportError): - import_module('egg_module.sub1.sub2.bad_module') + 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.assertFalse(module_has_submodule(egg_module, "no_such_module")) with self.assertRaises(ImportError): - import_module('egg_module.sub1.sub2.no_such_module') + import_module("egg_module.sub1.sub2.no_such_module") class ModuleImportTests(SimpleTestCase): def test_import_string(self): - cls = import_string('django.utils.module_loading.import_string') + cls = import_string("django.utils.module_loading.import_string") self.assertEqual(cls, import_string) # Test exceptions raised with self.assertRaises(ImportError): - import_string('no_dots_in_path') + import_string("no_dots_in_path") msg = 'Module "utils_tests" does not define a "unexistent" attribute' with self.assertRaisesMessage(ImportError, msg): - import_string('utils_tests.unexistent') + import_string("utils_tests.unexistent") -@modify_settings(INSTALLED_APPS={'append': 'utils_tests.test_module'}) +@modify_settings(INSTALLED_APPS={"append": "utils_tests.test_module"}) class AutodiscoverModulesTestCase(SimpleTestCase): - def tearDown(self): sys.path_importer_cache.clear() - sys.modules.pop('utils_tests.test_module.another_bad_module', None) - sys.modules.pop('utils_tests.test_module.another_good_module', None) - sys.modules.pop('utils_tests.test_module.bad_module', None) - sys.modules.pop('utils_tests.test_module.good_module', None) - sys.modules.pop('utils_tests.test_module', None) + sys.modules.pop("utils_tests.test_module.another_bad_module", None) + sys.modules.pop("utils_tests.test_module.another_good_module", None) + sys.modules.pop("utils_tests.test_module.bad_module", None) + sys.modules.pop("utils_tests.test_module.good_module", None) + sys.modules.pop("utils_tests.test_module", None) def test_autodiscover_modules_found(self): - autodiscover_modules('good_module') + autodiscover_modules("good_module") def test_autodiscover_modules_not_found(self): - autodiscover_modules('missing_module') + autodiscover_modules("missing_module") def test_autodiscover_modules_found_but_bad_module(self): - with self.assertRaisesMessage(ImportError, "No module named 'a_package_name_that_does_not_exist'"): - autodiscover_modules('bad_module') + with self.assertRaisesMessage( + ImportError, "No module named 'a_package_name_that_does_not_exist'" + ): + autodiscover_modules("bad_module") def test_autodiscover_modules_several_one_bad_module(self): - with self.assertRaisesMessage(ImportError, "No module named 'a_package_name_that_does_not_exist'"): - autodiscover_modules('good_module', 'bad_module') + with self.assertRaisesMessage( + ImportError, "No module named 'a_package_name_that_does_not_exist'" + ): + autodiscover_modules("good_module", "bad_module") def test_autodiscover_modules_several_found(self): - autodiscover_modules('good_module', 'another_good_module') + autodiscover_modules("good_module", "another_good_module") def test_autodiscover_modules_several_found_with_registry(self): from .test_module import site - autodiscover_modules('good_module', 'another_good_module', register_to=site) - self.assertEqual(site._registry, {'lorem': 'ipsum'}) + + autodiscover_modules("good_module", "another_good_module", register_to=site) + self.assertEqual(site._registry, {"lorem": "ipsum"}) def test_validate_registry_keeps_intact(self): from .test_module import site + with self.assertRaisesMessage(Exception, "Some random exception."): - autodiscover_modules('another_bad_module', register_to=site) + autodiscover_modules("another_bad_module", register_to=site) self.assertEqual(site._registry, {}) def test_validate_registry_resets_after_erroneous_module(self): from .test_module import site + with self.assertRaisesMessage(Exception, "Some random exception."): - autodiscover_modules('another_good_module', 'another_bad_module', register_to=site) - self.assertEqual(site._registry, {'lorem': 'ipsum'}) + autodiscover_modules( + "another_good_module", "another_bad_module", register_to=site + ) + self.assertEqual(site._registry, {"lorem": "ipsum"}) def test_validate_registry_resets_after_missing_module(self): from .test_module import site - autodiscover_modules('does_not_exist', 'another_good_module', 'does_not_exist2', register_to=site) - self.assertEqual(site._registry, {'lorem': 'ipsum'}) + + autodiscover_modules( + "does_not_exist", "another_good_module", "does_not_exist2", register_to=site + ) + self.assertEqual(site._registry, {"lorem": "ipsum"}) if PY310: + class TestFinder: def __init__(self, *args, **kwargs): self.importer = zipimporter(*args, **kwargs) def find_spec(self, path, target=None): return self.importer.find_spec(path, target) + else: + class TestFinder: def __init__(self, *args, **kwargs): self.importer = zipimporter(*args, **kwargs) @@ -218,6 +241,7 @@ class CustomLoader(EggLoader): it uses a custom defined Loader class. Although the EggLoader combines both functions into one class, this isn't required. """ + def setUp(self): super().setUp() sys.path_hooks.insert(0, TestFinder) |
