Introduce server fixture to run app in subprocess for Socket.IO integration tests and update sync scenarios to use it.

Author: owaisjunedi

server/app/__pycache__/database.cpython-312.pyc

@@ -3,7 +3,7 @@
 from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
 from sqlalchemy.orm import sessionmaker
 from app.database import Base, get_db
-from app.main import fastapi_app as app
+from app.main import app, fastapi_app
 import asyncio
 from fastapi.testclient import TestClient
 
@@ -47,10 +47,79 @@ def event_loop():
 def override_dependency(test_db):
     async def _get_db_override():
         yield test_db
-    app.dependency_overrides[get_db] = _get_db_override
+    fastapi_app.dependency_overrides[get_db] = _get_db_override
+    
+    # Also override SessionLocal in main.py for Socket.IO handlers
+    import app.main
+    app.main.SessionLocal = TestingSessionLocal
 
 @pytest.fixture
 def client():
     with TestClient(app) as c:
         yield c
 
+@pytest_asyncio.fixture(scope="session")
+async def server():
+    """Starts a real uvicorn server for Socket.IO tests using subprocess"""
+    import subprocess
+    import time
+    import socket
+    import os
+    from httpx import AsyncClient
+
+    # Find a free port
+    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    sock.bind(('127.0.0.1', 0))
+    port = sock.getsockname()[1]
+    sock.close()
+
+    # Use the same database for the subprocess
+    # Since we are using in-memory SQLite, this won't work across processes!
+    # We should use a temporary file-based database for these tests.
+    test_db_file = "test_sync.db"
+    if os.path.exists(test_db_file):
+        os.remove(test_db_file)
+    
+    db_url = f"sqlite+aiosqlite:///./{test_db_file}"
+    
+    cmd = [
+        "uv", "run", "uvicorn", "app.main:app",
+        "--host", "127.0.0.1",
+        "--port", str(port),
+        "--log-level", "error"
+    ]
+    
+    env = os.environ.copy()
+    env["PYTHONPATH"] = "."
+    env["DATABASE_URL"] = db_url
+    
+    process = subprocess.Popen(cmd, env=env, cwd=os.getcwd())
+    
+    base_url = f"http://127.0.0.1:{port}"
+    
+    # Wait for server to be ready
+    max_retries = 50
+    for _ in range(max_retries):
+        try:
+            async with AsyncClient() as client:
+                res = await client.get(f"{base_url}/health")
+                if res.status_code == 200:
+                    break
+        except Exception:
+            pass
+        await asyncio.sleep(0.2)
+    
+    yield base_url
+    
+    process.terminate()
+    try:
+        process.wait(timeout=5)
+    except subprocess.TimeoutExpired:
+        process.kill()
+    
+    if os.path.exists(test_db_file):
+        try:
+            os.remove(test_db_file)
+        except:
+            pass
+

server/tests/__pycache__/conftest.cpython-312-pytest-9.0.2.pyc

@@ -5,12 +5,10 @@
 from httpx import AsyncClient, ASGITransport
 from app.main import app
 
-BASE_URL = "http://localhost:8000"
-
 @pytest.mark.asyncio
-async def test_sync_and_persistence():
+async def test_sync_and_persistence(server):
     # 1. Create Session
-    async with AsyncClient(base_url=BASE_URL) as ac:
+    async with AsyncClient(base_url=server) as ac:
         res = await ac.post("/sessions", json={
             "candidateName": "Test Candidate",
             "candidateEmail": "test@example.com",
@@ -31,7 +29,7 @@ async def on_session_updated(data):
         if not future_session_update.done():
             future_session_update.set_result(data)
 
-    await sio_a.connect(BASE_URL, socketio_path='/socket.io')
+    await sio_a.connect(server, socketio_path='/socket.io')
     await sio_a.emit('join_room', {
         'roomId': session_id,
         'user': {'id': 'user_a', 'name': 'Interviewer', 'role': 'interviewer'}
@@ -67,7 +65,7 @@ async def on_code_change(data):
         if not future_code_sync.done():
             future_code_sync.set_result(data)
 
-    await sio_b.connect(BASE_URL, socketio_path='/socket.io')
+    await sio_b.connect(server, socketio_path='/socket.io')
     await sio_b.emit('join_room', {
         'roomId': session_id,
         'user': {'id': 'user_b', 'name': 'Candidate', 'role': 'candidate'}
@@ -80,9 +78,9 @@ async def on_code_change(data):
     await sio_b.disconnect()
 
 @pytest.mark.asyncio
-async def test_user_presence():
+async def test_user_presence(server):
     # 1. Create Session
-    async with AsyncClient(base_url=BASE_URL) as ac:
+    async with AsyncClient(base_url=server) as ac:
         res = await ac.post("/sessions", json={
             "candidateName": "Test Candidate",
             "candidateEmail": "test@example.com",
@@ -102,14 +100,14 @@ async def on_room_users_b(data):
             future_users_b.set_result(data)
 
     # A joins
-    await sio_a.connect(BASE_URL, socketio_path='/socket.io')
+    await sio_a.connect(server, socketio_path='/socket.io')
     await sio_a.emit('join_room', {
         'roomId': session_id,
         'user': {'id': 'user_a', 'name': 'Interviewer', 'role': 'interviewer'}
     })
 
     # B joins
-    await sio_b.connect(BASE_URL, socketio_path='/socket.io')
+    await sio_b.connect(server, socketio_path='/socket.io')
     await sio_b.emit('join_room', {
         'roomId': session_id,
         'user': {'id': 'user_b', 'name': 'Candidate', 'role': 'candidate'}
@@ -123,7 +121,7 @@ async def on_room_users_b(data):
     future_users_b_leave = asyncio.Future()
     @sio_b.on('room_users')
     async def on_room_users_b_leave(data):
-        if not future_users_b_leave.done():
+        if len(data['users']) == 1 and not future_users_b_leave.done():
             future_users_b_leave.set_result(data)
 
     await sio_a.emit('leave_room', {'roomId': session_id, 'userId': 'user_a'})
@@ -137,9 +135,9 @@ async def on_room_users_b_leave(data):
     await sio_b.disconnect()
 
 @pytest.mark.asyncio
-async def test_whiteboard_sync():
+async def test_whiteboard_sync(server):
     # 1. Create Session
-    async with AsyncClient(base_url=BASE_URL) as ac:
+    async with AsyncClient(base_url=server) as ac:
         res = await ac.post("/sessions", json={
             "candidateName": "Test Candidate",
             "candidateEmail": "test@example.com",
@@ -159,13 +157,13 @@ async def on_whiteboard_update(data):
             future_wb_update.set_result(data)
 
     # Connect both
-    await sio_a.connect(BASE_URL, socketio_path='/socket.io')
+    await sio_a.connect(server, socketio_path='/socket.io')
     await sio_a.emit('join_room', {
         'roomId': session_id,
         'user': {'id': 'user_a', 'name': 'Interviewer', 'role': 'interviewer'}
     })
 
-    await sio_b.connect(BASE_URL, socketio_path='/socket.io')
+    await sio_b.connect(server, socketio_path='/socket.io')
     await sio_b.emit('join_room', {
         'roomId': session_id,
         'user': {'id': 'user_b', 'name': 'Candidate', 'role': 'candidate'}