Coverage for tests / test_default_response_class.py: 100%

1from typing import Any 4 ctx1abcd

2 

3import orjson 4 ctx1abcd

4from fastapi import APIRouter, FastAPI 4 ctx1abcd

5from fastapi.responses import HTMLResponse, JSONResponse, PlainTextResponse 4 ctx1abcd

6from fastapi.testclient import TestClient 4 ctx1abcd

7 

8 

9class ORJSONResponse(JSONResponse): 4 ctx1abcd

10 media_type = "application/x-orjson" 4 ctx1abcd

11 

12 def render(self, content: Any) -> bytes: 4 ctx1abcd

13 return orjson.dumps(content) 9 ctx1efghijklm

14 

15 

16class OverrideResponse(JSONResponse): 4 ctx1abcd

17 media_type = "application/x-override" 4 ctx1abcd

18 

19 

20app = FastAPI(default_response_class=ORJSONResponse) 4 ctx1abcd

21router_a = APIRouter() 4 ctx1abcd

22router_a_a = APIRouter() 4 ctx1abcd

23router_a_b_override = APIRouter() # Overrides default class 4 ctx1abcd

24router_b_override = APIRouter() # Overrides default class 4 ctx1abcd

25router_b_a = APIRouter() 4 ctx1abcd

26router_b_a_c_override = APIRouter() # Overrides default class again 4 ctx1abcd

27 

28 

29@app.get("/") 4 ctx1abcd

30def get_root(): 4 ctx1abcd

31 return {"msg": "Hello World"} 3 ctx1ehk

32 

33 

34@app.get("/override", response_class=PlainTextResponse) 4 ctx1abcd

35def get_path_override(): 4 ctx1abcd

36 return "Hello World" 3 ctx1nop

37 

38 

39@router_a.get("/") 4 ctx1abcd

40def get_a(): 4 ctx1abcd

41 return {"msg": "Hello A"} 3 ctx1fil

42 

43 

44@router_a.get("/override", response_class=PlainTextResponse) 4 ctx1abcd

45def get_a_path_override(): 4 ctx1abcd

46 return "Hello A" 3 ctx1qrs

47 

48 

49@router_a_a.get("/") 4 ctx1abcd

50def get_a_a(): 4 ctx1abcd

51 return {"msg": "Hello A A"} 3 ctx1gjm

52 

53 

54@router_a_a.get("/override", response_class=PlainTextResponse) 4 ctx1abcd

55def get_a_a_path_override(): 4 ctx1abcd

56 return "Hello A A" 3 ctx1tuv

57 

58 

59@router_a_b_override.get("/") 4 ctx1abcd

60def get_a_b(): 4 ctx1abcd

61 return "Hello A B" 3 ctx1wxy

62 

63 

64@router_a_b_override.get("/override", response_class=HTMLResponse) 4 ctx1abcd

65def get_a_b_path_override(): 4 ctx1abcd

66 return "Hello A B" 3 ctx1zAB

67 

68 

69@router_b_override.get("/") 4 ctx1abcd

70def get_b(): 4 ctx1abcd

71 return "Hello B" 3 ctx1CDE

72 

73 

74@router_b_override.get("/override", response_class=HTMLResponse) 4 ctx1abcd

75def get_b_path_override(): 4 ctx1abcd

76 return "Hello B" 3 ctx1FGH

77 

78 

79@router_b_a.get("/") 4 ctx1abcd

80def get_b_a(): 4 ctx1abcd

81 return "Hello B A" 3 ctx1IJK

82 

83 

84@router_b_a.get("/override", response_class=HTMLResponse) 4 ctx1abcd

85def get_b_a_path_override(): 4 ctx1abcd

86 return "Hello B A" 3 ctx1LMN

87 

88 

89@router_b_a_c_override.get("/") 4 ctx1abcd

90def get_b_a_c(): 4 ctx1abcd

91 return "Hello B A C" 3 ctx1OPQ

92 

93 

94@router_b_a_c_override.get("/override", response_class=OverrideResponse) 4 ctx1abcd

95def get_b_a_c_path_override(): 4 ctx1abcd

96 return {"msg": "Hello B A C"} 3 ctx1RST

97 

98 

99router_b_a.include_router( 4 ctx1abcd

100 router_b_a_c_override, prefix="/c", default_response_class=HTMLResponse 

101) 

102router_b_override.include_router(router_b_a, prefix="/a") 4 ctx1abcd

103router_a.include_router(router_a_a, prefix="/a") 4 ctx1abcd

104router_a.include_router( 4 ctx1abcd

105 router_a_b_override, prefix="/b", default_response_class=PlainTextResponse 

106) 

107app.include_router(router_a, prefix="/a") 4 ctx1abcd

108app.include_router( 4 ctx1abcd

109 router_b_override, prefix="/b", default_response_class=PlainTextResponse 

110) 

