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

1from typing import Annotated 4 ctx1adbc

2 

3from fastapi import FastAPI, Header, HTTPException 4 ctx1adbc

4from pydantic import BaseModel 4 ctx1adbc

5 

6fake_secret_token = "coneofsilence" 4 ctx1adbc

7 

8fake_db = { 4 ctx1adbc

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

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

11} 

12 

13app = FastAPI() 4 ctx1adbc

14 

15 

16class Item(BaseModel): 4 ctx1adbc

17 id: str 3 ctx1abc

18 title: str 3 ctx1abc

19 description: str | None = None 4 ctx1adbc

20 

21 

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

23async def read_main(item_id: str, x_token: Annotated[str, Header()]): 4 ctx1adbc

24 if x_token != fake_secret_token: 4 ctx1efgh

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

26 if item_id not in fake_db: 4 ctx1efgh

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

28 return fake_db[item_id] 4 ctx1efgh

29 

30 

31@app.post("/items/") 4 ctx1adbc

32async def create_item(item: Item, x_token: Annotated[str, Header()]) -> Item: 4 ctx1adbc

33 if x_token != fake_secret_token: 4 ctx1efgh

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

35 if item.id in fake_db: 4 ctx1efgh

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

37 fake_db[item.id] = item.model_dump() 4 ctx1efgh

38 return item 4 ctx1efgh