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

1from typing import Union 7 ctx1abcdefg

2 

3from fastapi import FastAPI, Header, HTTPException 7 ctx1abcdefg

4from pydantic import BaseModel 7 ctx1abcdefg

5from typing_extensions import Annotated 7 ctx1abcdefg

6 

7fake_secret_token = "coneofsilence" 7 ctx1abcdefg

8 

9fake_db = { 7 ctx1abcdefg

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

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

12} 

13 

14app = FastAPI() 7 ctx1abcdefg

15 

16 

17class Item(BaseModel): 7 ctx1abcdefg

18 id: str 7 ctx1abcdefg

19 title: str 7 ctx1abcdefg

20 description: Union[str, None] = None 7 ctx1abcdefg

21 

22 

23@app.get("/items/{item_id}", response_model=Item) 7 ctx1abcdefg

24async def read_main(item_id: str, x_token: Annotated[str, Header()]): 7 ctx1abcdefg

25 if x_token != fake_secret_token: 7 ctx1hijklmn

26 raise HTTPException(status_code=400, detail="Invalid X-Token header") 7 ctx1hijklmn

27 if item_id not in fake_db: 7 ctx1hijklmn

28 raise HTTPException(status_code=404, detail="Item not found") 7 ctx1hijklmn

29 return fake_db[item_id] 7 ctx1hijklmn

30 

31 

32@app.post("/items/", response_model=Item) 7 ctx1abcdefg

33async def create_item(item: Item, x_token: Annotated[str, Header()]): 7 ctx1abcdefg

34 if x_token != fake_secret_token: 7 ctx1hijklmn

35 raise HTTPException(status_code=400, detail="Invalid X-Token header") 7 ctx1hijklmn

36 if item.id in fake_db: 7 ctx1hijklmn

37 raise HTTPException(status_code=409, detail="Item already exists") 7 ctx1hijklmn

38 fake_db[item.id] = item 7 ctx1hijklmn

39 return item 7 ctx1hijklmn