python测试websocket接口代码

下面是一个使用Python测试WebSocket接口的示例代码：

import websocket
import json

# 定义WebSocket接收到消息时的处理函数
def on_message(ws, message):
    print("Received message: " + message)

# 定义WebSocket连接建立时的处理函数
def on_open(ws):
    print("WebSocket connection established")
    # 向WebSocket服务器发送消息
    message = {"type": "test", "data": "hello"}
    ws.send(json.dumps(message))

# 定义WebSocket连接关闭时的处理函数
def on_close(ws):
    print("WebSocket connection closed")

# 定义WebSocket连接错误时的处理函数
def on_error(ws, error):
    print("WebSocket error: " + str(error))

# 创建WebSocket连接
ws = websocket.WebSocketApp("ws://localhost:8080/ws",
                            on_message = on_message,
                            on_open = on_open,
                            on_close = on_close,
                            on_error = on_error)

# 开始WebSocket连接
ws.run_forever()

在上面的代码中，我们使用了Python的`websocket`库，定义了四个回调函数：

- `on_message`: 当WebSocket接收到消息时调用的函数。
- `on_open`: 当WebSocket连接建立时调用的函数。
- `on_close`: 当WebSocket连接关闭时调用的函数。
- `on_error`: 当WebSocket连接出现错误时调用的函数。

然后我们创建了一个WebSocket连接对象，指定了上面定义的四个回调函数，并调用其`run_forever`方法开始连接。在`on_open`函数中，我们向WebSocket服务器发送了一条消息。

请注意，上面的代码仅供参考，具体的WebSocket接口测试代码需要根据实际情况进行调整。