Coverage for tests/test_tutorial/test_websockets/test_tutorial003.py: 100%

1from fastapi.testclient import TestClient 7 ctx1hijklmn

2 

3from docs_src.websockets.tutorial003 import app, html 7 ctx1hijklmn

4 

5client = TestClient(app) 7 ctx1hijklmn

6 

7 

8def test_get(): 7 ctx1hijklmn

9 response = client.get("/") 7 ctx1opqrstu

10 assert response.text == html 7 ctx1opqrstu

11 

12 

13def test_websocket_handle_disconnection(): 7 ctx1hijklmn

14 with client.websocket_connect("/ws/1234") as connection, client.websocket_connect( 7 ctx1abcdefg

15 "/ws/5678" 

16 ) as connection_two: 

17 connection.send_text("Hello from 1234") 7 ctx1abcdefg

18 data1 = connection.receive_text() 7 ctx1abcdefg

19 assert data1 == "You wrote: Hello from 1234" 7 ctx1abcdefg

20 data2 = connection_two.receive_text() 7 ctx1abcdefg

21 client1_says = "Client #1234 says: Hello from 1234" 7 ctx1abcdefg

22 assert data2 == client1_says 7 ctx1abcdefg

23 data1 = connection.receive_text() 7 ctx1abcdefg

24 assert data1 == client1_says 7 ctx1abcdefg

25 connection_two.close() 7 ctx1abcdefg

26 data1 = connection.receive_text() 7 ctx1abcdefg

27 assert data1 == "Client #5678 left the chat" 7 ctx1abcdefg