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():