Python code coverage for Lib/importlib/test/import_/test_meta_path.py

#	count	content
1	n/a	from .. import util
2	n/a	from . import util as import_util
3	n/a	import importlib._bootstrap
4	n/a	import sys
5	n/a	from types import MethodType
6	n/a	import unittest
7	n/a	import warnings
8	n/a
9	n/a
10	n/a	class CallingOrder(unittest.TestCase):
11	n/a
12	n/a	"""Calls to the importers on sys.meta_path happen in order that they are
13	n/a	specified in the sequence, starting with the first importer
14	n/a	[first called], and then continuing on down until one is found that doesn't
15	n/a	return None [continuing]."""
16	n/a
17	n/a
18	n/a	def test_first_called(self):
19	n/a	# [first called]
20	n/a	mod = 'top_level'
21	n/a	first = util.mock_modules(mod)
22	n/a	second = util.mock_modules(mod)
23	n/a	with util.mock_modules(mod) as first, util.mock_modules(mod) as second:
24	n/a	first.modules[mod] = 42
25	n/a	second.modules[mod] = -13
26	n/a	with util.import_state(meta_path=[first, second]):
27	n/a	self.assertEqual(import_util.import_(mod), 42)
28	n/a
29	n/a	def test_continuing(self):
30	n/a	# [continuing]
31	n/a	mod_name = 'for_real'
32	n/a	with util.mock_modules('nonexistent') as first, \
33	n/a	util.mock_modules(mod_name) as second:
34	n/a	first.find_module = lambda self, fullname, path=None: None
35	n/a	second.modules[mod_name] = 42
36	n/a	with util.import_state(meta_path=[first, second]):
37	n/a	self.assertEqual(import_util.import_(mod_name), 42)
38	n/a
39	n/a	def test_empty(self):
40	n/a	# Raise an ImportWarning if sys.meta_path is empty.
41	n/a	module_name = 'nothing'
42	n/a	try:
43	n/a	del sys.modules[module_name]
44	n/a	except KeyError:
45	n/a	pass
46	n/a	with util.import_state(meta_path=[]):
47	n/a	with warnings.catch_warnings(record=True) as w:
48	n/a	warnings.simplefilter('always')
49	n/a	self.assertIsNone(importlib._bootstrap._find_module('nothing',
50	n/a	None))
51	n/a	self.assertEqual(len(w), 1)
52	n/a	self.assertTrue(issubclass(w[-1].category, ImportWarning))
53	n/a
54	n/a
55	n/a	class CallSignature(unittest.TestCase):
56	n/a
57	n/a	"""If there is no __path__ entry on the parent module, then 'path' is None
58	n/a	[no path]. Otherwise, the value for __path__ is passed in for the 'path'
59	n/a	argument [path set]."""
60	n/a
61	n/a	def log(self, fxn):
62	n/a	log = []
63	n/a	def wrapper(self, args, *kwargs):
64	n/a	log.append([args, kwargs])
65	n/a	return fxn(args, *kwargs)
66	n/a	return log, wrapper
67	n/a
68	n/a
69	n/a	def test_no_path(self):
70	n/a	# [no path]
71	n/a	mod_name = 'top_level'
72	n/a	assert '.' not in mod_name
73	n/a	with util.mock_modules(mod_name) as importer:
74	n/a	log, wrapped_call = self.log(importer.find_module)
75	n/a	importer.find_module = MethodType(wrapped_call, importer)
76	n/a	with util.import_state(meta_path=[importer]):
77	n/a	import_util.import_(mod_name)
78	n/a	assert len(log) == 1
79	n/a	args = log[0][0]
80	n/a	kwargs = log[0][1]
81	n/a	# Assuming all arguments are positional.
82	n/a	self.assertEqual(len(args), 2)
83	n/a	self.assertEqual(len(kwargs), 0)
84	n/a	self.assertEqual(args[0], mod_name)
85	n/a	self.assertIsNone(args[1])
86	n/a
87	n/a	def test_with_path(self):
88	n/a	# [path set]
89	n/a	pkg_name = 'pkg'
90	n/a	mod_name = pkg_name + '.module'
91	n/a	path = [42]
92	n/a	assert '.' in mod_name
93	n/a	with util.mock_modules(pkg_name+'.__init__', mod_name) as importer:
94	n/a	importer.modules[pkg_name].__path__ = path
95	n/a	log, wrapped_call = self.log(importer.find_module)
96	n/a	importer.find_module = MethodType(wrapped_call, importer)
97	n/a	with util.import_state(meta_path=[importer]):
98	n/a	import_util.import_(mod_name)
99	n/a	assert len(log) == 2
100	n/a	args = log[1][0]
101	n/a	kwargs = log[1][1]
102	n/a	# Assuming all arguments are positional.
103	n/a	self.assertTrue(not kwargs)
104	n/a	self.assertEqual(args[0], mod_name)
105	n/a	self.assertIs(args[1], path)
106	n/a
107	n/a
108	n/a
109	n/a	def test_main():
110	n/a	from test.support import run_unittest
111	n/a	run_unittest(CallingOrder, CallSignature)
112	n/a
113	n/a
114	n/a	if __name__ == '__main__':
115	n/a	test_main()