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

1import importlib 4 ctx1abdc

2from types import ModuleType 4 ctx1abdc

3 

4import pytest 4 ctx1abdc

5from fastapi.testclient import TestClient 4 ctx1abdc

6 

7 

8@pytest.fixture( 4 ctx1abdc

9 name="mod", 

10 params=[ 

11 pytest.param("tutorial003_py310"), 

12 ], 

13) 

14def get_mod(request: pytest.FixtureRequest): 4 ctx1abdc

15 mod = importlib.import_module(f"docs_src.websockets.{request.param}") 3 ctx1abc

16 

17 return mod 3 ctx1abc

18 

19 

20@pytest.fixture(name="html") 4 ctx1abdc

21def get_html(mod: ModuleType): 4 ctx1abdc

22 return mod.html 3 ctx1abc

23 

24 

25@pytest.fixture(name="client") 4 ctx1abdc

26def get_client(mod: ModuleType): 4 ctx1abdc

27 client = TestClient(mod.app) 3 ctx1abc

28 

29 return client 3 ctx1abc

30 

31 

32def test_get(client: TestClient, html: str): 4 ctx1abdc

33 response = client.get("/") 3 ctx1hij

34 assert response.text == html 3 ctx1hij

35 

36 

37def test_websocket_handle_disconnection(client: TestClient): 4 ctx1abdc

38 with ( 1 ctx1e

39 client.websocket_connect("/ws/1234") as connection, 

40 client.websocket_connect("/ws/5678") as connection_two, 

41 ): 

42 connection.send_text("Hello from 1234") 3 ctx1fge

43 data1 = connection.receive_text() 3 ctx1fge

44 assert data1 == "You wrote: Hello from 1234" 3 ctx1fge

45 data2 = connection_two.receive_text() 3 ctx1fge

46 client1_says = "Client #1234 says: Hello from 1234" 3 ctx1fge

47 assert data2 == client1_says 3 ctx1fge

48 data1 = connection.receive_text() 3 ctx1fge

49 assert data1 == client1_says 3 ctx1fge

50 connection_two.close() 3 ctx1fge

51 data1 = connection.receive_text() 3 ctx1fge

52 assert data1 == "Client #5678 left the chat" 3 ctx1fge