Тестирование WebSocket¶
🌐 Перевод выполнен с помощью ИИ и людей
Этот перевод был сделан ИИ под руководством людей. 🤝
В нем могут быть ошибки из-за неправильного понимания оригинального смысла или неестественности и т. д. 🤖
Вы можете улучшить этот перевод,помогая нам лучше направлять ИИ LLM.
Вы можете использовать тот жеTestClient для тестирования WebSocket.
Для этого используйтеTestClient с менеджером контекстаwith, подключаясь к WebSocket:
fromfastapiimportFastAPIfromfastapi.testclientimportTestClientfromfastapi.websocketsimportWebSocketapp=FastAPI()@app.get("/")asyncdefread_main():return{"msg":"Hello World"}@app.websocket("/ws")asyncdefwebsocket(websocket:WebSocket):awaitwebsocket.accept()awaitwebsocket.send_json({"msg":"Hello WebSocket"})awaitwebsocket.close()deftest_read_main():client=TestClient(app)response=client.get("/")assertresponse.status_code==200assertresponse.json()=={"msg":"Hello World"}deftest_websocket():client=TestClient(app)withclient.websocket_connect("/ws")aswebsocket:data=websocket.receive_json()assertdata=={"msg":"Hello WebSocket"}Примечание
Подробности смотрите в документации Starlette потестированию WebSocket.