Python code coverage for Lib/importlib/test/test_util.py

#	count	content
1	n/a	from importlib import util
2	n/a	from . import util as test_util
3	n/a	import imp
4	n/a	import sys
5	n/a	import types
6	n/a	import unittest
7	n/a
8	n/a
9	n/a	class ModuleForLoaderTests(unittest.TestCase):
10	n/a
11	n/a	"""Tests for importlib.util.module_for_loader."""
12	n/a
13	n/a	def return_module(self, name):
14	n/a	fxn = util.module_for_loader(lambda self, module: module)
15	n/a	return fxn(self, name)
16	n/a
17	n/a	def raise_exception(self, name):
18	n/a	def to_wrap(self, module):
19	n/a	raise ImportError
20	n/a	fxn = util.module_for_loader(to_wrap)
21	n/a	try:
22	n/a	fxn(self, name)
23	n/a	except ImportError:
24	n/a	pass
25	n/a
26	n/a	def test_new_module(self):
27	n/a	# Test that when no module exists in sys.modules a new module is
28	n/a	# created.
29	n/a	module_name = 'a.b.c'
30	n/a	with test_util.uncache(module_name):
31	n/a	module = self.return_module(module_name)
32	n/a	self.assertIn(module_name, sys.modules)
33	n/a	self.assertIsInstance(module, types.ModuleType)
34	n/a	self.assertEqual(module.__name__, module_name)
35	n/a
36	n/a	def test_reload(self):
37	n/a	# Test that a module is reused if already in sys.modules.
38	n/a	name = 'a.b.c'
39	n/a	module = imp.new_module('a.b.c')
40	n/a	with test_util.uncache(name):
41	n/a	sys.modules[name] = module
42	n/a	returned_module = self.return_module(name)
43	n/a	self.assertIs(returned_module, sys.modules[name])
44	n/a
45	n/a	def test_new_module_failure(self):
46	n/a	# Test that a module is removed from sys.modules if added but an
47	n/a	# exception is raised.
48	n/a	name = 'a.b.c'
49	n/a	with test_util.uncache(name):
50	n/a	self.raise_exception(name)
51	n/a	self.assertNotIn(name, sys.modules)
52	n/a
53	n/a	def test_reload_failure(self):
54	n/a	# Test that a failure on reload leaves the module in-place.
55	n/a	name = 'a.b.c'
56	n/a	module = imp.new_module(name)
57	n/a	with test_util.uncache(name):
58	n/a	sys.modules[name] = module
59	n/a	self.raise_exception(name)
60	n/a	self.assertIs(module, sys.modules[name])
61	n/a
62	n/a	def test_decorator_attrs(self):
63	n/a	def fxn(self, module): pass
64	n/a	wrapped = util.module_for_loader(fxn)
65	n/a	self.assertEqual(wrapped.__name__, fxn.__name__)
66	n/a	self.assertEqual(wrapped.__qualname__, fxn.__qualname__)
67	n/a
68	n/a	def test_false_module(self):
69	n/a	# If for some odd reason a module is considered false, still return it
70	n/a	# from sys.modules.
71	n/a	class FalseModule(types.ModuleType):
72	n/a	def __bool__(self): return False
73	n/a
74	n/a	name = 'mod'
75	n/a	module = FalseModule(name)
76	n/a	with test_util.uncache(name):
77	n/a	self.assertFalse(module)
78	n/a	sys.modules[name] = module
79	n/a	given = self.return_module(name)
80	n/a	self.assertIs(given, module)
81	n/a
82	n/a	def test_attributes_set(self):
83	n/a	# __name__, __loader__, and __package__ should be set (when
84	n/a	# is_package() is defined; undefined implicitly tested elsewhere).
85	n/a	class FakeLoader:
86	n/a	def __init__(self, is_package):
87	n/a	self._pkg = is_package
88	n/a	def is_package(self, name):
89	n/a	return self._pkg
90	n/a	@util.module_for_loader
91	n/a	def load_module(self, module):
92	n/a	return module
93	n/a
94	n/a	name = 'pkg.mod'
95	n/a	with test_util.uncache(name):
96	n/a	loader = FakeLoader(False)
97	n/a	module = loader.load_module(name)
98	n/a	self.assertEqual(module.__name__, name)
99	n/a	self.assertIs(module.__loader__, loader)
100	n/a	self.assertEqual(module.__package__, 'pkg')
101	n/a
102	n/a	name = 'pkg.sub'
103	n/a	with test_util.uncache(name):
104	n/a	loader = FakeLoader(True)
105	n/a	module = loader.load_module(name)
106	n/a	self.assertEqual(module.__name__, name)
107	n/a	self.assertIs(module.__loader__, loader)
108	n/a	self.assertEqual(module.__package__, name)
109	n/a
110	n/a
111	n/a	class SetPackageTests(unittest.TestCase):
112	n/a
113	n/a	"""Tests for importlib.util.set_package."""
114	n/a
115	n/a	def verify(self, module, expect):
116	n/a	"""Verify the module has the expected value for __package__ after
117	n/a	passing through set_package."""
118	n/a	fxn = lambda: module
119	n/a	wrapped = util.set_package(fxn)
120	n/a	wrapped()
121	n/a	self.assertTrue(hasattr(module, '__package__'))
122	n/a	self.assertEqual(expect, module.__package__)
123	n/a
124	n/a	def test_top_level(self):
125	n/a	# __package__ should be set to the empty string if a top-level module.
126	n/a	# Implicitly tests when package is set to None.
127	n/a	module = imp.new_module('module')
128	n/a	module.__package__ = None
129	n/a	self.verify(module, '')
130	n/a
131	n/a	def test_package(self):
132	n/a	# Test setting __package__ for a package.
133	n/a	module = imp.new_module('pkg')
134	n/a	module.__path__ = ['<path>']
135	n/a	module.__package__ = None
136	n/a	self.verify(module, 'pkg')
137	n/a
138	n/a	def test_submodule(self):
139	n/a	# Test __package__ for a module in a package.
140	n/a	module = imp.new_module('pkg.mod')
141	n/a	module.__package__ = None
142	n/a	self.verify(module, 'pkg')
143	n/a
144	n/a	def test_setting_if_missing(self):
145	n/a	# __package__ should be set if it is missing.
146	n/a	module = imp.new_module('mod')
147	n/a	if hasattr(module, '__package__'):
148	n/a	delattr(module, '__package__')
149	n/a	self.verify(module, '')
150	n/a
151	n/a	def test_leaving_alone(self):
152	n/a	# If __package__ is set and not None then leave it alone.
153	n/a	for value in (True, False):
154	n/a	module = imp.new_module('mod')
155	n/a	module.__package__ = value
156	n/a	self.verify(module, value)
157	n/a
158	n/a	def test_decorator_attrs(self):
159	n/a	def fxn(module): pass
160	n/a	wrapped = util.set_package(fxn)
161	n/a	self.assertEqual(wrapped.__name__, fxn.__name__)
162	n/a	self.assertEqual(wrapped.__qualname__, fxn.__qualname__)
163	n/a
164	n/a
165	n/a	class ResolveNameTests(unittest.TestCase):
166	n/a
167	n/a	"""Tests importlib.util.resolve_name()."""
168	n/a
169	n/a	def test_absolute(self):
170	n/a	# bacon
171	n/a	self.assertEqual('bacon', util.resolve_name('bacon', None))
172	n/a
173	n/a	def test_aboslute_within_package(self):
174	n/a	# bacon in spam
175	n/a	self.assertEqual('bacon', util.resolve_name('bacon', 'spam'))
176	n/a
177	n/a	def test_no_package(self):
178	n/a	# .bacon in ''
179	n/a	with self.assertRaises(ValueError):
180	n/a	util.resolve_name('.bacon', '')
181	n/a
182	n/a	def test_in_package(self):
183	n/a	# .bacon in spam
184	n/a	self.assertEqual('spam.eggs.bacon',
185	n/a	util.resolve_name('.bacon', 'spam.eggs'))
186	n/a
187	n/a	def test_other_package(self):
188	n/a	# ..bacon in spam.bacon
189	n/a	self.assertEqual('spam.bacon',
190	n/a	util.resolve_name('..bacon', 'spam.eggs'))
191	n/a
192	n/a	def test_escape(self):
193	n/a	# ..bacon in spam
194	n/a	with self.assertRaises(ValueError):
195	n/a	util.resolve_name('..bacon', 'spam')
196	n/a
197	n/a
198	n/a	def test_main():
199	n/a	from test import support
200	n/a	support.run_unittest(
201	n/a	ModuleForLoaderTests,
202	n/a	SetPackageTests,
203	n/a	ResolveNameTests
204	n/a	)
205	n/a
206	n/a
207	n/a	if __name__ == '__main__':
208	n/a	test_main()