FastAPI（60）- 针对 WebSocket 进行单元测试

最新推荐文章于 2024-08-10 14:02:15 发布

小菠萝测试笔记

最新推荐文章于 2024-08-10 14:02:15 发布

阅读量790

点赞数

文章标签：单元测试 python java websocket socket

本文链接：https://blog.csdn.net/qq_33801641/article/details/120644603

版权

这篇博客介绍了如何对 FastAPI 中的 WebSocket 进行单元测试，包括如何运行测试、连接、发送和接收不同类型的数据，以及如何关闭连接。详细讲解了.send_text(), .send_bytes(), .send_json()等方法以及.receive_text(), .receive_bytes(), .receive_json()的使用，并提到了可能遇到的WebSocketDisconnect异常。" 112069348,10293602,15分钟生活圈距离衰减：高斯曲线拟合,"['GIS', '数据分析', '城市规划', '交通研究', '数据拟合']

摘要由CSDN通过智能技术生成

前言

还是比较简单的，直接写代码啦

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