test this

Author: xzrderek

tests/pytest/test_mcp_session_autocreate.py

@@ -3,8 +3,10 @@
 without requiring a prior initial state fetch, and returns JSON.
 """
 
+import asyncio
+import subprocess
+import sys
 import time
-from multiprocessing import Process
 
 import httpx
 import pytest
@@ -13,43 +15,46 @@
 from eval_protocol.types import MCPSession
 
 
-def _run_airline_server():
-    import os
-    import sys
-
+@pytest.mark.asyncio
+async def test_tool_call_returns_json_without_prior_initial_state():
     # Get Python version directly from sys.version_info
     minor_version = sys.version_info.minor  # 10, 11, 12
 
     # Map Python versions to port offsets: 3.10->0, 3.11->1, 3.12->2
     port_offset = minor_version - 10
-    port = str(9780)
-    print(f"[SERVER DEBUG] Python 3.{minor_version} -> Setting PORT={port}")
-    os.environ["PORT"] = port
-    from eval_protocol.mcp_servers.tau2.tau2_mcp import AirlineDomainMcp
-
-    print(f"[SERVER DEBUG] About to create AirlineDomainMcp with PORT={os.environ.get('PORT')}")
-    server = AirlineDomainMcp(seed=None)
-    print(f"[SERVER DEBUG] Server created, FastMCP port={server.mcp.settings.port}")
-    print(f"[SERVER DEBUG] About to run on port {port}")
-    server.run(transport="streamable-http")
-
+    port = str(9780 + port_offset)
+    print(f"[TEST DEBUG] Python 3.{minor_version} -> Looking for server on port {port}")
+
+    # Create server script to run as subprocess instead of multiprocessing
+    server_script = """
+import sys
+import os
+
+# Get Python version directly from sys.version_info
+minor_version = sys.version_info.minor  # 10, 11, 12
+
+# Map Python versions to port offsets: 3.10->0, 3.11->1, 3.12->2
+port_offset = minor_version - 10
+port = str(9780 + port_offset)
+print(f"[SERVER DEBUG] Python 3.{minor_version} -> Setting PORT={port}")
+os.environ["PORT"] = port
+
+from eval_protocol.mcp_servers.tau2.tau2_mcp import AirlineDomainMcp
+
+print(f"[SERVER DEBUG] About to create AirlineDomainMcp with PORT={os.environ.get('PORT')}")
+server = AirlineDomainMcp(seed=None)
+print(f"[SERVER DEBUG] Server created, FastMCP port={server.mcp.settings.port}")
+print(f"[SERVER DEBUG] About to run on port {port}")
+server.run(transport="streamable-http")
+"""
 
-@pytest.mark.asyncio
-async def test_tool_call_returns_json_without_prior_initial_state():
-    import sys
+    # Start server as subprocess instead of multiprocessing.Process
+    proc = subprocess.Popen([sys.executable, "-c", server_script])
 
-    proc = Process(target=_run_airline_server, daemon=True)
-    proc.start()
+    # Give server time to start
+    await asyncio.sleep(3)
 
     try:
-        # Get Python version directly from sys.version_info
-        minor_version = sys.version_info.minor  # 10, 11, 12
-
-        # Map Python versions to port offsets: 3.10->0, 3.11->1, 3.12->2
-        port_offset = minor_version - 10
-        port = str(9780)
-        print(f"[TEST DEBUG] Python 3.{minor_version} -> Looking for server on port {port}")
-
         base_url = f"http://127.0.0.1:{port}/mcp"
         print(f"[TEST DEBUG] base_url = {base_url}")
         client = httpx.Client(timeout=1.0)
@@ -66,7 +71,7 @@ async def test_tool_call_returns_json_without_prior_initial_state():
                 pass
             time.sleep(0.2)
         else:
-            pytest.fail("Server did not start on port 9780 in time")
+            pytest.fail(f"Server did not start on port {port} 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"
@@ -86,4 +91,4 @@ async def test_tool_call_returns_json_without_prior_initial_state():
         await mgr.close_session(session)
     finally:
         proc.terminate()
-        proc.join(timeout=5)
+        proc.wait(timeout=5)