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

1from fastapi import FastAPI 5 ctx1abcde

2from fastapi.testclient import TestClient 5 ctx1abcde

3from fastapi.websockets import WebSocket 5 ctx1abcde

4 

5app = FastAPI() 5 ctx1abcde

6 

7 

8@app.get("/") 5 ctx1abcde

9async def read_main(): 5 ctx1abcde

10 return {"msg": "Hello World"} 10 ctx1pqrstuvwxy

11 

12 

13@app.websocket("/ws") 5 ctx1abcde

14async def websocket(websocket: WebSocket): 5 ctx1abcde

15 await websocket.accept() 10 ctx1fghijklmno

16 await websocket.send_json({"msg": "Hello WebSocket"}) 10 ctx1fghijklmno

17 await websocket.close() 10 ctx1fghijklmno

18 

19 

20def test_read_main(): 5 ctx1abcde

21 client = TestClient(app) 10 ctx1pqrstuvwxy

22 response = client.get("/") 10 ctx1pqrstuvwxy

23 assert response.status_code == 200 10 ctx1pqrstuvwxy

24 assert response.json() == {"msg": "Hello World"} 10 ctx1pqrstuvwxy

25 

26 

27def test_websocket(): 5 ctx1abcde

28 client = TestClient(app) 10 ctx1fghijklmno

29 with client.websocket_connect("/ws") as websocket: 10 ctx1fghijklmno

30 data = websocket.receive_json() 10 ctx1fghijklmno

31 assert data == {"msg": "Hello WebSocket"} 10 ctx1fghijklmno