55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试WebSocket连接的脚本
|
||
模拟前端连接
|
||
"""
|
||
import asyncio
|
||
import websockets
|
||
import json
|
||
import sys
|
||
import os
|
||
|
||
# 设置环境变量
|
||
os.environ.setdefault('SECRET_KEY', 'your-secret-key-change-in-production')
|
||
|
||
async def test_streamer_connection():
|
||
"""测试主播WebSocket连接"""
|
||
# 使用有效的token（从日志中复制的）
|
||
token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3NjYzOTcyMTEsInN1YiI6IjEwIn0.Miublr4D4W5_xS80eNb9mEMQlH-k9fh2Vc4Ndjf8GPQ"
|
||
streamer_id = 10
|
||
|
||
uri = f"ws://localhost:8000/socket.io/?role=streamer&id={streamer_id}&token={token}"
|
||
|
||
try:
|
||
print(f"尝试连接到: {uri}")
|
||
async with websockets.connect(uri) as websocket:
|
||
print("✅ WebSocket连接成功！")
|
||
|
||
# 等待消息
|
||
for i in range(10):
|
||
try:
|
||
response = await asyncio.wait_for(websocket.recv(), timeout=3.0)
|
||
print(f"📥 收到消息 {i+1}: {response}")
|
||
except asyncio.TimeoutError:
|
||
print(f"⏱️ 3秒内未收到消息 {i+1}（继续等待）")
|
||
continue
|
||
|
||
print("\n✅ 测试完成！")
|
||
return True
|
||
|
||
except websockets.exceptions.InvalidURI as e:
|
||
print(f"❌ 无效的URI: {e}")
|
||
return False
|
||
except websockets.exceptions.ConnectionRefused as e:
|
||
print(f"❌ 连接被拒绝: {e}")
|
||
return False
|
||
except Exception as e:
|
||
print(f"❌ 连接错误: {type(e).__name__}: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
if __name__ == "__main__":
|
||
result = asyncio.run(test_streamer_connection())
|
||
sys.exit(0 if result else 1)
|