Coverage for docs_src/app_testing/app_b_py310/main.py: 100%

1from fastapi import FastAPI, Header, HTTPException 4 ctx1abcd

2from pydantic import BaseModel 4 ctx1abcd

3 

4fake_secret_token = "coneofsilence" 4 ctx1abcd

5 

6fake_db = { 4 ctx1abcd

7 "foo": {"id": "foo", "title": "Foo", "description": "There goes my hero"}, 

8 "bar": {"id": "bar", "title": "Bar", "description": "The bartenders"}, 

9} 

10 

11app = FastAPI() 4 ctx1abcd

12 

13 

14class Item(BaseModel): 4 ctx1abcd

15 id: str 4 ctx1abcd

16 title: str 4 ctx1abcd

17 description: str | None = None 4 ctx1abcd

18 

19 

20@app.get("/items/{item_id}", response_model=Item) 4 ctx1abcd

21async def read_main(item_id: str, x_token: str = Header()): 4 ctx1abcd

22 if x_token != fake_secret_token: 4 ctx1efgh

23 raise HTTPException(status_code=400, detail="Invalid X-Token header") 4 ctx1efgh

24 if item_id not in fake_db: 4 ctx1efgh

25 raise HTTPException(status_code=404, detail="Item not found") 4 ctx1efgh

26 return fake_db[item_id] 4 ctx1efgh

27 

28 

29@app.post("/items/", response_model=Item) 4 ctx1abcd

30async def create_item(item: Item, x_token: str = Header()): 4 ctx1abcd

31 if x_token != fake_secret_token: 4 ctx1efgh

32 raise HTTPException(status_code=400, detail="Invalid X-Token header") 4 ctx1efgh

33 if item.id in fake_db: 4 ctx1efgh

34 raise HTTPException(status_code=409, detail="Item already exists") 4 ctx1efgh

35 fake_db[item.id] = item 4 ctx1efgh

36 return item 4 ctx1efgh