Coverage for tests/test_serialize_response.py: 100%

1from typing import List, Optional 6 ctx1abcdef

2 

3from fastapi import FastAPI 6 ctx1abcdef

4from fastapi.testclient import TestClient 6 ctx1abcdef

5from pydantic import BaseModel 6 ctx1abcdef

6 

7app = FastAPI() 6 ctx1abcdef

8 

9 

10class Item(BaseModel): 6 ctx1abcdef

11 name: str 6 ctx1abcdef

12 price: Optional[float] = None 6 ctx1abcdef

13 owner_ids: Optional[List[int]] = None 6 ctx1abcdef

14 

15 

16@app.get("/items/valid", response_model=Item) 6 ctx1abcdef

17def get_valid(): 6 ctx1abcdef

18 return {"name": "valid", "price": 1.0} 6 ctx1ghijkl

19 

20 

21@app.get("/items/coerce", response_model=Item) 6 ctx1abcdef

22def get_coerce(): 6 ctx1abcdef

23 return {"name": "coerce", "price": "1.0"} 6 ctx1mnopqr

24 

25 

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

27def get_validlist(): 6 ctx1abcdef

28 return [ 6 ctx1stuvwx

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) 6 ctx1abcdef

36 

37 

38def test_valid(): 6 ctx1abcdef

39 response = client.get("/items/valid") 6 ctx1ghijkl

40 response.raise_for_status() 6 ctx1ghijkl

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

42 

43 

44def test_coerce(): 6 ctx1abcdef

45 response = client.get("/items/coerce") 6 ctx1mnopqr

46 response.raise_for_status() 6 ctx1mnopqr

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

48 

49 

50def test_validlist(): 6 ctx1abcdef

51 response = client.get("/items/validlist") 6 ctx1stuvwx

52 response.raise_for_status() 6 ctx1stuvwx

53 assert response.json() == [ 6 ctx1stuvwx

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 ]