Viewing file:  server.py (2.34 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
"""Basic http server for tests to simulate PyPI or custom indexes"""
import http.server
import os
import threading
import time
import urllib.parse
import urllib.request
class IndexServer(http.server.HTTPServer):
"""Basic single-threaded http server simulating a package index
You can use this server in unittest like this::
s = IndexServer()
s.start()
index_url = s.base_url() + 'mytestindex'
# do some test requests to the index
# The index files should be located in setuptools/tests/indexes
s.stop()
"""
def __init__(
self,
server_address=('', 0),
RequestHandlerClass=http.server.SimpleHTTPRequestHandler,
):
http.server.HTTPServer.__init__(self, server_address, RequestHandlerClass)
self._run = True
def start(self):
self.thread = threading.Thread(target=self.serve_forever)
self.thread.start()
def stop(self):
"Stop the server"
# Let the server finish the last request and wait for a new one.
time.sleep(0.1)
self.shutdown()
self.thread.join()
self.socket.close()
def base_url(self):
port = self.server_port
return f'http://127.0.0.1:{port}/setuptools/tests/indexes/'
class RequestRecorder(http.server.BaseHTTPRequestHandler):
def do_GET(self):
requests = vars(self.server).setdefault('requests', [])
requests.append(self)
self.send_response(200, 'OK')
class MockServer(http.server.HTTPServer, threading.Thread):
"""
A simple HTTP Server that records the requests made to it.
"""
def __init__(self, server_address=('', 0), RequestHandlerClass=RequestRecorder):
http.server.HTTPServer.__init__(self, server_address, RequestHandlerClass)
threading.Thread.__init__(self)
self.daemon = True
self.requests = []
def run(self):
self.serve_forever()
@property
def netloc(self):
return f'localhost:{self.server_port}'
@property
def url(self):
return f'http://{self.netloc}/'
def path_to_url(path, authority=None):
"""Convert a path to a file: URL."""
path = os.path.normpath(os.path.abspath(path))
base = 'file:'
if authority is not None:
base += '//' + authority
return urllib.parse.urljoin(base, urllib.request.pathname2url(path))
|