Coverage for tests/test_serialize_response.py: 100%

1from typing import List, Optional 5 ctx1abcde

2 

3from fastapi import FastAPI 5 ctx1abcde

4from fastapi.testclient import TestClient 5 ctx1abcde

5from pydantic import BaseModel 5 ctx1abcde

6 

7app = FastAPI() 5 ctx1abcde

8 

9 

10class Item(BaseModel): 5 ctx1abcde

11 name: str 5 ctx1abcde

12 price: Optional[float] = None 5 ctx1abcde

13 owner_ids: Optional[List[int]] = None 5 ctx1abcde

14 

15 

16@app.get("/items/valid", response_model=Item) 5 ctx1abcde

17def get_valid(): 5 ctx1abcde

18 return {"name": "valid", "price": 1.0} 5 ctx1abcde

19 

20 

21@app.get("/items/coerce", response_model=Item) 5 ctx1abcde

22def get_coerce(): 5 ctx1abcde

23 return {"name": "coerce", "price": "1.0"} 5 ctx1abcde

24 

25 

26@app.get("/items/validlist", response_model=List[Item]) 5 ctx1abcde

27def get_validlist(): 5 ctx1abcde

28 return [ 5 ctx1abcde

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) 5 ctx1abcde

36 

37 

38def test_valid(): 5 ctx1abcde

39 response = client.get("/items/valid") 5 ctx1abcde

40 response.raise_for_status() 5 ctx1abcde

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

42 

43 

44def test_coerce(): 5 ctx1abcde

45 response = client.get("/items/coerce") 5 ctx1abcde

46 response.raise_for_status() 5 ctx1abcde

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

48 

49 

50def test_validlist(): 5 ctx1abcde

51 response = client.get("/items/validlist") 5 ctx1abcde

52 response.raise_for_status() 5 ctx1abcde

53 assert response.json() == [ 5 ctx1abcde

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 ]