|
|
"""Basic http server for tests to simulate PyPI or custom indexes""" |
|
|
|
|
|
import os |
|
|
import time |
|
|
import threading |
|
|
import http.server |
|
|
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" |
|
|
|
|
|
|
|
|
time.sleep(0.1) |
|
|
|
|
|
self.shutdown() |
|
|
self.thread.join() |
|
|
self.socket.close() |
|
|
|
|
|
def base_url(self): |
|
|
port = self.server_port |
|
|
return 'http://127.0.0.1:%s/setuptools/tests/indexes/' % port |
|
|
|
|
|
|
|
|
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 'localhost:%s' % self.server_port |
|
|
|
|
|
@property |
|
|
def url(self): |
|
|
return 'http://%s/' % 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)) |
|
|
|
