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("测试完成")