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

#	count	content
1	n/a	from .. import util
2	n/a	import contextlib
3	n/a	import errno
4	n/a	import functools
5	n/a	import imp
6	n/a	import os
7	n/a	import os.path
8	n/a	import sys
9	n/a	import tempfile
10	n/a	from test import support
11	n/a
12	n/a
13	n/a	def writes_bytecode_files(fxn):
14	n/a	"""Decorator to protect sys.dont_write_bytecode from mutation and to skip
15	n/a	tests that require it to be set to False."""
16	n/a	if sys.dont_write_bytecode:
17	n/a	return lambda args, *kwargs: None
18	n/a	@functools.wraps(fxn)
19	n/a	def wrapper(args, *kwargs):
20	n/a	original = sys.dont_write_bytecode
21	n/a	sys.dont_write_bytecode = False
22	n/a	try:
23	n/a	to_return = fxn(args, *kwargs)
24	n/a	finally:
25	n/a	sys.dont_write_bytecode = original
26	n/a	return to_return
27	n/a	return wrapper
28	n/a
29	n/a
30	n/a	def ensure_bytecode_path(bytecode_path):
31	n/a	"""Ensure that the __pycache__ directory for PEP 3147 pyc file exists.
32	n/a
33	n/a	:param bytecode_path: File system path to PEP 3147 pyc file.
34	n/a	"""
35	n/a	try:
36	n/a	os.mkdir(os.path.dirname(bytecode_path))
37	n/a	except OSError as error:
38	n/a	if error.errno != errno.EEXIST:
39	n/a	raise
40	n/a
41	n/a
42	n/a	@contextlib.contextmanager
43	n/a	def create_modules(*names):
44	n/a	"""Temporarily create each named module with an attribute (named 'attr')
45	n/a	that contains the name passed into the context manager that caused the
46	n/a	creation of the module.
47	n/a
48	n/a	All files are created in a temporary directory returned by
49	n/a	tempfile.mkdtemp(). This directory is inserted at the beginning of
50	n/a	sys.path. When the context manager exits all created files (source and
51	n/a	bytecode) are explicitly deleted.
52	n/a
53	n/a	No magic is performed when creating packages! This means that if you create
54	n/a	a module within a package you must also create the package's __init__ as
55	n/a	well.
56	n/a
57	n/a	"""
58	n/a	source = 'attr = {0!r}'
59	n/a	created_paths = []
60	n/a	mapping = {}
61	n/a	state_manager = None
62	n/a	uncache_manager = None
63	n/a	try:
64	n/a	temp_dir = tempfile.mkdtemp()
65	n/a	mapping['.root'] = temp_dir
66	n/a	import_names = set()
67	n/a	for name in names:
68	n/a	if not name.endswith('__init__'):
69	n/a	import_name = name
70	n/a	else:
71	n/a	import_name = name[:-len('.__init__')]
72	n/a	import_names.add(import_name)
73	n/a	if import_name in sys.modules:
74	n/a	del sys.modules[import_name]
75	n/a	name_parts = name.split('.')
76	n/a	file_path = temp_dir
77	n/a	for directory in name_parts[:-1]:
78	n/a	file_path = os.path.join(file_path, directory)
79	n/a	if not os.path.exists(file_path):
80	n/a	os.mkdir(file_path)
81	n/a	created_paths.append(file_path)
82	n/a	file_path = os.path.join(file_path, name_parts[-1] + '.py')
83	n/a	with open(file_path, 'w') as file:
84	n/a	file.write(source.format(name))
85	n/a	created_paths.append(file_path)
86	n/a	mapping[name] = file_path
87	n/a	uncache_manager = util.uncache(*import_names)
88	n/a	uncache_manager.__enter__()
89	n/a	state_manager = util.import_state(path=[temp_dir])
90	n/a	state_manager.__enter__()
91	n/a	yield mapping
92	n/a	finally:
93	n/a	if state_manager is not None:
94	n/a	state_manager.__exit__(None, None, None)
95	n/a	if uncache_manager is not None:
96	n/a	uncache_manager.__exit__(None, None, None)
97	n/a	support.rmtree(temp_dir)