root
4 months ago
7 changed files with 252 additions and 7 deletions
@ -0,0 +1,146 @@ |
|||
#!/usr/bin/env python3 |
|||
""" |
|||
WebSocket连接测试脚本 |
|||
用于测试WebSocket客户端的连接功能 |
|||
""" |
|||
import asyncio |
|||
import json |
|||
import sys |
|||
import os |
|||
|
|||
# 添加项目根目录到Python路径 |
|||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) |
|||
|
|||
from app.core.websocket.manager import websocket_manager |
|||
from app.schemas.websocket import CreateWebSocketClientRequest |
|||
from app.utils.log import get_logger |
|||
|
|||
logger = get_logger(__name__) |
|||
|
|||
async def test_websocket_connection(): |
|||
"""测试WebSocket连接功能""" |
|||
print("=== WebSocket连接测试 ===") |
|||
|
|||
# 测试配置 |
|||
test_cases = [ |
|||
{ |
|||
"name": "test_ws_echo", |
|||
"url": "ws://echo.websocket.org", |
|||
"description": "WebSocket Echo服务器测试" |
|||
}, |
|||
{ |
|||
"name": "test_ws_local", |
|||
"url": "ws://localhost:8080/ws", |
|||
"description": "本地WebSocket服务器测试" |
|||
}, |
|||
{ |
|||
"name": "test_wss_echo", |
|||
"url": "wss://echo.websocket.org", |
|||
"description": "WSS Echo服务器测试" |
|||
} |
|||
] |
|||
|
|||
for test_case in test_cases: |
|||
print(f"\n--- 测试: {test_case['description']} ---") |
|||
print(f"URL: {test_case['url']}") |
|||
|
|||
try: |
|||
# 创建请求对象 |
|||
request = CreateWebSocketClientRequest( |
|||
name=test_case["name"], |
|||
url=test_case["url"], |
|||
heartbeat_interval=120 # 2分钟心跳间隔 |
|||
) |
|||
|
|||
# 创建客户端 |
|||
print("1. 创建WebSocket客户端...") |
|||
client = await websocket_manager.create_client(request.name, request.url) |
|||
print(f" 客户端创建成功: {client.name}") |
|||
|
|||
# 连接客户端 |
|||
print("2. 连接WebSocket客户端...") |
|||
success = await websocket_manager.connect_client(request.name) |
|||
|
|||
if success: |
|||
print(f" ✅ 连接成功: {request.name}") |
|||
|
|||
# 获取客户端状态 |
|||
client = websocket_manager.get_client(request.name) |
|||
if client: |
|||
stats = client.get_stats() |
|||
print(f" 状态: {stats['state']}") |
|||
print(f" 已连接: {stats['is_connected']}") |
|||
|
|||
# 等待一段时间观察连接状态 |
|||
print("3. 等待5秒观察连接状态...") |
|||
await asyncio.sleep(5) |
|||
|
|||
# 再次检查状态 |
|||
client = websocket_manager.get_client(request.name) |
|||
if client: |
|||
stats = client.get_stats() |
|||
print(f" 5秒后状态: {stats['state']}") |
|||
print(f" 5秒后已连接: {stats['is_connected']}") |
|||
|
|||
# 断开连接 |
|||
print("4. 断开WebSocket客户端...") |
|||
await websocket_manager.disconnect_client(request.name) |
|||
print(f" ✅ 断开成功: {request.name}") |
|||
|
|||
else: |
|||
print(f" ❌ 连接失败: {request.name}") |
|||
|
|||
except Exception as e: |
|||
print(f" ❌ 测试异常: {e}") |
|||
logger.error(f"WebSocket连接测试异常: {e}") |
|||
|
|||
# 清理 |
|||
try: |
|||
await websocket_manager.remove_client(test_case["name"]) |
|||
except: |
|||
pass |
|||
|
|||
print("\n=== 测试完成 ===") |
|||
|
|||
async def test_websocket_manager_stats(): |
|||
"""测试WebSocket管理器统计信息""" |
|||
print("\n=== WebSocket管理器统计信息 ===") |
|||
|
|||
try: |
|||
stats = websocket_manager.get_stats() |
|||
print(f"客户端数量: {stats['client_count']}") |
|||
print(f"Channel数量: {stats['channel_count']}") |
|||
print(f"适配器数量: {stats['adapter_count']}") |
|||
print(f"创建时间: {stats['created_at']}") |
|||
|
|||
if stats['clients']: |
|||
print("\n客户端详情:") |
|||
for name, client_stats in stats['clients'].items(): |
|||
print(f" - {name}: {client_stats['state']} (连接: {client_stats['is_connected']})") |
|||
|
|||
except Exception as e: |
|||
print(f"获取统计信息失败: {e}") |
|||
|
|||
async def main(): |
|||
"""主函数""" |
|||
try: |
|||
# 测试WebSocket连接 |
|||
await test_websocket_connection() |
|||
|
|||
# 测试管理器统计 |
|||
await test_websocket_manager_stats() |
|||
|
|||
except KeyboardInterrupt: |
|||
print("\n测试被用户中断") |
|||
except Exception as e: |
|||
print(f"测试过程中发生异常: {e}") |
|||
logger.error(f"测试异常: {e}") |
|||
finally: |
|||
# 清理资源 |
|||
print("\n清理WebSocket管理器资源...") |
|||
await websocket_manager.cleanup() |
|||
print("清理完成") |
|||
|
|||
if __name__ == "__main__": |
|||
# 运行测试 |
|||
asyncio.run(main()) |
|||
Loading…
Reference in new issue