Coverage for tests/test_serialize_response_model.py: 100%

1from typing import Dict, List, Optional 6 ctx1abcdef

2 

3from fastapi import FastAPI 6 ctx1abcdef

4from pydantic import BaseModel, Field 6 ctx1abcdef

5from starlette.testclient import TestClient 6 ctx1abcdef

6 

7app = FastAPI() 6 ctx1abcdef

8 

9 

10class Item(BaseModel): 6 ctx1abcdef

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

36def get_validdict(): 6 ctx1abcdef

37 return { 6 ctx1yzABCD

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

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

46) 

47def get_valid_exclude_unset(): 6 ctx1abcdef

48 return Item(aliased_name="valid", price=1.0) 6 ctx1EFGHIJ

49 

50 

51@app.get( 6 ctx1abcdef

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

53 response_model=Item, 

54 response_model_exclude_unset=True, 

55) 

56def get_coerce_exclude_unset(): 6 ctx1abcdef

57 return Item(aliased_name="coerce", price="1.0") 6 ctx1KLMNOP

58 

59 

60@app.get( 6 ctx1abcdef

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

62 response_model=List[Item], 

63 response_model_exclude_unset=True, 

64) 

65def get_validlist_exclude_unset(): 6 ctx1abcdef

66 return [ 6 ctx1QRSTUV

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

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

75 response_model=Dict[str, Item], 

76 response_model_exclude_unset=True, 

77) 

78def get_validdict_exclude_unset(): 6 ctx1abcdef

79 return { 6 ctx1WXYZ01

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

87 

88 

89def test_valid(): 6 ctx1abcdef

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

91 response.raise_for_status() 6 ctx1ghijkl

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

93 

94 

95def test_coerce(): 6 ctx1abcdef

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

97 response.raise_for_status() 6 ctx1mnopqr

98 assert response.json() == { 6 ctx1mnopqr

99 "aliased_name": "coerce", 

100 "price": 1.0, 

101 "owner_ids": None, 

102 } 

103 

104 

105def test_validlist(): 6 ctx1abcdef

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

107 response.raise_for_status() 6 ctx1stuvwx

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

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

116 response = client.get("/items/validdict") 6 ctx1yzABCD

117 response.raise_for_status() 6 ctx1yzABCD

118 assert response.json() == { 6 ctx1yzABCD

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

126 response = client.get("/items/valid-exclude-unset") 6 ctx1EFGHIJ

127 response.raise_for_status() 6 ctx1EFGHIJ

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

129 

130 

131def test_coerce_exclude_unset(): 6 ctx1abcdef

132 response = client.get("/items/coerce-exclude-unset") 6 ctx1KLMNOP

133 response.raise_for_status() 6 ctx1KLMNOP

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

135 

136 

137def test_validlist_exclude_unset(): 6 ctx1abcdef

138 response = client.get("/items/validlist-exclude-unset") 6 ctx1QRSTUV

139 response.raise_for_status() 6 ctx1QRSTUV

140 assert response.json() == [ 6 ctx1QRSTUV

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

148 response = client.get("/items/validdict-exclude-unset") 6 ctx1WXYZ01

149 response.raise_for_status() 6 ctx1WXYZ01

150 assert response.json() == { 6 ctx1WXYZ01

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 }