Coverage for tests / test_serialize_response.py: 100%

1from fastapi import FastAPI 4 ctx1abcd

2from fastapi.testclient import TestClient 4 ctx1abcd

3from pydantic import BaseModel 4 ctx1abcd

4 

5app = FastAPI() 4 ctx1abcd

6 

7 

8class Item(BaseModel): 4 ctx1abcd

9 name: str 4 ctx1abcd

10 price: float | None = None 4 ctx1abcd

11 owner_ids: list[int] | None = None 4 ctx1abcd

12 

13 

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

15def get_valid(): 4 ctx1abcd

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

17 

18 

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

20def get_coerce(): 4 ctx1abcd

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

22 

23 

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

25def get_validlist(): 4 ctx1abcd

26 return [ 3 ctx1klm

27 {"name": "foo"}, 

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

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

30 ] 

31 

32 

33client = TestClient(app) 4 ctx1abcd

34 

35 

36def test_valid(): 4 ctx1abcd

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

38 response.raise_for_status() 3 ctx1efg

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

40 

41 

42def test_coerce(): 4 ctx1abcd

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

44 response.raise_for_status() 3 ctx1hij

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

46 

47 

48def test_validlist(): 4 ctx1abcd

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

50 response.raise_for_status() 3 ctx1klm

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

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

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

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

55 ]