Coverage for tests / test_serialize_response.py: 100%

1from fastapi import FastAPI 4 ctx1adbc

2from fastapi.testclient import TestClient 4 ctx1adbc

3from pydantic import BaseModel 4 ctx1adbc

4 

5app = FastAPI() 4 ctx1adbc

6 

7 

8class Item(BaseModel): 4 ctx1adbc

9 name: str 3 ctx1abc

10 price: float | None = None 4 ctx1adbc

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

12 

13 

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

15def get_valid(): 4 ctx1adbc

16 return {"name": "valid", "price": 1.0} 4 ctx1efgh

17 

18 

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

20def get_coerce(): 4 ctx1adbc

21 return {"name": "coerce", "price": "1.0"} 4 ctx1ijkl

22 

23 

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

25def get_validlist(): 4 ctx1adbc

26 return [ 4 ctx1mnop

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 ctx1adbc

34 

35 

36def test_valid(): 4 ctx1adbc

37 response = client.get("/items/valid") 4 ctx1efgh

38 response.raise_for_status() 4 ctx1efgh

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

40 

41 

42def test_coerce(): 4 ctx1adbc

43 response = client.get("/items/coerce") 4 ctx1ijkl

44 response.raise_for_status() 4 ctx1ijkl

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

46 

47 

48def test_validlist(): 4 ctx1adbc

49 response = client.get("/items/validlist") 4 ctx1mnop

50 response.raise_for_status() 4 ctx1mnop

51 assert response.json() == [ 4 ctx1mnop

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 ]