在我们测试的过程中,肯定会对接口进行测试。之前分享过的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}")
其实要做的测试很简单,我们可以利用这个方式对于我们已经开发的接口进行测试。