用Python進行websocket接口測試
我們在做接口測試時，除了常見的http接口，還有一種比較多見，就是socket接口，今天講解下怎么用Python進行websocket接口測試。
現(xiàn)在大多數(shù)用的都是websocket，那我們就先來安裝一下websocket的安裝包。
pip install websocket-client
安裝完之后，我們就開始我們的websocket之旅了。
我們先來看個炒雞簡單的栗子：
import websocket
ws = websocket.WebSocket()
ws.connect("ws://example.com/websocket",
http_proxy_host="proxy_host_name",
http_proxy_port=3128)
這個栗子就是創(chuàng)建一個websocket連接，這個模塊支持通過http代理訪問websocket。代理服務器允許使用connect方法連接到websocket端口。默認的squid設置是“只允許連接HTTPS端口”。
在websocket里，我們有常用的這幾個方法：
on_message方法：
def on_message(ws, message): print(message)
on_message是用來接受消息的，server發(fā)送的所有消息都可以用on_message這個方法來收取。
on_error方法：
def on_error(ws, error): print(error)
這個方法是用來處理錯誤異常的，如果一旦socket的程序出現(xiàn)了通信的問題，就可以被這個方法捕捉到。
on_open方法：
def on_open(ws):
def run(*args):
for i in range(30):
# send the message, then wait
# so thread doesn't exit and socket
# isn't closed
ws.send("Hello %d" % i)
time.sleep(1)
time.sleep(1)
ws.close()
print("Thread terminating...")
Thread(target=run).start()
on_open方法是用來保持連接的，上面這樣的一個例子，就是保持連接的一個過程，每隔一段時間就會來做一件事,他會在30s內(nèi)一直發(fā)送hello。最后停止。
on_close方法：
def on_close(ws):
print("### closed ###")
onclose主要就是關閉socket連接的。
如何創(chuàng)建一個websocket應用：
ws = websocket.WebSocketApp("wss://echo.websocket.org")
括號里面就是你要連接的socket的地址，在WebSocketApp這個實例化的方法里面還可以有其他參數(shù)，這些參數(shù)就是我們剛剛介紹的這些方法。
ws = websocket.WebSocketApp("ws://echo.websocket.org/",
on_message=on_message,
on_error=on_error,
on_close=on_close)
指定了這些參數(shù)之后就可以直接進行調(diào)用了，例如：
ws.on_open = on_open
這樣就是調(diào)用了on_open方法
如果我們想讓我們的socket保持長連接，一直連接著，就可以使用run_forever方法：
ws.run_forever()
完整代碼：
import websocket
from threading import Thread
import time
import sys
def on_message(ws, message):
print(message)
def on_error(ws, error):
print(error)
def on_close(ws):
print("### closed ###")
def on_open(ws):
def run(*args):
for i in range(3):
# send the message, then wait
# so thread doesn't exit and socket
# isn't closed
ws.send("Hello %d" % i)
time.sleep(1)
time.sleep(1)
ws.close()
print("Thread terminating...")
Thread(target=run).start()
if __name__ == "__main__":
websocket.enableTrace(True)
host = "ws://echo.websocket.org/"
ws = websocket.WebSocketApp(host,
on_message=on_message,
on_error=on_error,
on_close=on_close)
ws.on_open = on_open
ws.run_forever()
如果想要通信一條短消息，并在完成后立即斷開連接，我們可以使用短連接：
from websocket import create_connection
ws = create_connection("ws://echo.websocket.org/")
print("Sending 'Hello, World'...")
ws.send("Hello, World")
print("Sent")
print("Receiving...")
result = ws.recv()
print("Received '%s'" % result)
ws.close()
關于websocket的介紹就到這兒了。
以上就是用Python進行websocket接口測試的詳細內(nèi)容，更多關于python 接口測試的資料請關注腳本之家其它相關文章！
相關文章
今天小編就為大家分享一篇Mac 使用python3的matplot畫圖不顯示的解決，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-11-11
Python人工智能之路 jieba gensim 最好別分家之最簡單的相似度實現(xiàn)
這篇文章主要介紹了Python人工智能之路 jieba gensim 最好別分家之最簡單的相似度實現(xiàn) ,非常不錯，具有一定的參考借鑒價值，需要的朋友可以參考下2019-08-08
這篇文章主要介紹了Python閉包和裝飾器用法,結合實例形式詳細分析了Python閉包和裝飾器的相關概念、原理、使用技巧與相關操作注意事項,需要的朋友可以參考下2019-05-05
關于Pandas?count()與values_count()的用法及區(qū)別
這篇文章主要介紹了關于Pandas?count()與values_count()的用法及區(qū)別，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
本文將從零開始，僅僅利用基礎的numpy庫，使用Python實現(xiàn)一個最簡單的神經(jīng)網(wǎng)絡(或者說是簡易的LR，因為LR就是一個單層的神經(jīng)網(wǎng)絡)，解決一個點擊率預估的問題。感興趣的朋友跟隨小白一起看看吧2018-10-10
Python threading Local()函數(shù)用法案例詳解
這篇文章主要介紹了Python threading Local()函數(shù)用法案例詳解,本篇文章通過簡要的案例,講解了該項技術的了解與使用,以下就是詳細內(nèi)容,需要的朋友可以參考下2021-09-09
這篇文章主要介紹了python數(shù)據(jù)類型強制轉換實例詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-06-06
在自然語言處理（NLP）中,將文本序列轉化為數(shù)字序列（tokenid）后,為了使模型能更好地理解這些數(shù)字背后的含義,引入了嵌入層（Embedding）通過簡單的示例,可以看出Embedding的獲取過程及其在理解語言中的關鍵作用2024-11-11
這篇文章主要介紹了如何用python開發(fā)一款翻譯工具，幫助大家更好的理解和學習python，感興趣的朋友可以了解下2020-10-10
python爬蟲實戰(zhàn)之最簡單的網(wǎng)頁爬蟲教程
在我們?nèi)粘Ｉ暇W(wǎng)瀏覽網(wǎng)頁的時候，經(jīng)常會看到一些好看的圖片，我們就希望把這些圖片保存下載，或者用戶用來做桌面壁紙，或者用來做設計的素材。下面這篇文章就來給大家介紹了關于利用python實現(xiàn)最簡單的網(wǎng)頁爬蟲的相關資料，需要的朋友可以參考借鑒，下面來一起看看吧。2017-08-08