FastAPI 學習之路(四十八)WebSockets(四)接口測試 FastAPI 學習之路(三十八)對開發接口進行測試

  在我們測試中,肯定會對接口進行測試。之前也分享過FastAPI 學習之路(三十八)對開發接口進行測試,那麼我們針對WebSockets接口怎麼測試呢。

        其實很簡單。

from fastapi.testclient import TestClient
def test_websocket():
    client = TestClient(app)
    with client.websocket_connect("/items/ws?token="+"leizishuoceshi") as websocket:
        websocket.send_text("Hello WebSocket")
        data = websocket.receive_text()
        assert str(data) =="消息是: Hello 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 WebSocket':
            await websocket.send_text(f"消息是: {data}")
            break
        else:
            await websocket.send_text(f"消息是: {data}")

   其實要做的測試很簡單,我們可以利用這個方式對於我們已經開發的接口進行單元測試。

文章首發在公衆號,歡迎關注。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章