Coverage for tests/test_tutorial/test_websockets/test_tutorial002_an_py39.py: 100%
80 statements
« prev ^ index » next coverage.py v7.6.1, created at 2025-01-13 13:38 +0000
1import pytest 1eabcd
2from fastapi import FastAPI 1eabcd
3from fastapi.testclient import TestClient 1eabcd
4from fastapi.websockets import WebSocketDisconnect 1eabcd
6from ...utils import needs_py39 1eabcd
9@pytest.fixture(name="app") 1eabcd
10def get_app(): 1eabcd
11 from docs_src.websockets.tutorial002_an_py39 import app 1abcd
13 return app 1abcd
16@needs_py39 1eabcd
17def test_main(app: FastAPI): 1eabcd
18 client = TestClient(app) 1rstu
19 response = client.get("/") 1rstu
20 assert response.status_code == 200, response.text 1rstu
21 assert b"<!DOCTYPE html>" in response.content 1rstu
24@needs_py39 1eabcd
25def test_websocket_with_cookie(app: FastAPI): 1eabcd
26 client = TestClient(app, cookies={"session": "fakesession"}) 1jklm
27 with pytest.raises(WebSocketDisconnect): 1jklm
28 with client.websocket_connect("/items/foo/ws") as websocket: 1jklm
29 message = "Message one" 1jklm
30 websocket.send_text(message) 1jklm
31 data = websocket.receive_text() 1jklm
32 assert data == "Session cookie or query token value is: fakesession" 1jklm
33 data = websocket.receive_text() 1jklm
34 assert data == f"Message text was: {message}, for item ID: foo" 1jklm
35 message = "Message two" 1jklm
36 websocket.send_text(message) 1jklm
37 data = websocket.receive_text() 1jklm
38 assert data == "Session cookie or query token value is: fakesession" 1jklm
39 data = websocket.receive_text() 1jklm
40 assert data == f"Message text was: {message}, for item ID: foo" 1jklm
43@needs_py39 1eabcd
44def test_websocket_with_header(app: FastAPI): 1eabcd
45 client = TestClient(app) 1nopq
46 with pytest.raises(WebSocketDisconnect): 1nopq
47 with client.websocket_connect("/items/bar/ws?token=some-token") as websocket: 1nopq
48 message = "Message one" 1nopq
49 websocket.send_text(message) 1nopq
50 data = websocket.receive_text() 1nopq
51 assert data == "Session cookie or query token value is: some-token" 1nopq
52 data = websocket.receive_text() 1nopq
53 assert data == f"Message text was: {message}, for item ID: bar" 1nopq
54 message = "Message two" 1nopq
55 websocket.send_text(message) 1nopq
56 data = websocket.receive_text() 1nopq
57 assert data == "Session cookie or query token value is: some-token" 1nopq
58 data = websocket.receive_text() 1nopq
59 assert data == f"Message text was: {message}, for item ID: bar" 1nopq
62@needs_py39 1eabcd
63def test_websocket_with_header_and_query(app: FastAPI): 1eabcd
64 client = TestClient(app) 1fghi
65 with pytest.raises(WebSocketDisconnect): 1fghi
66 with client.websocket_connect("/items/2/ws?q=3&token=some-token") as websocket: 1fghi
67 message = "Message one" 1fghi
68 websocket.send_text(message) 1fghi
69 data = websocket.receive_text() 1fghi
70 assert data == "Session cookie or query token value is: some-token" 1fghi
71 data = websocket.receive_text() 1fghi
72 assert data == "Query parameter q is: 3" 1fghi
73 data = websocket.receive_text() 1fghi
74 assert data == f"Message text was: {message}, for item ID: 2" 1fghi
75 message = "Message two" 1fghi
76 websocket.send_text(message) 1fghi
77 data = websocket.receive_text() 1fghi
78 assert data == "Session cookie or query token value is: some-token" 1fghi
79 data = websocket.receive_text() 1fghi
80 assert data == "Query parameter q is: 3" 1fghi
81 data = websocket.receive_text() 1fghi
82 assert data == f"Message text was: {message}, for item ID: 2" 1fghi
85@needs_py39 1eabcd
86def test_websocket_no_credentials(app: FastAPI): 1eabcd
87 client = TestClient(app) 1vxyz
88 with pytest.raises(WebSocketDisconnect): 1vxyz
89 with client.websocket_connect("/items/foo/ws"): 1vxyz
90 pytest.fail( 1v
91 "did not raise WebSocketDisconnect on __enter__"
92 ) # pragma: no cover
95@needs_py39 1eabcd
96def test_websocket_invalid_data(app: FastAPI): 1eabcd
97 client = TestClient(app) 1wABC
98 with pytest.raises(WebSocketDisconnect): 1wABC
99 with client.websocket_connect("/items/foo/ws?q=bar&token=some-token"): 1wABC
100 pytest.fail( 1w
101 "did not raise WebSocketDisconnect on __enter__"
102 ) # pragma: no cover