Coverage for tests / test_tutorial / test_websockets / test_tutorial001.py: 100%
20 statements
« prev ^ index » next coverage.py v7.13.3, created at 2026-04-06 01:24 +0000
1import pytest 1efgh
2from fastapi.testclient import TestClient 1efgh
3from fastapi.websockets import WebSocketDisconnect 1efgh
5from docs_src.websockets_.tutorial001_py310 import app 1efgh
7client = TestClient(app) 1efgh
10def test_main(): 1efgh
11 response = client.get("/") 1ijkl
12 assert response.status_code == 200, response.text 1ijkl
13 assert b"<!DOCTYPE html>" in response.content 1ijkl
16def test_websocket(): 1efgh
17 with pytest.raises(WebSocketDisconnect): 1abcd
18 with client.websocket_connect("/ws") as websocket: 1abcd
19 message = "Message one" 1abcd
20 websocket.send_text(message) 1abcd
21 data = websocket.receive_text() 1abcd
22 assert data == f"Message text was: {message}" 1abcd
23 message = "Message two" 1abcd
24 websocket.send_text(message) 1abcd
25 data = websocket.receive_text() 1abcd
26 assert data == f"Message text was: {message}" 1abcd