目錄

引言

WebSocket介紹

HTTP與WebSocket的區(qū)別

WebSocket測試方法

使用在線工具

使用Postman

使用Jmeter

使用Python

結語

引言

你是否曾經為 WebSocket 接口測試中復雜的協議和難以捕獲的數據而感到束手無策？WebSocket 協議與傳統的 HTTP 協議不同，測試方法也需要針對其特殊性進行優(yōu)化。

本文將向您介紹一些在 WebSocket 接口測試中提高效率的實用方法和技巧，希望對您學習WebSocket有所幫助。

WebSocket介紹

WebSocket 是一種基于在單個 TCP 連接上進行全雙工通信的協議，是從HTML5開始提供的一種瀏覽器與服務器之間進行全雙工通訊的網絡技術，解決了HTTP協議不適用于實時通信的缺點，相較于 HTTP 協議，WebSocket 協議實現了持久化網絡通信，可以實現客戶端和服務端的長連接，能夠進行雙向實時通信，協議名為"ws"。

HTTP與WebSocket的區(qū)別

1. 兩者都是基于TCP協議進行數據傳輸，具有可靠的傳輸能力，而且都是應用層協議
2. HTTP是單向的通信，只能由客戶端向服務端發(fā)送請求，服務端無法主動向客戶端發(fā)送消息
3. WebSocket是雙向通信，客戶端和服務端是通過握手建立連接，可實現全雙工通信，雙方可以同時主動向對方發(fā)送消息
4. WebSocket在建立連接時數據是通過HTTP協議傳輸的，但是建立之后，真正傳輸數據時使用的是更加可靠的TCP協議

二者關系

• WebSocket在建立連接時，HTTP通過Upgrade消息頭來通知客戶端需要采用WebSocket協議來完成這個請求，在發(fā)送完此響應最后的空行后，服務器就會切換到WebSocket協議，接口返回狀態(tài)碼為101，表示服務器已經理解了客戶端的請求，需要升級協議

WebSocket測試方法

使用在線工具

在線工具有很多，自行百度吧！個人習慣用這個工具

使用Postman

Postman支持測試WebSocket協議的接口，此文章發(fā)布時還處于公測階段，需要登錄后才能使用，入口如下圖所示

操作跟測試HTTP請求類似，不過多介紹啦！如下圖所示

使用Jmeter

使用Jmeter既可以單獨測接口，也可以進行性能測試，但測試WebSocket協議需要安裝插件，下載插件后放到Jmeter安裝路徑中的lib/ext目錄下，重啟Jmeter后在【選項】菜單中會出現【Plugins Manager】，然后在可用插件Tab頁搜索WebSocket，進行安裝即可，如下圖所示

安裝后新建【線程組】，在添加【取樣器】中會出現6項WebSocket的組件

• WebSocket Close：用于關閉WebSocket連接
• WebSocket Open Connection：只建立WebSocket連接，不發(fā)送數據
• WebSocket Ping/Pong：用于心跳檢測，為了保持長連接，防止客戶端被服務端判定為不活躍而被斷開
• WebSocket Single Read Sampler：用于接收一個（文本或二進制）數據
• WebSocket Single Write Sampler：用于發(fā)送一個（文本或二進制）數據
• WebSocket request-response Sampler：用于執(zhí)行基本的請求和響應交換，既能發(fā)送數據又可以接收數據

若有使用過Jmeter就肯定能明白這些組件的使用方法，以WebSocket request-response Sampler為例，頁面信息介紹如下圖所示

使用Python

首先需要安裝websocket-client

pip install websocket-client

對于只是測試WebSocket接口發(fā)送接收消息是否正常，不需要長時間連接，可以使用下面的短連接方法，在請求結束后會主動斷開連接

from websocket import create_connection

user1 = create_connection("ws://localhost:6688/1")	# 用戶1打開連接
user2 = create_connection("ws://localhost:6688/2")	# 用戶2打開連接

print("獲取響應狀態(tài)碼：", user1.getstatus()) # 正常應返回101
print("獲取響應頭：", user2.getheaders())

