Rewrite test_inspect_main_url to avoid mocking

Makes the test more realistic by issuing an HTTP request.
Reduces coupling between test and the code under test.

The `http_server` helper was factored out into a new tests.utils module.

1 files changed, 27 insertions, 0 deletions

diff --git a/tests/utils.py b/tests/utils.py
new file mode 100644
index 000000000..182dc1df0
--- /dev/null
+++ b/tests/utils.py
@@ -0,0 +1,27 @@
+import contextlib
+import http.server
+import threading
+
+
+class HttpServerThread(threading.Thread):
+    def __init__(self, handler, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.server = http.server.HTTPServer(("localhost", 7777), handler)
+
+    def run(self):
+        self.server.serve_forever(poll_interval=0.01)
+
+    def terminate(self):
+        self.server.shutdown()
+        self.server.server_close()
+        self.join()
+
+
+@contextlib.contextmanager
+def http_server(handler):
+    server_thread = HttpServerThread(handler, daemon=True)
+    server_thread.start()
+    try:
+        yield server_thread
+    finally:
+        server_thread.terminate()