还是比较简单的,直接写代码啦
FastAPI 的单元测试:https://cloud.tencent.com/developer/article/1886017
#!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
必须用作上下文管理器,使用 with ... as ... 来管理
可能会引发 starlette.websockets.WebSocketDisconnect
.close(code=1000):在客户端关闭 websocket 连接