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