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

1from contextlib import asynccontextmanager 4 ctx1hijk

2 

3from fastapi import FastAPI 4 ctx1hijk

4from fastapi.testclient import TestClient 4 ctx1hijk

5 

6items = {} 4 ctx1hijk

7 

8 

9@asynccontextmanager 4 ctx1hijk

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

11 items["foo"] = {"name": "Fighters"} 7 ctx1abcdefg

12 items["bar"] = {"name": "Tenders"} 7 ctx1abcdefg

13 yield 7 ctx1abcdefg

14 # clean up items 

15 items.clear() 7 ctx1abcdefg

16 

17 

18app = FastAPI(lifespan=lifespan) 4 ctx1hijk

19 

20 

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

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

23 return items[item_id] 7 ctx1abcdefg

24 

25 

26def test_read_items(): 4 ctx1hijk

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

28 assert items == {} 7 ctx1abcdefg

29 

30 with TestClient(app) as client: 7 ctx1abcdefg

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

32 assert items == {"foo": {"name": "Fighters"}, "bar": {"name": "Tenders"}} 7 ctx1abcdefg

33 

34 response = client.get("/items/foo") 7 ctx1abcdefg

35 assert response.status_code == 200 7 ctx1abcdefg

36 assert response.json() == {"name": "Fighters"} 7 ctx1abcdefg

37 

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

39 assert items == {"foo": {"name": "Fighters"}, "bar": {"name": "Tenders"}} 7 ctx1abcdefg

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 == {} 7 ctx1abcdefg