Coverage for tests / test_serialize_response.py: 100%

1from typing import Optional 4 ctx1abcd

2 

3from fastapi import FastAPI 4 ctx1abcd

4from fastapi.testclient import TestClient 4 ctx1abcd

5from pydantic import BaseModel 4 ctx1abcd

6 

7app = FastAPI() 4 ctx1abcd

8 

9 

10class Item(BaseModel): 4 ctx1abcd

11 name: str 4 ctx1abcd

12 price: Optional[float] = None 4 ctx1abcd

13 owner_ids: Optional[list[int]] = None 4 ctx1abcd

14 

15 

16@app.get("/items/valid", response_model=Item) 4 ctx1abcd

17def get_valid(): 4 ctx1abcd

18 return {"name": "valid", "price": 1.0} 3 ctx1efg

19 

20 

21@app.get("/items/coerce", response_model=Item) 4 ctx1abcd

22def get_coerce(): 4 ctx1abcd

23 return {"name": "coerce", "price": "1.0"} 3 ctx1hij

24 

25 

26@app.get("/items/validlist", response_model=list[Item]) 4 ctx1abcd

27def get_validlist(): 4 ctx1abcd

28 return [ 3 ctx1klm

29 {"name": "foo"}, 

30 {"name": "bar", "price": 1.0}, 

31 {"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 

32 ] 

33 

34 

35client = TestClient(app) 4 ctx1abcd

36 

37 

38def test_valid(): 4 ctx1abcd

39 response = client.get("/items/valid") 3 ctx1efg

40 response.raise_for_status() 3 ctx1efg

41 assert response.json() == {"name": "valid", "price": 1.0, "owner_ids": None} 3 ctx1efg

42 

43 

44def test_coerce(): 4 ctx1abcd

45 response = client.get("/items/coerce") 3 ctx1hij

46 response.raise_for_status() 3 ctx1hij

47 assert response.json() == {"name": "coerce", "price": 1.0, "owner_ids": None} 3 ctx1hij

48 

49 

50def test_validlist(): 4 ctx1abcd

51 response = client.get("/items/validlist") 3 ctx1klm

52 response.raise_for_status() 3 ctx1klm

53 assert response.json() == [ 3 ctx1klm

54 {"name": "foo", "price": None, "owner_ids": None}, 

55 {"name": "bar", "price": 1.0, "owner_ids": None}, 

56 {"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 

57 ]