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

#	count	content
1	n/a	"""Tests for packaging.command.bdist."""
2	n/a	import urllib.request
3	n/a	import urllib.parse
4	n/a	import urllib.error
5	n/a
6	n/a	try:
7	n/a	import threading
8	n/a	from packaging.tests.pypi_server import (
9	n/a	PyPIServer, PYPI_DEFAULT_STATIC_PATH)
10	n/a	except ImportError:
11	n/a	threading = None
12	n/a	PyPIServer = None
13	n/a	PYPI_DEFAULT_STATIC_PATH = None
14	n/a
15	n/a	from packaging.tests import unittest
16	n/a
17	n/a
18	n/a	@unittest.skipIf(threading is None, "Needs threading")
19	n/a	class PyPIServerTest(unittest.TestCase):
20	n/a
21	n/a	def test_records_requests(self):
22	n/a	# We expect that PyPIServer can log our requests
23	n/a	server = PyPIServer()
24	n/a	server.default_response_status = 200
25	n/a
26	n/a	try:
27	n/a	server.start()
28	n/a	self.assertEqual(len(server.requests), 0)
29	n/a
30	n/a	data = b'Rock Around The Bunker'
31	n/a
32	n/a	headers = {"X-test-header": "Mister Iceberg"}
33	n/a
34	n/a	request = urllib.request.Request(
35	n/a	server.full_address, data, headers)
36	n/a	urllib.request.urlopen(request)
37	n/a	self.assertEqual(len(server.requests), 1)
38	n/a	handler, request_data = server.requests[-1]
39	n/a	self.assertIn(data, request_data)
40	n/a	self.assertIn("x-test-header", handler.headers)
41	n/a	self.assertEqual(handler.headers["x-test-header"],
42	n/a	"Mister Iceberg")
43	n/a
44	n/a	finally:
45	n/a	server.stop()
46	n/a
47	n/a	def test_serve_static_content(self):
48	n/a	# PYPI Mocked server can serve static content from disk.
49	n/a
50	n/a	def uses_local_files_for(server, url_path):
51	n/a	"""Test that files are served statically (eg. the output from the
52	n/a	server is the same than the one made by a simple file read.
53	n/a	"""
54	n/a	url = server.full_address + url_path
55	n/a	request = urllib.request.Request(url)
56	n/a	response = urllib.request.urlopen(request)
57	n/a	with open(PYPI_DEFAULT_STATIC_PATH + "/test_pypi_server"
58	n/a	+ url_path) as file:
59	n/a	return response.read().decode() == file.read()
60	n/a
61	n/a	server = PyPIServer(static_uri_paths=["simple", "external"],
62	n/a	static_filesystem_paths=["test_pypi_server"])
63	n/a	server.start()
64	n/a	try:
65	n/a	# the file does not exists on the disc, so it might not be served
66	n/a	url = server.full_address + "/simple/unexisting_page"
67	n/a	request = urllib.request.Request(url)
68	n/a	try:
69	n/a	urllib.request.urlopen(request)
70	n/a	except urllib.error.HTTPError as e:
71	n/a	self.assertEqual(e.code, 404)
72	n/a
73	n/a	# now try serving a content that do exists
74	n/a	self.assertTrue(uses_local_files_for(server, "/simple/index.html"))
75	n/a
76	n/a	# and another one in another root path
77	n/a	self.assertTrue(uses_local_files_for(server,
78	n/a	"/external/index.html"))
79	n/a
80	n/a	finally:
81	n/a	server.stop()
82	n/a
83	n/a
84	n/a	def test_suite():
85	n/a	return unittest.makeSuite(PyPIServerTest)
86	n/a
87	n/a	if __name__ == '__main__':
88	n/a	unittest.main(defaultTest="test_suite")