「案例」使用python对websocket接口测试

功能说明：

连接websocket服务发送请求数据并接收，最后关闭连接，完成一次websocket接口调试，适用场景：单接口调试，或接口自动化测试

代码：

from websocket import create_connectionimport websocketimport jsonimport timeimport sysdef 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)

测试执行：

测试结果：