Python測試基于websocket協定的即時通訊接口【多測師】

使用python中的websocket-client子產品來做相關的接口測試

import webclient

def creatWS(user,password):#生成ws連結的函數
    token = getToken(user,password)#調用系統的登入接口，擷取token
    return "ws://url?token=%s"%token#生成ws連結


def msgChat(user,password):#即時聊天消息
    websocket.enableTrace(True)#打開跟蹤，檢視日志
    strUrl= creatWS(user,password)
    ws = websocket.create_connection(strUrl)#建立連結
    data={json格式的資料}
    ws.send(json.dumps(data))
    #print json.loads(ws.recv())
    print ws.recv()#檢視傳回
    ws.close()      

import websocket  
url = 'wss://host:port/rt/tr'  #websocket連接配接位址
ws = websocket.create_connection(url)  #建立連接配接'''data為json格式'''
data = {"body":{"address":"陝西省","location":{"latitude":11.23256618923611,"longitude":218.9637478298611,"timestamp":1541735071192},"state":"arrivalDepot","timestamp":1541862074000,"trackId":3218111000038001,"transportId":218111000038},"token":"token","type":"roadtrace_state"}
ws.send(json.dumps(data))   #json轉化為字元串，必須轉化
print(ws.recv())    #伺服器響應資料
ws.close()   #關閉連接配接