comfyui/test_connection.py
2025-06-08 17:44:58 +08:00

31 lines
804 B
Python

from websocket import create_connection, WebSocketTimeoutException, WebSocketConnectionClosedException
print("=== WebSocket客户端测试 ===")
try:
# 测试连接到公共的 WebSocket 测试服务
ws = create_connection("ws://echo.websocket.org/", timeout=10)
print("连接成功!")
# 发送测试消息
message = "Hello, WebSocket!"
ws.send(message)
print(f"发送消息: {message}")
# 接收回应
result = ws.recv()
print(f"接收回应: {result}")
# 关闭连接
ws.close()
print("连接已正常关闭")
except WebSocketTimeoutException:
print("连接超时")
except WebSocketConnectionClosedException:
print("连接意外关闭")
except Exception as e:
print(f"其他错误: {str(e)}")
print("测试完成")