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}")

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

相关推荐
GuokLiu19 小时前
241121-REST、WebService与FastAPI之间的关系及示例代码
fastapi
T0uken3 天前
【Python】FastAPI:Token认证
开发语言·python·fastapi
OT.Ter4 天前
基于FastAPI实现本地大模型API封装调用
人工智能·算法·大模型·fastapi
skywalk81635 天前
三周精通FastAPI:42 手动运行服务器 - Uvicorn & Gunicorn with Uvicorn
运维·服务器·fastapi·gunicorn
股票GPT分析6 天前
如何利用必盈接口在 C#中完成股票量化程序
运维·服务器·数据库·python·fastapi
Magicapprentice6 天前
gunicorn 和 uvicorn部署fastapi 或者flask
flask·fastapi·gunicorn
股票GPT分析7 天前
如何在C#中处理必盈接口返回的股票数据?
python·fastapi
m0_637146937 天前
如何在FastAPI中灵活使用路径参数、查询参数和请求体参数进行接口设计
数据库·oracle·fastapi
m0_594526307 天前
探索 TraceBoard:统计你的键盘按键使用情况
python·计算机外设·开源软件·fastapi
晴天のVlog9 天前
Fastapi使用MongoDB作为数据库
数据库·python·mongodb·fastapi