FastAPI（60）- 針對 WebSocket 進行單元測試

前言

還是比較簡單的，直接寫程式碼啦

FastAPI 的單元測試：https://www.cnblogs.com/poloyy/p/15354901.html

直接上程式碼

#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
# author: 小菠蘿測試筆記
# blog:  https://www.cnblogs.com/poloyy/
# time: 2021/10/7 9:05 下午
# file: 51_test_websocket.py
"""

from fastapi import FastAPI
from
 
 fastapi.testclient import TestClient
from fastapi.websockets import WebSocket

app = FastAPI()
testClient = TestClient(app)


@app.websocket("/ws")
async def connect(websocket: WebSocket):
    await websocket.accept()
    await websocket.send_json({"msg": "Hello WebSocket"})
    await websocket.close()


def
 
 test_websocket():
    # 建立 websocket 連線
    with testClient.websocket_connect("/ws") as websocket:
        # 接收資料
        data = websocket.receive_json()
        # 斷言資料
        assert data == {"msg": "Hello WebSocket"}

執行

命令列敲

pytest test.py

websocket_connect()

必須用作上下文管理器，使用 with ... as ... 來管理

傳送資料的三種方法

• .send_text(data)：將給定的文字傳送到應用程式
• .send_bytes(data)：將給定的位元組傳送到應用程式
• .send_json(data, mode="text")：將給定的資料傳送到應用程式，使用 mode="binary" 通過二進位制資料幀傳送 JSON

接收資料的三種方法

• .receive_text()：等待應用程式傳送的傳入文字並返回它
• .receive_bytes()：等待應用程式傳送的傳入位元組串並返回它
• .receive_json(mode="text")：等待應用程式傳送的傳入 json 資料並返回它，使用 mode="binary" 通過二進位制資料幀傳送 JSON

可能會引發 starlette.websockets.WebSocketDisconnect

關閉連線

.close(code=1000)：在客戶端關閉 websocket 連線

• 前言
• 直接上程式碼
• 執行
• websocket_connect()
• 傳送資料的三種方法
• 接收資料的三種方法
• 關閉連線

• 本文作者： 小菠蘿測試筆記
• 本文連結： https://www.cnblogs.com/poloyy/p/15377141.html