python測試基於websocket協議的即時通信接口

隨着html5的普遍應用，基於websocket協議的即時通信有了愈來愈多的使用場景，本文使用python中的websocket-client模塊來作相關的接口測試

 1 import webclient
 2 
 3 def creatWS(user,password):#生成ws連接的函數
 4     token = getToken(user,password)#調用系統的登陸接口，獲取token
 5     return "ws://url?token=%s"%token#生成ws連接
 6 
 7 
 8 def msgChat(user,password):#即時聊天消息
 9     websocket.enableTrace(True)#打開跟蹤，查看日誌
10     strUrl= creatWS(user,password)
11     ws = websocket.create_connection(strUrl)#建立連接
12     data={json格式的數據}
13     ws.send(json.dumps(data))
14     #print json.loads(ws.recv())
15     print ws.recv()#查看返回
16     ws.close()