user1.send("你好呀！")	 # 用戶1向服務端發(fā)送消息

res1 = user1.recv()		# 查看服務端推送的消息
res2 = user2.recv()

if res1 == res2:
    print("服務端消息推送成功！\n")
else:
    print("消息獲取異常！\n")
    
print("查看用戶1收到的消息：",res1,"\n查看用戶2收到的消息：",res2)

若是要對WebSocket接口做持續(xù)監(jiān)聽工作，需要長連接，但不需要傳參，只是監(jiān)聽消息，使用WebSocketApp類就更合適啦！方法如下

import websocket

def on_open(ws):				# 定義用來處理打開連接的方法
    print("打開連接")
    
def on_message(ws, message):	# 定義用來監(jiān)聽服務器返回消息的方法
    print("監(jiān)聽到服務器返回的消息，：\n", message)

def on_error(ws, error):		# 定義用來處理錯誤的方法
    print("連接出現異常：\n", error)

def on_close(ws):				# 定義用來處理斷開連接的方法
    print("關閉連接")

if __name__ == "__main__":
    websocket.enableTrace(True) # 可選擇開啟跟蹤，在控制臺可以看到詳細的信息
    ws = websocket.WebSocketApp("ws://localhost:6688",
                                on_open=on_open,
                                on_message=on_message,
                                on_error=on_error,
                                on_close=on_close)
    ws.run_forever()	# 調用run_forever方法，保持長連接

還可以使用Pytest對WebSocket接口實現自動化測試，示例如下

import pytest
from websocket import create_connection

class TestDyd():
    url = "ws://localhost:6688"

    @classmethod
    def setup_class(cls):
        cls.ws = create_connection(cls.url)  # 建立連接
        cls.ws.settimeout(5)  # 設置超時時間

    def test_connect(self):
        # 通過狀態(tài)碼判斷連接是否正常
        assert self.ws.getstatus() == 101

    def test_send(self):
        params = "你好呀！"  			# 定義傳參

        self.ws.send(params)			# 發(fā)送請求
        result = self.ws.recv()			# 獲取響應結果
        print("收到來自服務端的消息：", result)	# 打印響應結果
        # 因為該測試項目傳參會顯示在響應中，所以通過判斷傳參是否在響應結果中進行斷言
        assert params in result

if __name__ == '__main__':
    pytest.main(["-vs"])

傳參可以使用@pytest.mark.parametrize()裝飾器，示例如下

import pytest
from websocket import create_connection

class TestDyd():
    
    data = [("/user/2", "今天開直播賣魚?。。。?),("/user/3", "魚可以買來放生嗎？"),
            ("/user/4", "那條翻白肚的魚有死亡證明嗎？"),("/user/5", "你的魚會唱歌嗎？"),
            ("/user/6", "你的魚上大學了嗎？"),("/user/7", "你的魚買回來需要隔離嗎？")]

    @pytest.mark.parametrize("user, word", data)
    def test_send(self, user, word):
        ws = create_connection("ws://localhost:6688"+user)
        params = word  

        ws.send(params)  # 發(fā)送請求
        result = ws.recv()  # 獲取響應結果
        print("來自服務端的消息：", result)
        assert params in result	# 斷言

先簡單介紹這些吧，關于Python腳本，可以參考官方文檔，里面有示例和常見問題解決方法

結語

這篇貼子到這里就結束了，最后，希望看這篇帖子的朋友能夠有所收獲。

?獲取方式：留言【WebSocket學習】即可

如果你覺得文章還不錯，請大家 點贊、分享、留言 下，因為這將是我持續(xù)輸出更多優(yōu)質文章的最強動力！

?文章來源地址http://www.zghlxwxcb.cn/news/detail-469069.html

到了這里，關于如何讓你的 WebSocket 接口測試更高效？拯救你的接口測試工作的文章就介紹完了。如果您還想了解更多內容，請在右上角搜索TOY模板網以前的文章或繼續(xù)瀏覽下面的相關文章，希望大家以后多多支持TOY模板網！