FastAPI 学习之路(四十八)WebSockets(四)接口测试

在我们测试的过程中,肯定会对接口进行测试。之前分享过的FastApi学习之路(三十八)对开发接口进行测试,那么我们针对websockets接口怎么测试呢。

其实也很简单

from fastapi.testclient import TestClient
from main import app


def test_websocket():
    client = TestClient(app)
    with client.websocket_connect("/items/ws?token=fake-token") as websocket:
        websocket.send_text("Hello, this is testing websocket")
        data = websocket.receive_text()
        assert str(data) == f"Message is: Hello, this is testing websocket"


if __name__ == '__main__':
    test_websocket()

执行测试发现报错

这个错误,主要是我们在最后的时候没有释放链接,我们可以在代码中链接接受到消息后,返回完毕关闭链接,或者说我们单元测试的时候关闭链接。

@app.websocket("/items/ws")
async def websocket_endpoint(
    websocket: WebSocket,
    cookie_or_token: str = Depends(get_cookie_or_token),
):
    await websocket.accept()
    while True:
        data = await websocket.receive_text()
        if data == "Hello, this is testing websocket":
            await websocket.send_text(f"Message is: {data}")
            break
        else:
            await websocket.send_text(f"Message is: {data}")

其实要做的测试很简单,我们可以利用这个方式对于我们已经开发的接口进行测试。

相关推荐
skywalk81631 天前
三周精通FastAPI:33 在编辑器中调试
python·编辑器·fastapi
敲代码不忘补水3 天前
使用 PyCharm 构建 FastAPI 项目:零基础入门 Web API 开发
后端·python·fastapi
skywalk81633 天前
三周精通FastAPI:32 探索如何使用pytest进行高效、全面的项目测试!
开发语言·python·fastapi
德育处主任4 天前
『FastAPI』快速掌握“请求与响应”的基础用法
后端·python·fastapi
花酒锄作田6 天前
[python]Gunicorn加持下的Flask性能测试
python·nginx·golang·flask·fastapi
萤火架构8 天前
使用FastAPI整合Gradio和Django
django·fastapi·gradio
练习两年半的工程师9 天前
建立一个简单的todo应用程序(前端React;后端FastAPI;数据库MongoDB)
前端·数据库·react.js·fastapi
岳涛@心馨电脑12 天前
【硬啃Dash-Fastapi-Admin】03-requirements-pg.txt 速览
信息可视化·fastapi·dash
skywalk816312 天前
三周精通FastAPI:16 Handling Errors处理错误
开发语言·python·fastapi
zhiyong_will13 天前
Uvicorn 原理及源码分析
fastapi