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