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

隨着html5的廣泛應用，基於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()