Python code coverage for Lib/packaging/tests/test_command_test.py

#	count	content
1	n/a	import os
2	n/a	import re
3	n/a	import sys
4	n/a	import shutil
5	n/a	import unittest as ut1
6	n/a	import packaging.database
7	n/a
8	n/a	from os.path import join
9	n/a	from operator import getitem, setitem, delitem
10	n/a	from packaging.command.build import build
11	n/a	from packaging.tests import unittest
12	n/a	from packaging.tests.support import (TempdirManager, EnvironRestorer,
13	n/a	LoggingCatcher)
14	n/a	from packaging.command.test import test
15	n/a	from packaging.command import set_command
16	n/a	from packaging.dist import Distribution
17	n/a
18	n/a
19	n/a	EXPECTED_OUTPUT_RE = r'''FAIL: test_blah \(myowntestmodule.SomeTest\)
20	n/a	----------------------------------------------------------------------
21	n/a	Traceback \(most recent call last\):
22	n/a	File ".+/myowntestmodule.py", line \d+, in test_blah
23	n/a	self.fail\("horribly"\)
24	n/a	AssertionError: horribly
25	n/a	'''
26	n/a
27	n/a	here = os.path.dirname(os.path.abspath(__file__))
28	n/a
29	n/a
30	n/a	class MockBuildCmd(build):
31	n/a	build_lib = "mock build lib"
32	n/a	command_name = 'build'
33	n/a	plat_name = 'whatever'
34	n/a
35	n/a	def initialize_options(self):
36	n/a	pass
37	n/a
38	n/a	def finalize_options(self):
39	n/a	pass
40	n/a
41	n/a	def run(self):
42	n/a	self._record.append("build has run")
43	n/a
44	n/a
45	n/a	class TestTest(TempdirManager,
46	n/a	EnvironRestorer,
47	n/a	LoggingCatcher,
48	n/a	unittest.TestCase):
49	n/a
50	n/a	restore_environ = ['PYTHONPATH']
51	n/a
52	n/a	def setUp(self):
53	n/a	super(TestTest, self).setUp()
54	n/a	self.addCleanup(packaging.database.clear_cache)
55	n/a	new_pythonpath = os.path.dirname(os.path.dirname(here))
56	n/a	pythonpath = os.environ.get('PYTHONPATH')
57	n/a	if pythonpath is not None:
58	n/a	new_pythonpath = os.pathsep.join((new_pythonpath, pythonpath))
59	n/a	os.environ['PYTHONPATH'] = new_pythonpath
60	n/a
61	n/a	def assert_re_match(self, pattern, string):
62	n/a	def quote(s):
63	n/a	lines = ['## ' + line for line in s.split('\n')]
64	n/a	sep = ["#" * 60]
65	n/a	return [''] + sep + lines + sep
66	n/a	msg = quote(pattern) + ["didn't match"] + quote(string)
67	n/a	msg = "\n".join(msg)
68	n/a	if not re.search(pattern, string):
69	n/a	self.fail(msg)
70	n/a
71	n/a	def prepare_dist(self, dist_name):
72	n/a	pkg_dir = join(os.path.dirname(__file__), "dists", dist_name)
73	n/a	temp_pkg_dir = join(self.mkdtemp(), dist_name)
74	n/a	shutil.copytree(pkg_dir, temp_pkg_dir)
75	n/a	return temp_pkg_dir
76	n/a
77	n/a	def safely_replace(self, obj, attr,
78	n/a	new_val=None, delete=False, dictionary=False):
79	n/a	"""Replace a object's attribute returning to its original state at the
80	n/a	end of the test run. Creates the attribute if not present before
81	n/a	(deleting afterwards). When delete=True, makes sure the value is del'd
82	n/a	for the test run. If dictionary is set to True, operates of its items
83	n/a	rather than attributes."""
84	n/a	if dictionary:
85	n/a	_setattr, _getattr, _delattr = setitem, getitem, delitem
86	n/a
87	n/a	def _hasattr(_dict, value):
88	n/a	return value in _dict
89	n/a	else:
90	n/a	_setattr, _getattr, _delattr, _hasattr = (setattr, getattr,
91	n/a	delattr, hasattr)
92	n/a
93	n/a	orig_has_attr = _hasattr(obj, attr)
94	n/a	if orig_has_attr:
95	n/a	orig_val = _getattr(obj, attr)
96	n/a
97	n/a	if delete is False:
98	n/a	_setattr(obj, attr, new_val)
99	n/a	elif orig_has_attr:
100	n/a	_delattr(obj, attr)
101	n/a
102	n/a	def do_cleanup():
103	n/a	if orig_has_attr:
104	n/a	_setattr(obj, attr, orig_val)
105	n/a	elif _hasattr(obj, attr):
106	n/a	_delattr(obj, attr)
107	n/a
108	n/a	self.addCleanup(do_cleanup)
109	n/a
110	n/a	def test_runs_unittest(self):
111	n/a	module_name, a_module = self.prepare_a_module()
112	n/a	record = []
113	n/a	a_module.recorder = lambda *args: record.append("suite")
114	n/a
115	n/a	class MockTextTestRunner:
116	n/a	def __init__(_, *__):
117	n/a	pass
118	n/a
119	n/a	def run(_self, suite):
120	n/a	record.append("run")
121	n/a
122	n/a	self.safely_replace(ut1, "TextTestRunner", MockTextTestRunner)
123	n/a
124	n/a	dist = Distribution()
125	n/a	cmd = test(dist)
126	n/a	cmd.suite = "%s.recorder" % module_name
127	n/a	cmd.run()
128	n/a	self.assertEqual(record, ["suite", "run"])
129	n/a
130	n/a	def test_builds_before_running_tests(self):
131	n/a	self.addCleanup(set_command, 'packaging.command.build.build')
132	n/a	set_command('packaging.tests.test_command_test.MockBuildCmd')
133	n/a
134	n/a	dist = Distribution()
135	n/a	dist.get_command_obj('build')._record = record = []
136	n/a	cmd = test(dist)
137	n/a	cmd.runner = self.prepare_named_function(lambda: None)
138	n/a	cmd.ensure_finalized()
139	n/a	cmd.run()
140	n/a	self.assertEqual(['build has run'], record)
141	n/a
142	n/a	@unittest.skip('needs to be written')
143	n/a	def test_works_with_2to3(self):
144	n/a	pass
145	n/a
146	n/a	def test_checks_requires(self):
147	n/a	dist = Distribution()
148	n/a	cmd = test(dist)
149	n/a	phony_project = 'ohno_ohno-impossible_1234-name_stop-that!'
150	n/a	cmd.tests_require = [phony_project]
151	n/a	cmd.ensure_finalized()
152	n/a	logs = self.get_logs()
153	n/a	self.assertIn(phony_project, logs[-1])
154	n/a
155	n/a	def prepare_a_module(self):
156	n/a	tmp_dir = self.mkdtemp()
157	n/a	sys.path.append(tmp_dir)
158	n/a	self.addCleanup(sys.path.remove, tmp_dir)
159	n/a
160	n/a	self.write_file((tmp_dir, 'packaging_tests_a.py'), '')
161	n/a	import packaging_tests_a as a_module
162	n/a	return "packaging_tests_a", a_module
163	n/a
164	n/a	def prepare_named_function(self, func):
165	n/a	module_name, a_module = self.prepare_a_module()
166	n/a	a_module.recorder = func
167	n/a	return "%s.recorder" % module_name
168	n/a
169	n/a	def test_custom_runner(self):
170	n/a	dist = Distribution()
171	n/a	cmd = test(dist)
172	n/a	record = []
173	n/a	cmd.runner = self.prepare_named_function(
174	n/a	lambda: record.append("runner called"))
175	n/a	cmd.ensure_finalized()
176	n/a	cmd.run()
177	n/a	self.assertEqual(["runner called"], record)
178	n/a
179	n/a	def prepare_mock_ut2(self):
180	n/a	class MockUTClass:
181	n/a	def __init__(_, *__):
182	n/a	pass
183	n/a
184	n/a	def discover(self):
185	n/a	pass
186	n/a
187	n/a	def run(self, _):
188	n/a	pass
189	n/a
190	n/a	class MockUTModule:
191	n/a	TestLoader = MockUTClass
192	n/a	TextTestRunner = MockUTClass
193	n/a
194	n/a	mock_ut2 = MockUTModule()
195	n/a	self.safely_replace(sys.modules, "unittest2",
196	n/a	mock_ut2, dictionary=True)
197	n/a	return mock_ut2
198	n/a
199	n/a	def test_gets_unittest_discovery(self):
200	n/a	mock_ut2 = self.prepare_mock_ut2()
201	n/a	dist = Distribution()
202	n/a	cmd = test(dist)
203	n/a	self.safely_replace(ut1.TestLoader, "discover", lambda: None)
204	n/a	self.assertEqual(cmd.get_ut_with_discovery(), ut1)
205	n/a
206	n/a	del ut1.TestLoader.discover
207	n/a	self.assertEqual(cmd.get_ut_with_discovery(), mock_ut2)
208	n/a
209	n/a	def test_calls_discover(self):
210	n/a	self.safely_replace(ut1.TestLoader, "discover", delete=True)
211	n/a	mock_ut2 = self.prepare_mock_ut2()
212	n/a	record = []
213	n/a	mock_ut2.TestLoader.discover = lambda self, path: record.append(path)
214	n/a	dist = Distribution()
215	n/a	cmd = test(dist)
216	n/a	cmd.run()
217	n/a	self.assertEqual([os.curdir], record)
218	n/a
219	n/a
220	n/a	def test_suite():
221	n/a	return unittest.makeSuite(TestTest)
222	n/a
223	n/a	if __name__ == "__main__":
224	n/a	unittest.main(defaultTest="test_suite")