目錄

引言

WebSocket介紹

HTTP與WebSocket的區(qū)別

WebSocket測試方法

使用在線工具

使用Postman

使用Jmeter

使用Python

結(jié)語

---

引言

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

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

WebSocket介紹

WebSocket 是一種基于在單個(gè) TCP 連接上進(jìn)行全雙工通信的協(xié)議，是從HTML5開始提供的一種瀏覽器與服務(wù)器之間進(jìn)行全雙工通訊的網(wǎng)絡(luò)技術(shù)，解決了HTTP協(xié)議不適用于實(shí)時(shí)通信的缺點(diǎn)，相較于 HTTP 協(xié)議，WebSocket 協(xié)議實(shí)現(xiàn)了持久化網(wǎng)絡(luò)通信，可以實(shí)現(xiàn)客戶端和服務(wù)端的長連接，能夠進(jìn)行雙向?qū)崟r(shí)通信，協(xié)議名為"ws"。

HTTP與WebSocket的區(qū)別

1. 兩者都是基于TCP協(xié)議進(jìn)行數(shù)據(jù)傳輸，具有可靠的傳輸能力，而且都是應(yīng)用層協(xié)議
2. HTTP是單向的通信，只能由客戶端向服務(wù)端發(fā)送請求，服務(wù)端無法主動向客戶端發(fā)送消息
3. WebSocket是雙向通信，客戶端和服務(wù)端是通過握手建立連接，可實(shí)現(xiàn)全雙工通信，雙方可以同時(shí)主動向?qū)Ψ桨l(fā)送消息
4. WebSocket在建立連接時(shí)數(shù)據(jù)是通過HTTP協(xié)議傳輸?shù)?#xff0c;但是建立之后，真正傳輸數(shù)據(jù)時(shí)使用的是更加可靠的TCP協(xié)議

二者關(guān)系

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

WebSocket測試方法

使用在線工具

在線工具有很多，自行百度吧！個(gè)人習(xí)慣用這個(gè)工具

使用Postman

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

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

使用Jmeter

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

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

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

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

使用Python

首先需要安裝websocket-client

pip install websocket-client

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

from websocket import create_connectionuser1 = create_connection("ws://localhost:6688/1")	# 用戶1打開連接
user2 = create_connection("ws://localhost:6688/2")	# 用戶2打開連接print("獲取響應(yīng)狀態(tài)碼：", user1.getstatus()) # 正常應(yīng)返回101
print("獲取響應(yīng)頭：", user2.getheaders())user1.send("你好呀！")	 # 用戶1向服務(wù)端發(fā)送消息res1 = user1.recv()		# 查看服務(wù)端推送的消息
res2 = user2.recv()if res1 == res2:print("服務(wù)端消息推送成功！\n")
else:print("消息獲取異常！\n")print("查看用戶1收到的消息：",res1,"\n查看用戶2收到的消息：",res2)

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

import websocketdef on_open(ws):				# 定義用來處理打開連接的方法print("打開連接")def on_message(ws, message):	# 定義用來監(jiān)聽服務(wù)器返回消息的方法print("監(jiān)聽到服務(wù)器返回的消息，：\n", message)def on_error(ws, error):		# 定義用來處理錯(cuò)誤的方法print("連接出現(xiàn)異常：\n", error)def on_close(ws):				# 定義用來處理斷開連接的方法print("關(guān)閉連接")if __name__ == "__main__":websocket.enableTrace(True) # 可選擇開啟跟蹤，在控制臺可以看到詳細(xì)的信息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()	# 調(diào)用run_forever方法，保持長連接

還可以使用Pytest對WebSocket接口實(shí)現(xiàn)自動化測試，示例如下

import pytest
from websocket import create_connectionclass TestDyd():url = "ws://localhost:6688"@classmethoddef setup_class(cls):cls.ws = create_connection(cls.url)  # 建立連接cls.ws.settimeout(5)  # 設(shè)置超時(shí)時(shí)間def test_connect(self):# 通過狀態(tài)碼判斷連接是否正常assert self.ws.getstatus() == 101def test_send(self):params = "你好呀！"  			# 定義傳參self.ws.send(params)			# 發(fā)送請求result = self.ws.recv()			# 獲取響應(yīng)結(jié)果print("收到來自服務(wù)端的消息：", result)	# 打印響應(yīng)結(jié)果# 因?yàn)樵摐y試項(xiàng)目傳參會顯示在響應(yīng)中，所以通過判斷傳參是否在響應(yīng)結(jié)果中進(jìn)行斷言assert params in resultif __name__ == '__main__':pytest.main(["-vs"])

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

import pytest
from websocket import create_connectionclass TestDyd():data = [("/user/2", "今天開直播賣魚！！！！"),("/user/3", "魚可以買來放生嗎？"),("/user/4", "那條翻白肚的魚有死亡證明嗎？"),("/user/5", "你的魚會唱歌嗎？"),("/user/6", "你的魚上大學(xué)了嗎？"),("/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()  # 獲取響應(yīng)結(jié)果print("來自服務(wù)端的消息：", result)assert params in result	# 斷言

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

結(jié)語

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

?獲取方式：留言【W(wǎng)ebSocket學(xué)習(xí)】即可

如果你覺得文章還不錯(cuò)，請大家 點(diǎn)贊、分享、留言 下，因?yàn)檫@將是我持續(xù)輸出更多優(yōu)質(zhì)文章的最強(qiáng)動力！

?