在我們測驗中，肯定會對介面進行測驗，之前也分享過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=https://www.cnblogs.com/leiziv5/p/='Hello WebSocket': await websocket.send_text(f"訊息是: {data}") break else: await websocket.send_text(f"訊息是: {data}")
其實要做的測驗很簡單，我們可以利用這個方式對于我們已經開發的介面進行單元測驗，
文章首發在公眾號，歡迎關注，
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/335093.html
標籤：其他