功能说明:
连接websocket服务发送请求数据并接收,最后关闭连接,完成一次websocket接口调试,适用场景:单接口调试,或接口自动化测试
代码:
from websocket import create_connection
import websocket
import json
import time
import sys
def client(host):
try:
ws = create_connection(host) # 创建连接
ws.settimeout(10) # 设置超时时间
print (ws.getstatus())
recv = ws.recv()
print ('连接信息',recv)
# 请求数据
t = time.time()
timeStamp = int(round(t * 1000)) #毫秒级时间戳
data_dict = {
"cmd":"goTest",
"timestamp":timeStamp,
"body": {}
}
data_json = json.dumps(data_dict,indent=4)
print ("请求参数:%s"%data_json)
ws.send(data_json)
# 接收数据
try:
result = ws.recv()
res_dict = json.loads(result)
print ('接收数据:\n',json.dumps(res_dict,indent=4,ensure_ascii=False))
except Exception as e:
result = e
print (result)
except Exception as e:
print ('服务连接失败:',e)
sys.exit()
ws.shutdown()
if __name__ == "__main__":
host = "ws://192.168.x.x:端口"
client(host)
测试执行:
测试结果: