Coverage for docs_src / app_testing / tutorial002_py310.py: 100%

1from fastapi import FastAPI 4 ctx1abcd

2from fastapi.testclient import TestClient 4 ctx1abcd

3from fastapi.websockets import WebSocket 4 ctx1abcd

4 

5app = FastAPI() 4 ctx1abcd

6 

7 

8@app.get("/") 4 ctx1abcd

9async def read_main(): 4 ctx1abcd

10 return {"msg": "Hello World"} 7 ctx1lmnopqr

11 

12 

13@app.websocket("/ws") 4 ctx1abcd

14async def websocket(websocket: WebSocket): 4 ctx1abcd

15 await websocket.accept() 7 ctx1efghijk

16 await websocket.send_json({"msg": "Hello WebSocket"}) 7 ctx1efghijk

17 await websocket.close() 7 ctx1efghijk

18 

19 

20def test_read_main(): 4 ctx1abcd

21 client = TestClient(app) 7 ctx1lmnopqr

22 response = client.get("/") 7 ctx1lmnopqr

23 assert response.status_code == 200 7 ctx1lmnopqr

24 assert response.json() == {"msg": "Hello World"} 7 ctx1lmnopqr

25 

26 

27def test_websocket(): 4 ctx1abcd

28 client = TestClient(app) 7 ctx1efghijk

29 with client.websocket_connect("/ws") as websocket: 7 ctx1efghijk

30 data = websocket.receive_json() 7 ctx1efghijk

31 assert data == {"msg": "Hello WebSocket"} 7 ctx1efghijk