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

1import pytest 6 ctx1fabcde

2from fastapi import FastAPI 6 ctx1fabcde

3from fastapi.testclient import TestClient 6 ctx1fabcde

4 

5from ...utils import needs_py39 6 ctx1fabcde

6 

7 

8@pytest.fixture(name="app") 6 ctx1fabcde

9def get_app(): 6 ctx1fabcde

10 from docs_src.websockets.tutorial003_py39 import app 5 ctx1abcde

11 

12 return app 5 ctx1abcde

13 

14 

15@pytest.fixture(name="html") 6 ctx1fabcde

16def get_html(): 6 ctx1fabcde

17 from docs_src.websockets.tutorial003_py39 import html 5 ctx1abcde

18 

19 return html 5 ctx1abcde

20 

21 

22@pytest.fixture(name="client") 6 ctx1fabcde

23def get_client(app: FastAPI): 6 ctx1fabcde

24 client = TestClient(app) 5 ctx1abcde

25 

26 return client 5 ctx1abcde

27 

28 

29@needs_py39 6 ctx1fabcde

30def test_get(client: TestClient, html: str): 6 ctx1fabcde

31 response = client.get("/") 5 ctx1lmnop

32 assert response.text == html 5 ctx1lmnop

33 

34 

35@needs_py39 6 ctx1fabcde

36def test_websocket_handle_disconnection(client: TestClient): 6 ctx1fabcde

37 with client.websocket_connect("/ws/1234") as connection, client.websocket_connect( 5 ctx1ghijk

38 "/ws/5678" 

39 ) as connection_two: 

40 connection.send_text("Hello from 1234") 5 ctx1ghijk

41 data1 = connection.receive_text() 5 ctx1ghijk

42 assert data1 == "You wrote: Hello from 1234" 5 ctx1ghijk

43 data2 = connection_two.receive_text() 5 ctx1ghijk

44 client1_says = "Client #1234 says: Hello from 1234" 5 ctx1ghijk

45 assert data2 == client1_says 5 ctx1ghijk

46 data1 = connection.receive_text() 5 ctx1ghijk

47 assert data1 == client1_says 5 ctx1ghijk

48 connection_two.close() 5 ctx1ghijk

49 data1 = connection.receive_text() 5 ctx1ghijk

50 assert data1 == "Client #5678 left the chat" 5 ctx1ghijk