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

#	count	content
1	n/a	"""Tests for packaging.depgraph """
2	n/a	import os
3	n/a	import re
4	n/a	import sys
5	n/a	from io import StringIO
6	n/a
7	n/a	from packaging import depgraph
8	n/a	from packaging.database import get_distribution, enable_cache, disable_cache
9	n/a
10	n/a	from packaging.tests import unittest, support
11	n/a	from packaging.tests.support import requires_zlib
12	n/a
13	n/a
14	n/a	class DepGraphTestCase(support.LoggingCatcher,
15	n/a	unittest.TestCase):
16	n/a
17	n/a	DISTROS_DIST = ('choxie', 'grammar', 'towel-stuff')
18	n/a	DISTROS_EGG = ('bacon', 'banana', 'strawberry', 'cheese')
19	n/a	BAD_EGGS = ('nut',)
20	n/a
21	n/a	EDGE = re.compile(
22	n/a	r'"(?P<from>.)" -> "(?P<to>.)" \[label="(?P<label>.*)"\]')
23	n/a
24	n/a	def checkLists(self, l1, l2):
25	n/a	""" Compare two lists without taking the order into consideration """
26	n/a	self.assertListEqual(sorted(l1), sorted(l2))
27	n/a
28	n/a	def setUp(self):
29	n/a	super(DepGraphTestCase, self).setUp()
30	n/a	path = os.path.join(os.path.dirname(__file__), 'fake_dists')
31	n/a	path = os.path.abspath(path)
32	n/a	sys.path.insert(0, path)
33	n/a	self.addCleanup(sys.path.remove, path)
34	n/a	self.addCleanup(enable_cache)
35	n/a	disable_cache()
36	n/a
37	n/a	def test_generate_graph(self):
38	n/a	dists = []
39	n/a	for name in self.DISTROS_DIST:
40	n/a	dist = get_distribution(name)
41	n/a	self.assertNotEqual(dist, None)
42	n/a	dists.append(dist)
43	n/a
44	n/a	choxie, grammar, towel = dists
45	n/a
46	n/a	graph = depgraph.generate_graph(dists)
47	n/a
48	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[choxie]]
49	n/a	self.checkLists([('towel-stuff', 'towel-stuff (0.1)')], deps)
50	n/a	self.assertIn(choxie, graph.reverse_list[towel])
51	n/a	self.checkLists(graph.missing[choxie], ['nut'])
52	n/a
53	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[grammar]]
54	n/a	self.checkLists([], deps)
55	n/a	self.checkLists(graph.missing[grammar], ['truffles (>=1.2)'])
56	n/a
57	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[towel]]
58	n/a	self.checkLists([], deps)
59	n/a	self.checkLists(graph.missing[towel], ['bacon (<=0.2)'])
60	n/a
61	n/a	@requires_zlib
62	n/a	def test_generate_graph_egg(self):
63	n/a	dists = []
64	n/a	for name in self.DISTROS_DIST + self.DISTROS_EGG:
65	n/a	dist = get_distribution(name, use_egg_info=True)
66	n/a	self.assertNotEqual(dist, None)
67	n/a	dists.append(dist)
68	n/a
69	n/a	choxie, grammar, towel, bacon, banana, strawberry, cheese = dists
70	n/a
71	n/a	graph = depgraph.generate_graph(dists)
72	n/a
73	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[choxie]]
74	n/a	self.checkLists([('towel-stuff', 'towel-stuff (0.1)')], deps)
75	n/a	self.assertIn(choxie, graph.reverse_list[towel])
76	n/a	self.checkLists(graph.missing[choxie], ['nut'])
77	n/a
78	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[grammar]]
79	n/a	self.checkLists([('bacon', 'truffles (>=1.2)')], deps)
80	n/a	self.checkLists(graph.missing[grammar], [])
81	n/a	self.assertIn(grammar, graph.reverse_list[bacon])
82	n/a
83	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[towel]]
84	n/a	self.checkLists([('bacon', 'bacon (<=0.2)')], deps)
85	n/a	self.checkLists(graph.missing[towel], [])
86	n/a	self.assertIn(towel, graph.reverse_list[bacon])
87	n/a
88	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[bacon]]
89	n/a	self.checkLists([], deps)
90	n/a	self.checkLists(graph.missing[bacon], [])
91	n/a
92	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[banana]]
93	n/a	self.checkLists([('strawberry', 'strawberry (>=0.5)')], deps)
94	n/a	self.checkLists(graph.missing[banana], [])
95	n/a	self.assertIn(banana, graph.reverse_list[strawberry])
96	n/a
97	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[strawberry]]
98	n/a	self.checkLists([], deps)
99	n/a	self.checkLists(graph.missing[strawberry], [])
100	n/a
101	n/a	deps = [(x.name, y) for x, y in graph.adjacency_list[cheese]]
102	n/a	self.checkLists([], deps)
103	n/a	self.checkLists(graph.missing[cheese], [])
104	n/a
105	n/a	def test_dependent_dists(self):
106	n/a	dists = []
107	n/a	for name in self.DISTROS_DIST:
108	n/a	dist = get_distribution(name)
109	n/a	self.assertNotEqual(dist, None)
110	n/a	dists.append(dist)
111	n/a
112	n/a	choxie, grammar, towel = dists
113	n/a
114	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, choxie)]
115	n/a	self.checkLists([], deps)
116	n/a
117	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, grammar)]
118	n/a	self.checkLists([], deps)
119	n/a
120	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, towel)]
121	n/a	self.checkLists(['choxie'], deps)
122	n/a
123	n/a	@requires_zlib
124	n/a	def test_dependent_dists_egg(self):
125	n/a	dists = []
126	n/a	for name in self.DISTROS_DIST + self.DISTROS_EGG:
127	n/a	dist = get_distribution(name, use_egg_info=True)
128	n/a	self.assertNotEqual(dist, None)
129	n/a	dists.append(dist)
130	n/a
131	n/a	choxie, grammar, towel, bacon, banana, strawberry, cheese = dists
132	n/a
133	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, choxie)]
134	n/a	self.checkLists([], deps)
135	n/a
136	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, grammar)]
137	n/a	self.checkLists([], deps)
138	n/a
139	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, towel)]
140	n/a	self.checkLists(['choxie'], deps)
141	n/a
142	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, bacon)]
143	n/a	self.checkLists(['choxie', 'towel-stuff', 'grammar'], deps)
144	n/a
145	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, strawberry)]
146	n/a	self.checkLists(['banana'], deps)
147	n/a
148	n/a	deps = [d.name for d in depgraph.dependent_dists(dists, cheese)]
149	n/a	self.checkLists([], deps)
150	n/a
151	n/a	@requires_zlib
152	n/a	def test_graph_to_dot(self):
153	n/a	expected = (
154	n/a	('towel-stuff', 'bacon', 'bacon (<=0.2)'),
155	n/a	('grammar', 'bacon', 'truffles (>=1.2)'),
156	n/a	('choxie', 'towel-stuff', 'towel-stuff (0.1)'),
157	n/a	('banana', 'strawberry', 'strawberry (>=0.5)'),
158	n/a	)
159	n/a
160	n/a	dists = []
161	n/a	for name in self.DISTROS_DIST + self.DISTROS_EGG:
162	n/a	dist = get_distribution(name, use_egg_info=True)
163	n/a	self.assertNotEqual(dist, None)
164	n/a	dists.append(dist)
165	n/a
166	n/a	graph = depgraph.generate_graph(dists)
167	n/a	buf = StringIO()
168	n/a	depgraph.graph_to_dot(graph, buf)
169	n/a	buf.seek(0)
170	n/a	matches = []
171	n/a	lines = buf.readlines()
172	n/a	for line in lines[1:-1]: # skip the first and the last lines
173	n/a	if line[-1] == '\n':
174	n/a	line = line[:-1]
175	n/a	match = self.EDGE.match(line.strip())
176	n/a	self.assertIsNot(match, None)
177	n/a	matches.append(match.groups())
178	n/a
179	n/a	self.checkLists(matches, expected)
180	n/a
181	n/a	@requires_zlib
182	n/a	def test_graph_disconnected_to_dot(self):
183	n/a	dependencies_expected = (
184	n/a	('towel-stuff', 'bacon', 'bacon (<=0.2)'),
185	n/a	('grammar', 'bacon', 'truffles (>=1.2)'),
186	n/a	('choxie', 'towel-stuff', 'towel-stuff (0.1)'),
187	n/a	('banana', 'strawberry', 'strawberry (>=0.5)'),
188	n/a	)
189	n/a	disconnected_expected = ('cheese', 'bacon', 'strawberry')
190	n/a
191	n/a	dists = []
192	n/a	for name in self.DISTROS_DIST + self.DISTROS_EGG:
193	n/a	dist = get_distribution(name, use_egg_info=True)
194	n/a	self.assertNotEqual(dist, None)
195	n/a	dists.append(dist)
196	n/a
197	n/a	graph = depgraph.generate_graph(dists)
198	n/a	buf = StringIO()
199	n/a	depgraph.graph_to_dot(graph, buf, skip_disconnected=False)
200	n/a	buf.seek(0)
201	n/a	lines = buf.readlines()
202	n/a
203	n/a	dependencies_lines = []
204	n/a	disconnected_lines = []
205	n/a
206	n/a	# First sort output lines into dependencies and disconnected lines.
207	n/a	# We also skip the attribute lines, and don't include the "{" and "}"
208	n/a	# lines.
209	n/a	disconnected_active = False
210	n/a	for line in lines[1:-1]: # Skip first and last line
211	n/a	if line.startswith('subgraph disconnected'):
212	n/a	disconnected_active = True
213	n/a	continue
214	n/a	if line.startswith('}') and disconnected_active:
215	n/a	disconnected_active = False
216	n/a	continue
217	n/a
218	n/a	if disconnected_active:
219	n/a	# Skip the 'label = "Disconnected"', etc. attribute lines.
220	n/a	if ' = ' not in line:
221	n/a	disconnected_lines.append(line)
222	n/a	else:
223	n/a	dependencies_lines.append(line)
224	n/a
225	n/a	dependencies_matches = []
226	n/a	for line in dependencies_lines:
227	n/a	if line[-1] == '\n':
228	n/a	line = line[:-1]
229	n/a	match = self.EDGE.match(line.strip())
230	n/a	self.assertIsNot(match, None)
231	n/a	dependencies_matches.append(match.groups())
232	n/a
233	n/a	disconnected_matches = []
234	n/a	for line in disconnected_lines:
235	n/a	if line[-1] == '\n':
236	n/a	line = line[:-1]
237	n/a	line = line.strip('"')
238	n/a	disconnected_matches.append(line)
239	n/a
240	n/a	self.checkLists(dependencies_matches, dependencies_expected)
241	n/a	self.checkLists(disconnected_matches, disconnected_expected)
242	n/a
243	n/a	@requires_zlib
244	n/a	def test_graph_bad_version_to_dot(self):
245	n/a	expected = (
246	n/a	('towel-stuff', 'bacon', 'bacon (<=0.2)'),
247	n/a	('grammar', 'bacon', 'truffles (>=1.2)'),
248	n/a	('choxie', 'towel-stuff', 'towel-stuff (0.1)'),
249	n/a	('banana', 'strawberry', 'strawberry (>=0.5)'),
250	n/a	)
251	n/a
252	n/a	dists = []
253	n/a	for name in self.DISTROS_DIST + self.DISTROS_EGG + self.BAD_EGGS:
254	n/a	dist = get_distribution(name, use_egg_info=True)
255	n/a	self.assertNotEqual(dist, None)
256	n/a	dists.append(dist)
257	n/a
258	n/a	graph = depgraph.generate_graph(dists)
259	n/a	buf = StringIO()
260	n/a	depgraph.graph_to_dot(graph, buf)
261	n/a	buf.seek(0)
262	n/a	matches = []
263	n/a	lines = buf.readlines()
264	n/a	for line in lines[1:-1]: # skip the first and the last lines
265	n/a	if line[-1] == '\n':
266	n/a	line = line[:-1]
267	n/a	match = self.EDGE.match(line.strip())
268	n/a	self.assertIsNot(match, None)
269	n/a	matches.append(match.groups())
270	n/a
271	n/a	self.checkLists(matches, expected)
272	n/a
273	n/a	@requires_zlib
274	n/a	def test_repr(self):
275	n/a	dists = []
276	n/a	for name in self.DISTROS_DIST + self.DISTROS_EGG + self.BAD_EGGS:
277	n/a	dist = get_distribution(name, use_egg_info=True)
278	n/a	self.assertNotEqual(dist, None)
279	n/a	dists.append(dist)
280	n/a
281	n/a	graph = depgraph.generate_graph(dists)
282	n/a	self.assertTrue(repr(graph))
283	n/a
284	n/a	@requires_zlib
285	n/a	def test_main(self):
286	n/a	tempout = StringIO()
287	n/a	old = sys.stdout
288	n/a	sys.stdout = tempout
289	n/a	oldargv = sys.argv[:]
290	n/a	sys.argv[:] = ['script.py']
291	n/a	try:
292	n/a	try:
293	n/a	depgraph.main()
294	n/a	except SystemExit:
295	n/a	pass
296	n/a	finally:
297	n/a	sys.stdout = old
298	n/a	sys.argv[:] = oldargv
299	n/a
300	n/a	# checks what main did XXX could do more here
301	n/a	tempout.seek(0)
302	n/a	res = tempout.read()
303	n/a	self.assertIn('towel', res)
304	n/a
305	n/a
306	n/a	def test_suite():
307	n/a	return unittest.makeSuite(DepGraphTestCase)
308	n/a
309	n/a	if __name__ == "__main__":
310	n/a	unittest.main(defaultTest="test_suite")