Untitled

"""
Manager WebSocket test.
Run with: python tests/ai/ai_review/test_manager_websocket.py

Tests manager-employee review WebSocket connection.
Endpoint: /manager-realtime
Params:
  - employee_id: the employee being reviewed
  - manager_id: the manager conducting the review
  - session_id: from the employee's voice_conversations row
  - token: manager's JWT token
"""
import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../..")))

import asyncio
import json
import websockets


# ── PASTE YOUR VALUES HERE ────────────────────────────────────────────────────
EMPLOYEE_ID = "paste-employee-id-here"       # employee being reviewed
MANAGER_ID  = "paste-manager-id-here"        # manager conducting the review
SESSION_ID  = "paste-session-id-here"        # from employee's voice_conversations row
TOKEN       = "paste-manager-jwt-token-here" # manager's auth token

# Built from above — no need to edit
FULL_URL = (
    f"ws://localhost:8000/ws/v1/manager-realtime"
    f"?employee_id={EMPLOYEE_ID}"
    f"&manager_id={MANAGER_ID}"
    f"&session_id={SESSION_ID}"
    f"&token={TOKEN}"
)
# ─────────────────────────────────────────────────────────────────────────────


async def test_manager_websocket():
    print(f"Connecting to: {FULL_URL}")
    print(f"  employee_id : {EMPLOYEE_ID}")
    print(f"  manager_id  : {MANAGER_ID}")
    print(f"  session_id  : {SESSION_ID}")
    print("-" * 60)

    try:
        async with websockets.connect(FULL_URL) as ws:
            print("✅ WebSocket connected")

            async def listen():
                async for message in ws:
                    if isinstance(message, bytes):
                        print(f"📦 Binary audio: {len(message)} bytes")
                    else:
                        try:
                            data = json.loads(message)
                            msg_type = data.get("type", "unknown")

                            if msg_type == "status":
                                print(f"✅ Status: {data.get('message')}")
                            elif msg_type == "error":
                                print(f"❌ Error: {data.get('message')}")
                            elif msg_type == "session.created":
                                print(f"✅ Session created with OpenAI")
                            elif msg_type == "session.updated":
                                session = data.get("session", {})
                                instructions = session.get("instructions", "") or ""
                                tools = session.get("tools", []) or []
                                print(f"✅ Session updated:")
                                print(f"   instructions length : {len(instructions)} chars")
                                print(f"   tools registered   : {[t.get('name') for t in tools]}")
                            elif msg_type == "review_progress":
                                print(f"📊 Progress: {data.get('percentage')}%")
                            elif msg_type == "heartbeat":
                                print(f"💓 Heartbeat")
                            else:
                                print(f"📨 {msg_type}")

                        except json.JSONDecodeError:
                            print(f"📨 Raw: {message[:100]}")

            try:
                await asyncio.wait_for(listen(), timeout=15.0)
            except asyncio.TimeoutError:
                print("\n⏱ 15 second test complete")

    except Exception as e:
        print(f"❌ Connection failed: {e}")

    print("\n── Expected ────────────────────────────────────────────")
    print("✅ WebSocket connected")
    print("✅ Status: Connected to proxy server")
    print("✅ Status: Connected to OpenAI")
    print("✅ Session created with OpenAI")
    print("✅ Session updated: instructions + tools")
    print("📨 Audio events (Archie speaks to manager)")


if __name__ == "__main__":
    asyncio.run(test_manager_websocket())