前言
上一篇我们分享了FastAPI 学习之路（四十六）WebSockets（三）登录后才可以聊天，那么我们这次看下WebSockets接口怎么测试？
正文
在我们测试中，肯定会对接口进行测试。之前也分享过FastAPI 学习之路（三十八）对开发接口进行测试。那么我们针对WebSockets接口怎么测试呢。
其实很简单。
代码语言：javascript复制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"然后执行测试即可。
这个错误，主要是我们在最后的时候没有释放链接，我们可以在代码中链接接受到消息后，返回完毕关闭，或者说我们单元测试的时候关闭连接.
代码语言：javascript复制@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}")其实要做的测试很简单，我们可以利用这个方式对于我们已经开发的接口进行单元测试。
后记
发现问题，解决问题。遇到问题，慢慢解决问题即可。
欢迎关注雷子说测试开发，后续将会持续为大家分享更多的技术知识
如果你有问题可以留言或者加我微信:952943386。
2021，一起牛转钱坤，一起牛逼。
如果觉得这篇文章还不错，来个【分享、点赞、在看】三连吧，让更多的人也看到~
