diff options
Diffstat (limited to 'tests/utils.py')
-rw-r--r-- | tests/utils.py | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/tests/utils.py b/tests/utils.py new file mode 100644 index 0000000..32636b7 --- /dev/null +++ b/tests/utils.py @@ -0,0 +1,55 @@ +import contextlib +import http.server +import pathlib +import threading +from ssl import PROTOCOL_TLS_SERVER, SSLContext + +import filelock + +# Generated with: +# $ openssl req -new -x509 -days 3650 -nodes -out cert.pem \ +# -keyout cert.pem -addext "subjectAltName = DNS:localhost" +TESTS_ROOT = pathlib.Path(__file__).parent +CERT_FILE = str(TESTS_ROOT / "certs" / "cert.pem") + +# File lock for tests +LOCK_PATH = str(TESTS_ROOT / 'test-server.lock') + + +class HttpServerThread(threading.Thread): + def __init__(self, handler, *args, **kwargs): + super().__init__(*args, **kwargs) + self.server = http.server.ThreadingHTTPServer(("localhost", 7777), handler) + + def run(self): + self.server.serve_forever(poll_interval=0.001) + + def terminate(self): + self.server.shutdown() + self.server.server_close() + self.join() + + +class HttpsServerThread(HttpServerThread): + def __init__(self, handler, *args, **kwargs): + super().__init__(handler, *args, **kwargs) + sslcontext = SSLContext(PROTOCOL_TLS_SERVER) + sslcontext.load_cert_chain(CERT_FILE) + self.server.socket = sslcontext.wrap_socket(self.server.socket, server_side=True) + + +def create_server(thread_class): + def server(handler): + lock = filelock.FileLock(LOCK_PATH) + with lock: + server_thread = thread_class(handler, daemon=True) + server_thread.start() + try: + yield server_thread + finally: + server_thread.terminate() + return contextlib.contextmanager(server) + + +http_server = create_server(HttpServerThread) +https_server = create_server(HttpsServerThread) |