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

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