python+websocket接口测试

1.连接websocket服务器

import logging
from websocket import creat_connertion
logger = logging.getLogger(__name__)
url = 'ws://echo.websocket.org' #一个在线回环websocket接口，必须以websocket的方式连接后访问，无法直接在网页端输入该地址访问
wss = creat_connertion(url, timeout=timeout)

2.发送websocket消息

wss.send('hello world')

3.接受websocket消息

res = wss.recv()
logger.info(res)

4.关闭websocket连接

wss.close()

websocket第三方库的调用不支持直接发送除字符串外的其他数据类型，所以在发送请求之前需要将python结构化的格式，转换成字符串类型或者json字符串后，再发起websocket的接口请求。

#待发送的数据体格式为：
data= {
      "a" : "abcd",
      "b" : 123
}
#发送前需要把数据处理成json字符串
new_data = json.dumps(data,ensure_ascii=Flase)
wss.send(new_data)