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

#	count	content
1	n/a	"""Test PyPI Server implementation at testpypi.python.org, to use in tests.
2	n/a
3	n/a	This is a drop-in replacement for the mock pypi server for testing against a
4	n/a	real pypi server hosted by python.org especially for testing against.
5	n/a	"""
6	n/a
7	n/a	import unittest
8	n/a
9	n/a	PYPI_DEFAULT_STATIC_PATH = None
10	n/a
11	n/a
12	n/a	def use_xmlrpc_server(server_args, *server_kwargs):
13	n/a	server_kwargs['serve_xmlrpc'] = True
14	n/a	return use_pypi_server(server_args, *server_kwargs)
15	n/a
16	n/a
17	n/a	def use_http_server(server_args, *server_kwargs):
18	n/a	server_kwargs['serve_xmlrpc'] = False
19	n/a	return use_pypi_server(server_args, *server_kwargs)
20	n/a
21	n/a
22	n/a	def use_pypi_server(server_args, *server_kwargs):
23	n/a	"""Decorator to make use of the PyPIServer for test methods,
24	n/a	just when needed, and not for the entire duration of the testcase.
25	n/a	"""
26	n/a	def wrapper(func):
27	n/a	def wrapped(args, *kwargs):
28	n/a	server = PyPIServer(server_args, *server_kwargs)
29	n/a	func(server=server, args, *kwargs)
30	n/a	return wrapped
31	n/a	return wrapper
32	n/a
33	n/a
34	n/a	class PyPIServerTestCase(unittest.TestCase):
35	n/a
36	n/a	def setUp(self):
37	n/a	super(PyPIServerTestCase, self).setUp()
38	n/a	self.pypi = PyPIServer()
39	n/a	self.pypi.start()
40	n/a	self.addCleanup(self.pypi.stop)
41	n/a
42	n/a
43	n/a	class PyPIServer:
44	n/a	"""Shim to access testpypi.python.org, for testing a real server."""
45	n/a
46	n/a	def __init__(self, test_static_path=None,
47	n/a	static_filesystem_paths=["default"],
48	n/a	static_uri_paths=["simple"], serve_xmlrpc=False):
49	n/a	self.address = ('testpypi.python.org', '80')
50	n/a
51	n/a	def start(self):
52	n/a	pass
53	n/a
54	n/a	def stop(self):
55	n/a	pass
56	n/a
57	n/a	@property
58	n/a	def full_address(self):
59	n/a	return "http://%s:%s" % self.address