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