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

#	count	content
1	n/a	from contextlib import contextmanager
2	n/a	import imp
3	n/a	import os.path
4	n/a	from test import support
5	n/a	import unittest
6	n/a	import sys
7	n/a
8	n/a
9	n/a	CASE_INSENSITIVE_FS = True
10	n/a	# Windows is the only OS that is always case-insensitive
11	n/a	# (OS X can be case-sensitive).
12	n/a	if sys.platform not in ('win32', 'cygwin'):
13	n/a	changed_name = __file__.upper()
14	n/a	if changed_name == __file__:
15	n/a	changed_name = __file__.lower()
16	n/a	if not os.path.exists(changed_name):
17	n/a	CASE_INSENSITIVE_FS = False
18	n/a
19	n/a
20	n/a	def case_insensitive_tests(test):
21	n/a	"""Class decorator that nullifies tests requiring a case-insensitive
22	n/a	file system."""
23	n/a	return unittest.skipIf(not CASE_INSENSITIVE_FS,
24	n/a	"requires a case-insensitive filesystem")(test)
25	n/a
26	n/a
27	n/a	@contextmanager
28	n/a	def uncache(*names):
29	n/a	"""Uncache a module from sys.modules.
30	n/a
31	n/a	A basic sanity check is performed to prevent uncaching modules that either
32	n/a	cannot/shouldn't be uncached.
33	n/a
34	n/a	"""
35	n/a	for name in names:
36	n/a	if name in ('sys', 'marshal', 'imp'):
37	n/a	raise ValueError(
38	n/a	"cannot uncache {0}".format(name))
39	n/a	try:
40	n/a	del sys.modules[name]
41	n/a	except KeyError:
42	n/a	pass
43	n/a	try:
44	n/a	yield
45	n/a	finally:
46	n/a	for name in names:
47	n/a	try:
48	n/a	del sys.modules[name]
49	n/a	except KeyError:
50	n/a	pass
51	n/a
52	n/a	@contextmanager
53	n/a	def import_state(**kwargs):
54	n/a	"""Context manager to manage the various importers and stored state in the
55	n/a	sys module.
56	n/a
57	n/a	The 'modules' attribute is not supported as the interpreter state stores a
58	n/a	pointer to the dict that the interpreter uses internally;
59	n/a	reassigning to sys.modules does not have the desired effect.
60	n/a
61	n/a	"""
62	n/a	originals = {}
63	n/a	try:
64	n/a	for attr, default in (('meta_path', []), ('path', []),
65	n/a	('path_hooks', []),
66	n/a	('path_importer_cache', {})):
67	n/a	originals[attr] = getattr(sys, attr)
68	n/a	if attr in kwargs:
69	n/a	new_value = kwargs[attr]
70	n/a	del kwargs[attr]
71	n/a	else:
72	n/a	new_value = default
73	n/a	setattr(sys, attr, new_value)
74	n/a	if len(kwargs):
75	n/a	raise ValueError(
76	n/a	'unrecognized arguments: {0}'.format(kwargs.keys()))
77	n/a	yield
78	n/a	finally:
79	n/a	for attr, value in originals.items():
80	n/a	setattr(sys, attr, value)
81	n/a
82	n/a
83	n/a	class mock_modules:
84	n/a
85	n/a	"""A mock importer/loader."""
86	n/a
87	n/a	def __init__(self, *names, module_code={}):
88	n/a	self.modules = {}
89	n/a	self.module_code = {}
90	n/a	for name in names:
91	n/a	if not name.endswith('.__init__'):
92	n/a	import_name = name
93	n/a	else:
94	n/a	import_name = name[:-len('.__init__')]
95	n/a	if '.' not in name:
96	n/a	package = None
97	n/a	elif import_name == name:
98	n/a	package = name.rsplit('.', 1)[0]
99	n/a	else:
100	n/a	package = import_name
101	n/a	module = imp.new_module(import_name)
102	n/a	module.__loader__ = self
103	n/a	module.__file__ = '<mock __file__>'
104	n/a	module.__package__ = package
105	n/a	module.attr = name
106	n/a	if import_name != name:
107	n/a	module.__path__ = ['<mock __path__>']
108	n/a	self.modules[import_name] = module
109	n/a	if import_name in module_code:
110	n/a	self.module_code[import_name] = module_code[import_name]
111	n/a
112	n/a	def __getitem__(self, name):
113	n/a	return self.modules[name]
114	n/a
115	n/a	def find_module(self, fullname, path=None):
116	n/a	if fullname not in self.modules:
117	n/a	return None
118	n/a	else:
119	n/a	return self
120	n/a
121	n/a	def load_module(self, fullname):
122	n/a	if fullname not in self.modules:
123	n/a	raise ImportError
124	n/a	else:
125	n/a	sys.modules[fullname] = self.modules[fullname]
126	n/a	if fullname in self.module_code:
127	n/a	try:
128	n/a	self.module_code[fullname]()
129	n/a	except Exception:
130	n/a	del sys.modules[fullname]
131	n/a	raise
132	n/a	return self.modules[fullname]
133	n/a
134	n/a	def __enter__(self):
135	n/a	self._uncache = uncache(*self.modules.keys())
136	n/a	self._uncache.__enter__()
137	n/a	return self
138	n/a
139	n/a	def __exit__(self, *exc_info):
140	n/a	self._uncache.__exit__(None, None, None)