Coverage for tests/test_serialize_response_model.py: 100%

1from typing import Dict, List, Optional 5 ctx1abcde

2 

3from fastapi import FastAPI 5 ctx1abcde

4from pydantic import BaseModel, Field 5 ctx1abcde

5from starlette.testclient import TestClient 5 ctx1abcde

6 

7app = FastAPI() 5 ctx1abcde

8 

9 

10class Item(BaseModel): 5 ctx1abcde

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

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

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

32 ] 

33 

34 

35@app.get("/items/validdict", response_model=Dict[str, Item]) 5 ctx1abcde

36def get_validdict(): 5 ctx1abcde

37 return { 5 ctx1abcde

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

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

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

41 } 

42 

43 

44@app.get( 5 ctx1abcde

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

46) 

47def get_valid_exclude_unset(): 5 ctx1abcde

48 return Item(aliased_name="valid", price=1.0) 5 ctx1abcde

49 

50 

51@app.get( 5 ctx1abcde

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

53 response_model=Item, 

54 response_model_exclude_unset=True, 

55) 

56def get_coerce_exclude_unset(): 5 ctx1abcde

57 return Item(aliased_name="coerce", price="1.0") 5 ctx1abcde

58 

59 

60@app.get( 5 ctx1abcde

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

62 response_model=List[Item], 

63 response_model_exclude_unset=True, 

64) 

65def get_validlist_exclude_unset(): 5 ctx1abcde

66 return [ 5 ctx1abcde

67 Item(aliased_name="foo"), 

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

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

70 ] 

71 

72 

73@app.get( 5 ctx1abcde

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

75 response_model=Dict[str, Item], 

76 response_model_exclude_unset=True, 

77) 

78def get_validdict_exclude_unset(): 5 ctx1abcde

79 return { 5 ctx1abcde

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

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

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

83 } 

84 

85 

86client = TestClient(app) 5 ctx1abcde

87 

88 

89def test_valid(): 5 ctx1abcde

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

91 response.raise_for_status() 5 ctx1abcde

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

93 

94 

95def test_coerce(): 5 ctx1abcde

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

97 response.raise_for_status() 5 ctx1abcde

98 assert response.json() == { 5 ctx1abcde

99 "aliased_name": "coerce", 

100 "price": 1.0, 

101 "owner_ids": None, 

102 } 

103 

104 

105def test_validlist(): 5 ctx1abcde

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

107 response.raise_for_status() 5 ctx1abcde

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

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

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

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

112 ] 

113 

114 

115def test_validdict(): 5 ctx1abcde

116 response = client.get("/items/validdict") 5 ctx1abcde

117 response.raise_for_status() 5 ctx1abcde

118 assert response.json() == { 5 ctx1abcde

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

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

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

122 } 

123 

124 

125def test_valid_exclude_unset(): 5 ctx1abcde

126 response = client.get("/items/valid-exclude-unset") 5 ctx1abcde

127 response.raise_for_status() 5 ctx1abcde

128 assert response.json() == {"aliased_name": "valid", "price": 1.0} 5 ctx1abcde

129 

130 

131def test_coerce_exclude_unset(): 5 ctx1abcde

132 response = client.get("/items/coerce-exclude-unset") 5 ctx1abcde

133 response.raise_for_status() 5 ctx1abcde

134 assert response.json() == {"aliased_name": "coerce", "price": 1.0} 5 ctx1abcde

135 

136 

137def test_validlist_exclude_unset(): 5 ctx1abcde

138 response = client.get("/items/validlist-exclude-unset") 5 ctx1abcde

139 response.raise_for_status() 5 ctx1abcde

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

141 {"aliased_name": "foo"}, 

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

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

144 ] 

145 

146 

147def test_validdict_exclude_unset(): 5 ctx1abcde

148 response = client.get("/items/validdict-exclude-unset") 5 ctx1abcde

149 response.raise_for_status() 5 ctx1abcde

150 assert response.json() == { 5 ctx1abcde

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

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

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

154 }