Better testing maybe

Author: thewhaleking

.github/workflows/unit-and-integration-test.yml

@@ -56,4 +56,4 @@ jobs:
           PYTHONUNBUFFERED: "1"
         run: |
           source venv/bin/activate
-          python -m uv run pytest -n 2 tests/integration_tests/ --reruns 3
+          python -m uv run pytest tests/integration_tests/ --reruns 3

tests/helpers/proxy_server.py

@@ -9,14 +9,15 @@
 
 
 class ProxyServer:
-    def __init__(self, upstream: str, time_til_pause: float, time_til_resume: float):
+    def __init__(self, upstream: str, time_til_pause: float, time_til_resume: float, port: int = 8080):
         self.upstream_server = upstream
         self.time_til_pause = time_til_pause
         self.time_til_resume = time_til_resume
         self.upstream_connection = None
         self.connection_time = 0
         self.shutdown_time = 0
         self.resume_time = 0
+        self.port = port
 
     def connect(self):
         self.upstream_connection = connect(self.upstream_server)
@@ -41,7 +42,7 @@ def proxy_request(self, websocket: ServerConnection):
             websocket.send(recd)
 
     def serve(self):
-        with serve(self.proxy_request, "localhost", 8080) as self.server:
+        with serve(self.proxy_request, "localhost", self.port) as self.server:
             self.server.serve_forever()
 
     def connect_and_serve(self):

tests/integration_tests/test_async_substrate_interface.py

@@ -210,14 +210,14 @@ async def test_improved_reconnection():
         os.remove(asi_logger_path)
     logger.setLevel(logging.DEBUG)
     logger.addHandler(logging.FileHandler(asi_logger_path))
-
-    proxy = ProxyServer("wss://archive.sub.latent.to", 10, 20)
+    port = 8079
+    proxy = ProxyServer("wss://archive.sub.latent.to", 10, 20, port=port)
 
     server_thread = threading.Thread(target=proxy.connect_and_serve, daemon=True)
     server_thread.start()
     await asyncio.sleep(3)  # give the server start up time
     async with AsyncSubstrateInterface(
-        "ws://localhost:8080",
+        f"ws://localhost:{port}",
         ss58_format=42,
         chain_name="Bittensor",
         retry_timeout=10.0,