debug_websocket.py

#!/usr/bin/env python3
import asyncio
import websockets
import json
import logging
from utils import setup_logging

setup_logging(logging.INFO)
logger = logging.getLogger("webseek_backend.debug_websocket")

async def test_websocket():
    try:
        logger.info("🔌 Attempting to connect to ws://localhost:8000/ws")
        async with websockets.connect('ws://localhost:8000/ws') as websocket:
            logger.info("✅ Connected successfully!")
            
            ping_message = {"type": "ping"}
            await websocket.send(json.dumps(ping_message))
            logger.info("📤 Sent ping")
            
            try:
                response = await asyncio.wait_for(websocket.recv(), timeout=5.0)
                logger.info(f"📥 Received: {response}")
            except asyncio.TimeoutError:
                logger.warning("⏰ Timeout waiting for response")
            except Exception as e:
                logger.error(f"📥 Failed to receive message: {e}")
            
            await asyncio.sleep(1)
            
    except Exception as e:
        logger.error(f"❌ Connection failed: {e}")
        logger.error(f"Error type: {type(e)}")

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