Coverage for tests / test_serialize_response_model.py: 100%

1from fastapi import FastAPI 4 ctx1abcd

2from pydantic import BaseModel, Field 4 ctx1abcd

3from starlette.testclient import TestClient 4 ctx1abcd

4 

5app = FastAPI() 4 ctx1abcd

6 

7 

8class Item(BaseModel): 4 ctx1abcd

9 name: str = Field(alias="aliased_name") 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 Item(aliased_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 Item(aliased_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 Item(aliased_name="foo"), 

28 Item(aliased_name="bar", price=1.0), 

29 Item(aliased_name="baz", price=2.0, owner_ids=[1, 2, 3]), 

30 ] 

31 

32 

33@app.get("/items/validdict", response_model=dict[str, Item]) 4 ctx1abcd

34def get_validdict(): 4 ctx1abcd

35 return { 3 ctx1nop

36 "k1": Item(aliased_name="foo"), 

37 "k2": Item(aliased_name="bar", price=1.0), 

38 "k3": Item(aliased_name="baz", price=2.0, owner_ids=[1, 2, 3]), 

39 } 

40 

41 

42@app.get( 4 ctx1abcd

43 "/items/valid-exclude-unset", response_model=Item, response_model_exclude_unset=True 

44) 

45def get_valid_exclude_unset(): 4 ctx1abcd

46 return Item(aliased_name="valid", price=1.0) 3 ctx1qrs

47 

48 

49@app.get( 4 ctx1abcd

50 "/items/coerce-exclude-unset", 

51 response_model=Item, 

52 response_model_exclude_unset=True, 

53) 

54def get_coerce_exclude_unset(): 4 ctx1abcd

55 return Item(aliased_name="coerce", price="1.0") 3 ctx1tuv

56 

57 

58@app.get( 4 ctx1abcd

59 "/items/validlist-exclude-unset", 

60 response_model=list[Item], 

61 response_model_exclude_unset=True, 

62) 

63def get_validlist_exclude_unset(): 4 ctx1abcd

64 return [ 3 ctx1wxy

65 Item(aliased_name="foo"), 

66 Item(aliased_name="bar", price=1.0), 

67 Item(aliased_name="baz", price=2.0, owner_ids=[1, 2, 3]), 

68 ] 

69 

70 

71@app.get( 4 ctx1abcd

72 "/items/validdict-exclude-unset", 

73 response_model=dict[str, Item], 

74 response_model_exclude_unset=True, 

75) 

76def get_validdict_exclude_unset(): 4 ctx1abcd

77 return { 3 ctx1zAB

78 "k1": Item(aliased_name="foo"), 

79 "k2": Item(aliased_name="bar", price=1.0), 

80 "k3": Item(aliased_name="baz", price=2.0, owner_ids=[1, 2, 3]), 

81 } 

82 

83 

84client = TestClient(app) 4 ctx1abcd

85 

86 

87def test_valid(): 4 ctx1abcd

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

89 response.raise_for_status() 3 ctx1efg

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

91 

92 

93def test_coerce(): 4 ctx1abcd

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

95 response.raise_for_status() 3 ctx1hij

96 assert response.json() == { 3 ctx1hij

97 "aliased_name": "coerce", 

98 "price": 1.0, 

99 "owner_ids": None, 

100 } 

101 

102 

103def test_validlist(): 4 ctx1abcd

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

105 response.raise_for_status() 3 ctx1klm

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

107 {"aliased_name": "foo", "price": None, "owner_ids": None}, 

108 {"aliased_name": "bar", "price": 1.0, "owner_ids": None}, 

109 {"aliased_name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 

110 ] 

111 

112 

113def test_validdict(): 4 ctx1abcd

114 response = client.get("/items/validdict") 3 ctx1nop

115 response.raise_for_status() 3 ctx1nop

116 assert response.json() == { 3 ctx1nop

117 "k1": {"aliased_name": "foo", "price": None, "owner_ids": None}, 

118 "k2": {"aliased_name": "bar", "price": 1.0, "owner_ids": None}, 

119 "k3": {"aliased_name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 

120 } 

121 

122 

123def test_valid_exclude_unset(): 4 ctx1abcd

124 response = client.get("/items/valid-exclude-unset") 3 ctx1qrs

125 response.raise_for_status() 3 ctx1qrs

126 assert response.json() == {"aliased_name": "valid", "price": 1.0} 3 ctx1qrs

127 

128 

129def test_coerce_exclude_unset(): 4 ctx1abcd

130 response = client.get("/items/coerce-exclude-unset") 3 ctx1tuv

131 response.raise_for_status() 3 ctx1tuv

132 assert response.json() == {"aliased_name": "coerce", "price": 1.0} 3 ctx1tuv

133 

134 

135def test_validlist_exclude_unset(): 4 ctx1abcd

136 response = client.get("/items/validlist-exclude-unset") 3 ctx1wxy

137 response.raise_for_status() 3 ctx1wxy

138 assert response.json() == [ 3 ctx1wxy

139 {"aliased_name": "foo"}, 

140 {"aliased_name": "bar", "price": 1.0}, 

141 {"aliased_name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 

142 ] 

143 

144 

145def test_validdict_exclude_unset(): 4 ctx1abcd

146 response = client.get("/items/validdict-exclude-unset") 3 ctx1zAB

147 response.raise_for_status() 3 ctx1zAB

148 assert response.json() == { 3 ctx1zAB

149 "k1": {"aliased_name": "foo"}, 

150 "k2": {"aliased_name": "bar", "price": 1.0}, 

151 "k3": {"aliased_name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 

152 }