FastAPI(60)- 针对 WebSocket 进行单元测试

FastAPI(60)- 针对 WebSocket 进行单元测试 

 

前言

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

FastAPI 的单元测试:https://www.cnblogs.com/poloyy/p/15354901.html

 

直接上代码

FastAPI(60)- 针对 WebSocket 进行单元测试
#!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"}
FastAPI(60)- 针对 WebSocket 进行单元测试

 

运行

命令行敲

pytest test.py

  

websocket_connect()

必须用作上下文管理器,使用 with ... as ... 来管理

FastAPI(60)- 针对 WebSocket 进行单元测试

 

发送数据的三种方法

  • .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 连接

 

上一篇:Solution -「CF 232E」Quick Tortoise


下一篇:数据结构——栈(一)(2022考研)