You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
134 lines
4.8 KiB
134 lines
4.8 KiB
#!/usr/bin/env python3
|
|
"""
|
|
简单测试WebSocket消息格式组装功能
|
|
"""
|
|
import asyncio
|
|
import sys
|
|
import os
|
|
|
|
# 添加项目根目录到Python路径
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from app.core.websocket.manager import WebSocketManager
|
|
from app.core.device.event_types import EventType
|
|
|
|
|
|
async def test_message_format_assembly():
|
|
"""测试消息格式组装功能"""
|
|
print("=== 简单测试WebSocket消息格式组装功能 ===")
|
|
|
|
try:
|
|
# 1. 创建WebSocket管理器
|
|
websocket_manager = WebSocketManager()
|
|
|
|
# 2. 创建测试客户端和Channel（不连接WebSocket）
|
|
await websocket_manager.create_client("test_client", "ws://localhost:8080")
|
|
|
|
# 手动启动Channel（不连接WebSocket客户端）
|
|
await websocket_manager._channel_manager.start_client_channels("test_client")
|
|
|
|
print("1. WebSocket客户端和Channel创建启动成功")
|
|
|
|
# 3. 测试发送消息到send_channel
|
|
test_data = {
|
|
"type": EventType.DEVICE_STATUS_UPDATE.value,
|
|
"device_id": "AMFU6R1813008221",
|
|
"status": "device",
|
|
"device_info": {
|
|
"brand": "HONOR",
|
|
"model": "JLH-AN00"
|
|
},
|
|
"event_type": EventType.DEVICE_CONNECTED.value
|
|
}
|
|
|
|
success = await websocket_manager.send_message(
|
|
"test_client",
|
|
"terminal",
|
|
test_data,
|
|
priority=1
|
|
)
|
|
|
|
print(f"2. 发送消息到send_channel: {'成功' if success else '失败'}")
|
|
|
|
# 4. 验证消息格式
|
|
send_channel = websocket_manager.get_channel("test_client_send")
|
|
if send_channel and send_channel.total_queue_size > 0:
|
|
message = await send_channel.receive_message(timeout=1.0)
|
|
if message:
|
|
print(f"3. 接收到的消息类型: {message.type}")
|
|
print(f" 消息数据: {message.data}")
|
|
|
|
# 检查是否包含Type字段
|
|
if "Type" in message.data:
|
|
print(f" ✅ 消息格式组装成功，包含Type字段: {message.data['Type']}")
|
|
else:
|
|
print(f" ❌ 消息格式组装失败，缺少Type字段")
|
|
|
|
# 检查原始数据是否保留
|
|
if "type" in message.data and "device_id" in message.data:
|
|
print(f" ✅ 原始数据保留完整")
|
|
else:
|
|
print(f" ❌ 原始数据丢失")
|
|
else:
|
|
print("3. ❌ 无法接收消息")
|
|
else:
|
|
print("3. ❌ send_channel中没有消息")
|
|
|
|
# 5. 测试发送消息到其他Channel（不应该组装格式）
|
|
heartbeat_channel = websocket_manager.get_channel("test_client_heartbeat")
|
|
if heartbeat_channel:
|
|
# 直接通过channel_manager发送，绕过manager的组装逻辑
|
|
success = await websocket_manager._channel_manager.send_message_to_channel(
|
|
"test_client_heartbeat",
|
|
"heartbeat",
|
|
{"message": "ping"},
|
|
priority=1
|
|
)
|
|
|
|
print(f"4. 发送消息到heartbeat_channel: {'成功' if success else '失败'}")
|
|
|
|
if heartbeat_channel.total_queue_size > 0:
|
|
message = await heartbeat_channel.receive_message(timeout=1.0)
|
|
if message:
|
|
print(f"5. heartbeat_channel消息类型: {message.type}")
|
|
print(f" 消息数据: {message.data}")
|
|
|
|
# 检查是否不包含Type字段（不应该组装）
|
|
if "Type" not in message.data:
|
|
print(f" ✅ heartbeat_channel消息格式正确，不包含Type字段")
|
|
else:
|
|
print(f" ❌ heartbeat_channel消息格式错误，不应该包含Type字段")
|
|
else:
|
|
print("5. ❌ 无法接收heartbeat_channel消息")
|
|
else:
|
|
print("5. ❌ heartbeat_channel中没有消息")
|
|
|
|
# 6. 清理资源
|
|
await websocket_manager.cleanup()
|
|
print("6. 资源清理完成")
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 测试异常: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
|
|
async def main():
|
|
"""主函数"""
|
|
print("开始简单测试WebSocket消息格式组装功能...")
|
|
|
|
success = await test_message_format_assembly()
|
|
|
|
if success:
|
|
print("\n✅ 所有测试通过！")
|
|
else:
|
|
print("\n❌ 测试失败！")
|
|
|
|
return success
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|
|
|