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

1from contextlib import asynccontextmanager 4 ctx1ghij

2 

3from fastapi import FastAPI 4 ctx1ghij

4from fastapi.testclient import TestClient 4 ctx1ghij

5 

6items = {} 4 ctx1ghij

7 

8 

9@asynccontextmanager 4 ctx1ghij

10async def lifespan(app: FastAPI): 4 ctx1ghij

11 items["foo"] = {"name": "Fighters"} 6 ctx1abcdef

12 items["bar"] = {"name": "Tenders"} 6 ctx1abcdef

13 yield 6 ctx1abcdef

14 # clean up items 

15 items.clear() 6 ctx1abcdef

16 

17 

18app = FastAPI(lifespan=lifespan) 4 ctx1ghij

19 

20 

21@app.get("/items/{item_id}") 4 ctx1ghij

22async def read_items(item_id: str): 4 ctx1ghij

23 return items[item_id] 6 ctx1abcdef

24 

25 

26def test_read_items(): 4 ctx1ghij

27 # Before the lifespan starts, "items" is still empty 

28 assert items == {} 6 ctx1abcdef

29 

30 with TestClient(app) as client: 6 ctx1abcdef

31 # Inside the "with TestClient" block, the lifespan starts and items added 

32 assert items == {"foo": {"name": "Fighters"}, "bar": {"name": "Tenders"}} 6 ctx1abcdef

33 

34 response = client.get("/items/foo") 6 ctx1abcdef

35 assert response.status_code == 200 6 ctx1abcdef

36 assert response.json() == {"name": "Fighters"} 6 ctx1abcdef

37 

38 # After the requests is done, the items are still there 

39 assert items == {"foo": {"name": "Fighters"}, "bar": {"name": "Tenders"}} 6 ctx1abcdef

40 

41 # The end of the "with TestClient" block simulates terminating the app, so 

42 # the lifespan ends and items are cleaned up 

43 assert items == {} 6 ctx1abcdef