mirror of https://github.com/python/cpython
88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
"""Tests for packaging.command.bdist."""
|
|
import sys
|
|
|
|
import urllib.request
|
|
import urllib.parse
|
|
import urllib.error
|
|
|
|
try:
|
|
import threading
|
|
from packaging.tests.pypi_server import PyPIServer, PYPI_DEFAULT_STATIC_PATH
|
|
except ImportError:
|
|
threading = None
|
|
PyPIServer = None
|
|
PYPI_DEFAULT_STATIC_PATH = None
|
|
|
|
from packaging.tests import unittest
|
|
|
|
|
|
@unittest.skipIf(threading is None, "Needs threading")
|
|
class PyPIServerTest(unittest.TestCase):
|
|
|
|
def test_records_requests(self):
|
|
# We expect that PyPIServer can log our requests
|
|
server = PyPIServer()
|
|
server.default_response_status = 200
|
|
|
|
try:
|
|
server.start()
|
|
self.assertEqual(len(server.requests), 0)
|
|
|
|
data = b'Rock Around The Bunker'
|
|
|
|
headers = {"X-test-header": "Mister Iceberg"}
|
|
|
|
request = urllib.request.Request(server.full_address, data, headers)
|
|
urllib.request.urlopen(request)
|
|
self.assertEqual(len(server.requests), 1)
|
|
handler, request_data = server.requests[-1]
|
|
self.assertIn(data, request_data)
|
|
self.assertIn("x-test-header", handler.headers)
|
|
self.assertEqual(handler.headers["x-test-header"], "Mister Iceberg")
|
|
|
|
finally:
|
|
server.stop()
|
|
|
|
|
|
def test_serve_static_content(self):
|
|
# PYPI Mocked server can serve static content from disk.
|
|
|
|
def uses_local_files_for(server, url_path):
|
|
"""Test that files are served statically (eg. the output from the
|
|
server is the same than the one made by a simple file read.
|
|
"""
|
|
url = server.full_address + url_path
|
|
request = urllib.request.Request(url)
|
|
response = urllib.request.urlopen(request)
|
|
with open(PYPI_DEFAULT_STATIC_PATH + "/test_pypi_server"
|
|
+ url_path) as file:
|
|
return response.read().decode() == file.read()
|
|
|
|
server = PyPIServer(static_uri_paths=["simple", "external"],
|
|
static_filesystem_paths=["test_pypi_server"])
|
|
server.start()
|
|
try:
|
|
# the file does not exists on the disc, so it might not be served
|
|
url = server.full_address + "/simple/unexisting_page"
|
|
request = urllib.request.Request(url)
|
|
try:
|
|
urllib.request.urlopen(request)
|
|
except urllib.error.HTTPError as e:
|
|
self.assertEqual(e.code, 404)
|
|
|
|
# now try serving a content that do exists
|
|
self.assertTrue(uses_local_files_for(server, "/simple/index.html"))
|
|
|
|
# and another one in another root path
|
|
self.assertTrue(uses_local_files_for(server, "/external/index.html"))
|
|
|
|
finally:
|
|
server.stop()
|
|
|
|
|
|
def test_suite():
|
|
return unittest.makeSuite(PyPIServerTest)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest="test_suite")
|