hard code port

Author: xzrderek

tests/pytest/test_mcp_session_autocreate.py

@@ -4,7 +4,7 @@
 """
 
 import time
-from multiprocessing import Process, Queue
+from multiprocessing import Process
 
 import httpx
 import pytest
@@ -13,15 +13,12 @@
 from eval_protocol.types import MCPSession
 
 
-def _run_airline_server(port_queue):
+def _run_airline_server():
     import os
 
-    # Use different ports based on Python version to avoid conflicts in parallel CI runs
     python_version = os.environ.get("PYTHON_VERSION", "3.10").replace(".", "")
-    port = str(9780 + int(python_version[-2:]))  # 9780, 9781, 9782
+    port = str(9780 + int(python_version[-1:]))
     os.environ["PORT"] = port
-
-    port_queue.put(int(port))  # Send the port back to the test
     from eval_protocol.mcp_servers.tau2.tau2_mcp import AirlineDomainMcp
 
     server = AirlineDomainMcp(seed=None)
@@ -30,26 +27,34 @@ def _run_airline_server(port_queue):
 
 @pytest.mark.asyncio
 async def test_tool_call_returns_json_without_prior_initial_state():
-    port_queue = Queue()
-    proc = Process(target=_run_airline_server, args=(port_queue,), daemon=True)
+    import os
+
+    proc = Process(target=_run_airline_server, daemon=True)
     proc.start()
 
     try:
-        # Get the dynamically assigned port
-        port = port_queue.get(timeout=10)
+        python_version = os.environ.get("PYTHON_VERSION", "3.10").replace(".", "")
+        port = str(9780 + int(python_version[-1:]))
+
         base_url = f"http://127.0.0.1:{port}/mcp"
         client = httpx.Client(timeout=1.0)
-        deadline = time.time() + 20
+        start_time = time.time()
+        deadline = start_time + 20
+        ready_time = None
         while time.time() < deadline:
             try:
                 r = client.get(base_url)
                 if r.status_code in (200, 307, 406):
+                    ready_time = time.time()
                     break
             except Exception:
                 pass
             time.sleep(0.2)
         else:
-            pytest.fail(f"Server did not start on port {port} in time")
+            pytest.fail("Server did not start on port 9780 in time")
+
+        assert ready_time is not None, "Server did not return a successful status before exiting loop"
+        assert ready_time - start_time < 20, f"Server took too long to respond: {ready_time - start_time:.2f}s"
 
         session = MCPSession(base_url=base_url, session_id="test-autocreate", seed=None, model_id="test-model")