111 

112 

113client = TestClient(app) 4 ctx1abcd

114 

115orjson_type = "application/x-orjson" 4 ctx1abcd

116text_type = "text/plain; charset=utf-8" 4 ctx1abcd

117html_type = "text/html; charset=utf-8" 4 ctx1abcd

118override_type = "application/x-override" 4 ctx1abcd

119 

120 

121def test_app(): 4 ctx1abcd

122 with client: 3 ctx1ehk

123 response = client.get("/") 3 ctx1ehk

124 assert response.json() == {"msg": "Hello World"} 3 ctx1ehk

125 assert response.headers["content-type"] == orjson_type 3 ctx1ehk

126 

127 

128def test_app_override(): 4 ctx1abcd

129 with client: 3 ctx1nop

130 response = client.get("/override") 3 ctx1nop

131 assert response.content == b"Hello World" 3 ctx1nop

132 assert response.headers["content-type"] == text_type 3 ctx1nop

133 

134 

135def test_router_a(): 4 ctx1abcd

136 with client: 3 ctx1fil

137 response = client.get("/a") 3 ctx1fil

138 assert response.json() == {"msg": "Hello A"} 3 ctx1fil

139 assert response.headers["content-type"] == orjson_type 3 ctx1fil

140 

141 

142def test_router_a_override(): 4 ctx1abcd

143 with client: 3 ctx1qrs

144 response = client.get("/a/override") 3 ctx1qrs

145 assert response.content == b"Hello A" 3 ctx1qrs

146 assert response.headers["content-type"] == text_type 3 ctx1qrs

147 

148 

149def test_router_a_a(): 4 ctx1abcd

150 with client: 3 ctx1gjm

151 response = client.get("/a/a") 3 ctx1gjm

152 assert response.json() == {"msg": "Hello A A"} 3 ctx1gjm

153 assert response.headers["content-type"] == orjson_type 3 ctx1gjm

154 

155 

156def test_router_a_a_override(): 4 ctx1abcd

157 with client: 3 ctx1tuv

158 response = client.get("/a/a/override") 3 ctx1tuv

159 assert response.content == b"Hello A A" 3 ctx1tuv

160 assert response.headers["content-type"] == text_type 3 ctx1tuv

161 

162 

163def test_router_a_b(): 4 ctx1abcd

164 with client: 3 ctx1wxy

165 response = client.get("/a/b") 3 ctx1wxy

166 assert response.content == b"Hello A B" 3 ctx1wxy

167 assert response.headers["content-type"] == text_type 3 ctx1wxy

168 

169 

170def test_router_a_b_override(): 4 ctx1abcd

171 with client: 3 ctx1zAB

172 response = client.get("/a/b/override") 3 ctx1zAB

173 assert response.content == b"Hello A B" 3 ctx1zAB

174 assert response.headers["content-type"] == html_type 3 ctx1zAB

175 

176 

177def test_router_b(): 4 ctx1abcd

178 with client: 3 ctx1CDE

179 response = client.get("/b") 3 ctx1CDE

180 assert response.content == b"Hello B" 3 ctx1CDE

181 assert response.headers["content-type"] == text_type 3 ctx1CDE

182 

183 

184def test_router_b_override(): 4 ctx1abcd

185 with client: 3 ctx1FGH

186 response = client.get("/b/override") 3 ctx1FGH

187 assert response.content == b"Hello B" 3 ctx1FGH

188 assert response.headers["content-type"] == html_type 3 ctx1FGH

189 

190 

191def test_router_b_a(): 4 ctx1abcd

192 with client: 3 ctx1IJK

193 response = client.get("/b/a") 3 ctx1IJK

194 assert response.content == b"Hello B A" 3 ctx1IJK

195 assert response.headers["content-type"] == text_type 3 ctx1IJK

196 

197 

198def test_router_b_a_override(): 4 ctx1abcd

199 with client: 3 ctx1LMN

200 response = client.get("/b/a/override") 3 ctx1LMN

201 assert response.content == b"Hello B A" 3 ctx1LMN

202 assert response.headers["content-type"] == html_type 3 ctx1LMN

203 

204 

205def test_router_b_a_c(): 4 ctx1abcd

206 with client: 3 ctx1OPQ

207 response = client.get("/b/a/c") 3 ctx1OPQ

208 assert response.content == b"Hello B A C" 3 ctx1OPQ

209 assert response.headers["content-type"] == html_type 3 ctx1OPQ

210 

211 

212def test_router_b_a_c_override(): 4 ctx1abcd

213 with client: 3 ctx1RST

214 response = client.get("/b/a/c/override") 3 ctx1RST

215 assert response.json() == {"msg": "Hello B A C"} 3 ctx1RST

216 assert response.headers["content-type"] == override_type 3 ctx1RST