cpython/Lib/packaging/tests/test_pypi_server.py

"""Tests for packaging.command.bdist."""
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")