| from .. import abc |
| from .. import util |
| |
| machinery = util.import_importlib('importlib.machinery') |
| |
| import os.path |
| import sys |
| import types |
| import unittest |
| import importlib.util |
| import importlib |
| |
| |
| class LoaderTests(abc.LoaderTests): |
| |
| """Test load_module() for extension modules.""" |
| |
| def setUp(self): |
| self.loader = self.machinery.ExtensionFileLoader(util.EXTENSIONS.name, |
| util.EXTENSIONS.file_path) |
| |
| def load_module(self, fullname): |
| return self.loader.load_module(fullname) |
| |
| def test_load_module_API(self): |
| # Test the default argument for load_module(). |
| self.loader.load_module() |
| self.loader.load_module(None) |
| with self.assertRaises(ImportError): |
| self.load_module('XXX') |
| |
| def test_equality(self): |
| other = self.machinery.ExtensionFileLoader(util.EXTENSIONS.name, |
| util.EXTENSIONS.file_path) |
| self.assertEqual(self.loader, other) |
| |
| def test_inequality(self): |
| other = self.machinery.ExtensionFileLoader('_' + util.EXTENSIONS.name, |
| util.EXTENSIONS.file_path) |
| self.assertNotEqual(self.loader, other) |
| |
| def test_module(self): |
| with util.uncache(util.EXTENSIONS.name): |
| module = self.load_module(util.EXTENSIONS.name) |
| for attr, value in [('__name__', util.EXTENSIONS.name), |
| ('__file__', util.EXTENSIONS.file_path), |
| ('__package__', '')]: |
| self.assertEqual(getattr(module, attr), value) |
| self.assertIn(util.EXTENSIONS.name, sys.modules) |
| self.assertIsInstance(module.__loader__, |
| self.machinery.ExtensionFileLoader) |
| |
| # No extension module as __init__ available for testing. |
| test_package = None |
| |
| # No extension module in a package available for testing. |
| test_lacking_parent = None |
| |
| def test_module_reuse(self): |
| with util.uncache(util.EXTENSIONS.name): |
| module1 = self.load_module(util.EXTENSIONS.name) |
| module2 = self.load_module(util.EXTENSIONS.name) |
| self.assertIs(module1, module2) |
| |
| # No easy way to trigger a failure after a successful import. |
| test_state_after_failure = None |
| |
| def test_unloadable(self): |
| name = 'asdfjkl;' |
| with self.assertRaises(ImportError) as cm: |
| self.load_module(name) |
| self.assertEqual(cm.exception.name, name) |
| |
| def test_is_package(self): |
| self.assertFalse(self.loader.is_package(util.EXTENSIONS.name)) |
| for suffix in self.machinery.EXTENSION_SUFFIXES: |
| path = os.path.join('some', 'path', 'pkg', '__init__' + suffix) |
| loader = self.machinery.ExtensionFileLoader('pkg', path) |
| self.assertTrue(loader.is_package('pkg')) |
| |
| (Frozen_LoaderTests, |
| Source_LoaderTests |
| ) = util.test_both(LoaderTests, machinery=machinery) |
| |
| class MultiPhaseExtensionModuleTests(abc.LoaderTests): |
| """Test loading extension modules with multi-phase initialization (PEP 489) |
| """ |
| |
| def setUp(self): |
| self.name = '_testmultiphase' |
| finder = self.machinery.FileFinder(None) |
| self.spec = importlib.util.find_spec(self.name) |
| assert self.spec |
| self.loader = self.machinery.ExtensionFileLoader( |
| self.name, self.spec.origin) |
| |
| # No extension module as __init__ available for testing. |
| test_package = None |
| |
| # No extension module in a package available for testing. |
| test_lacking_parent = None |
| |
| # Handling failure on reload is the up to the module. |
| test_state_after_failure = None |
| |
| def test_module(self): |
| '''Test loading an extension module''' |
| with util.uncache(self.name): |
| module = self.load_module() |
| for attr, value in [('__name__', self.name), |
| ('__file__', self.spec.origin), |
| ('__package__', '')]: |
| self.assertEqual(getattr(module, attr), value) |
| with self.assertRaises(AttributeError): |
| module.__path__ |
| self.assertIs(module, sys.modules[self.name]) |
| self.assertIsInstance(module.__loader__, |
| self.machinery.ExtensionFileLoader) |
| |
| def test_functionality(self): |
| '''Test basic functionality of stuff defined in an extension module''' |
| with util.uncache(self.name): |
| module = self.load_module() |
| self.assertIsInstance(module, types.ModuleType) |
| ex = module.Example() |
| self.assertEqual(ex.demo('abcd'), 'abcd') |
| self.assertEqual(ex.demo(), None) |
| with self.assertRaises(AttributeError): |
| ex.abc |
| ex.abc = 0 |
| self.assertEqual(ex.abc, 0) |
| self.assertEqual(module.foo(9, 9), 18) |
| self.assertIsInstance(module.Str(), str) |
| self.assertEqual(module.Str(1) + '23', '123') |
| with self.assertRaises(module.error): |
| raise module.error() |
| self.assertEqual(module.int_const, 1969) |
| self.assertEqual(module.str_const, 'something different') |
| |
| def test_reload(self): |
| '''Test that reload didn't re-set the module's attributes''' |
| with util.uncache(self.name): |
| module = self.load_module() |
| ex_class = module.Example |
| importlib.reload(module) |
| self.assertIs(ex_class, module.Example) |
| |
| def test_try_registration(self): |
| '''Assert that the PyState_{Find,Add,Remove}Module C API doesn't work''' |
| module = self.load_module() |
| with self.subTest('PyState_FindModule'): |
| self.assertEqual(module.call_state_registration_func(0), None) |
| with self.subTest('PyState_AddModule'): |
| with self.assertRaises(SystemError): |
| module.call_state_registration_func(1) |
| with self.subTest('PyState_RemoveModule'): |
| with self.assertRaises(SystemError): |
| module.call_state_registration_func(2) |
| |
| def load_module(self): |
| '''Load the module from the test extension''' |
| return self.loader.load_module(self.name) |
| |
| def load_module_by_name(self, fullname): |
| '''Load a module from the test extension by name''' |
| origin = self.spec.origin |
| loader = self.machinery.ExtensionFileLoader(fullname, origin) |
| spec = importlib.util.spec_from_loader(fullname, loader) |
| module = importlib.util.module_from_spec(spec) |
| loader.exec_module(module) |
| return module |
| |
| def test_load_submodule(self): |
| '''Test loading a simulated submodule''' |
| module = self.load_module_by_name('pkg.' + self.name) |
| self.assertIsInstance(module, types.ModuleType) |
| self.assertEqual(module.__name__, 'pkg.' + self.name) |
| self.assertEqual(module.str_const, 'something different') |
| |
| def test_load_short_name(self): |
| '''Test loading module with a one-character name''' |
| module = self.load_module_by_name('x') |
| self.assertIsInstance(module, types.ModuleType) |
| self.assertEqual(module.__name__, 'x') |
| self.assertEqual(module.str_const, 'something different') |
| self.assertNotIn('x', sys.modules) |
| |
| def test_load_twice(self): |
| '''Test that 2 loads result in 2 module objects''' |
| module1 = self.load_module_by_name(self.name) |
| module2 = self.load_module_by_name(self.name) |
| self.assertIsNot(module1, module2) |
| |
| def test_unloadable(self): |
| '''Test nonexistent module''' |
| name = 'asdfjkl;' |
| with self.assertRaises(ImportError) as cm: |
| self.load_module_by_name(name) |
| self.assertEqual(cm.exception.name, name) |
| |
| def test_unloadable_nonascii(self): |
| '''Test behavior with nonexistent module with non-ASCII name''' |
| name = 'fo\xf3' |
| with self.assertRaises(ImportError) as cm: |
| self.load_module_by_name(name) |
| self.assertEqual(cm.exception.name, name) |
| |
| def test_nonmodule(self): |
| '''Test returning a non-module object from create works''' |
| name = self.name + '_nonmodule' |
| mod = self.load_module_by_name(name) |
| self.assertNotEqual(type(mod), type(unittest)) |
| self.assertEqual(mod.three, 3) |
| |
| def test_null_slots(self): |
| '''Test that NULL slots aren't a problem''' |
| name = self.name + '_null_slots' |
| module = self.load_module_by_name(name) |
| self.assertIsInstance(module, types.ModuleType) |
| self.assertEqual(module.__name__, name) |
| |
| def test_bad_modules(self): |
| '''Test SystemError is raised for misbehaving extensions''' |
| for name_base in [ |
| 'bad_slot_large', |
| 'bad_slot_negative', |
| 'create_int_with_state', |
| 'negative_size', |
| 'export_null', |
| 'export_uninitialized', |
| 'export_raise', |
| 'export_unreported_exception', |
| 'create_null', |
| 'create_raise', |
| 'create_unreported_exception', |
| 'nonmodule_with_exec_slots', |
| 'exec_err', |
| 'exec_raise', |
| 'exec_unreported_exception', |
| ]: |
| with self.subTest(name_base): |
| name = self.name + '_' + name_base |
| with self.assertRaises(SystemError): |
| self.load_module_by_name(name) |
| |
| def test_nonascii(self): |
| '''Test that modules with non-ASCII names can be loaded''' |
| # punycode behaves slightly differently in some-ASCII and no-ASCII |
| # cases, so test both |
| cases = [ |
| (self.name + '_zkou\u0161ka_na\u010dten\xed', 'Czech'), |
| ('\uff3f\u30a4\u30f3\u30dd\u30fc\u30c8\u30c6\u30b9\u30c8', |
| 'Japanese'), |
| ] |
| for name, lang in cases: |
| with self.subTest(name): |
| module = self.load_module_by_name(name) |
| self.assertEqual(module.__name__, name) |
| self.assertEqual(module.__doc__, "Module named in %s" % lang) |
| |
| |
| (Frozen_MultiPhaseExtensionModuleTests, |
| Source_MultiPhaseExtensionModuleTests |
| ) = util.test_both(MultiPhaseExtensionModuleTests, machinery=machinery) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |