如何用python编写一个简单的ws接口自动化测试脚本

我们通常做接口自动化测试有两种方式：1、使用工具  2、写脚本

目前能够做接口自动化测试的工具的有JMeter、Postman、LoadRunner、soapUI、Apipost等，但是大部分工具都是不能发送ws请求，如JMeter、Postman；或者不能进行ws接口的自动化测试，如Apipost支持发送ws接口，但是不支持ws接口进行自动化测试。

因此，我们通常可以使用编写简单的脚本来满足我们的需求。

例子：调用一个websocket接口20次，每次间隔2s，接收服务器的响应。

import websocket
import time

# WebSocket的URL
url = "xxxxx"

def test(url):
    # 创建一个WebSocket连接
    ws = websocket.create_connection(url)

    # 发送数据到服务器 
    # ws.send("Your message")

    # 接收服务器的响应
    response = ws.recv()
    print("Received: %s" % response)

    # 关闭连接
    ws.close()


# 循环20次
for i in range(20):
    time.sleep(2)  # 等待2秒
    test(url)

注意：需要安装websocket-client库