Fix flaky test: Increment request count before responding

[paperdoorknob] / testing / fakeserver.py

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:
        self._notify_server()
        with open("testdata/this-is-glowfic.html", "rb") as f:
            body = f.read(9999)
        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)


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