X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/07f9b178a58e87f90aa7ab1e459c17561345154d..41b11505f297962d099b75a0372a4068e26773a5:/testing/fakeserver.py diff --git a/testing/fakeserver.py b/testing/fakeserver.py new file mode 100644 index 0000000..2ed8149 --- /dev/null +++ b/testing/fakeserver.py @@ -0,0 +1,72 @@ +from typing import Any, Self +import threading +from http.server import BaseHTTPRequestHandler, HTTPServer + + +class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler): + + def _notify_server(self) -> None: + raise NotImplementedError() + + def _response_code(self) -> int: + if self.path == "/not_found": + return 404 + if self.path == "/server_error": + return 500 + return 200 + + def do_GET(self) -> None: + body = b''' + +
+
We don't want edit boxes
+ This is glowfic +
We don't want footers
+
+
+
+
We don't want edit boxes
+ You sure? +
We don't want footers
+
+
+ Pretty sure. +
+
+ +''' + self.send_response(self._response_code()) + self.send_header("Content-type", "text/html") + self.send_header("Content-Length", str(len(body))) + self.end_headers() + self.wfile.write(body) + self._notify_server() + + +class FakeGlowficServer(): + def __init__(self) -> None: + self._request_counter = 0 + handler = type("Handler", (FakeGlowficHTTPRequestHandler,), { + '_notify_server': lambda _: self._count_request()}) + self._web_server = HTTPServer(('', 0), handler) + self._thread = threading.Thread(target=self._web_server.serve_forever) + + def __enter__(self) -> Self: + self._thread.start() + return self + + def __exit__(self, *_: Any) -> None: + self._web_server.shutdown() + self._thread.join() + self._web_server.server_close() + + def _count_request(self) -> None: + self._request_counter += 1 + + def port(self) -> int: + p = self._web_server.socket.getsockname()[1] + assert isinstance(p, int) + return p + + def request_count(self) -> int: + return self._request_counter