Coverage for tests/test_default_response_class.py: 100%

1from typing import Any 5 ctx1abcde

2 

3import orjson 5 ctx1abcde

4from fastapi import APIRouter, FastAPI 5 ctx1abcde

5from fastapi.responses import HTMLResponse, JSONResponse, PlainTextResponse 5 ctx1abcde

6from fastapi.testclient import TestClient 5 ctx1abcde

7 

8 

9class ORJSONResponse(JSONResponse): 5 ctx1abcde

10 media_type = "application/x-orjson" 5 ctx1abcde

11 

12 def render(self, content: Any) -> bytes: 5 ctx1abcde

13 return orjson.dumps(content) 5 ctx1abcde

14 

15 

16class OverrideResponse(JSONResponse): 5 ctx1abcde

17 media_type = "application/x-override" 5 ctx1abcde

18 

19 

20app = FastAPI(default_response_class=ORJSONResponse) 5 ctx1abcde

21router_a = APIRouter() 5 ctx1abcde

22router_a_a = APIRouter() 5 ctx1abcde

23router_a_b_override = APIRouter() # Overrides default class 5 ctx1abcde

24router_b_override = APIRouter() # Overrides default class 5 ctx1abcde

25router_b_a = APIRouter() 5 ctx1abcde

26router_b_a_c_override = APIRouter() # Overrides default class again 5 ctx1abcde

27 

28 

29@app.get("/") 5 ctx1abcde

30def get_root(): 5 ctx1abcde

31 return {"msg": "Hello World"} 5 ctx1abcde

32 

33 

34@app.get("/override", response_class=PlainTextResponse) 5 ctx1abcde

35def get_path_override(): 5 ctx1abcde

36 return "Hello World" 5 ctx1abcde

37 

38 

39@router_a.get("/") 5 ctx1abcde

40def get_a(): 5 ctx1abcde

41 return {"msg": "Hello A"} 5 ctx1abcde

42 

43 

44@router_a.get("/override", response_class=PlainTextResponse) 5 ctx1abcde

45def get_a_path_override(): 5 ctx1abcde

46 return "Hello A" 5 ctx1abcde

47 

48 

49@router_a_a.get("/") 5 ctx1abcde

50def get_a_a(): 5 ctx1abcde

51 return {"msg": "Hello A A"} 5 ctx1abcde

52 

53 

54@router_a_a.get("/override", response_class=PlainTextResponse) 5 ctx1abcde

55def get_a_a_path_override(): 5 ctx1abcde

56 return "Hello A A" 5 ctx1abcde

57 

58 

59@router_a_b_override.get("/") 5 ctx1abcde

60def get_a_b(): 5 ctx1abcde

61 return "Hello A B" 5 ctx1abcde

62 

63 

64@router_a_b_override.get("/override", response_class=HTMLResponse) 5 ctx1abcde

65def get_a_b_path_override(): 5 ctx1abcde

66 return "Hello A B" 5 ctx1abcde

67 

68 

69@router_b_override.get("/") 5 ctx1abcde

70def get_b(): 5 ctx1abcde

71 return "Hello B" 5 ctx1abcde

72 

73 

74@router_b_override.get("/override", response_class=HTMLResponse) 5 ctx1abcde

75def get_b_path_override(): 5 ctx1abcde

76 return "Hello B" 5 ctx1abcde

77 

78 

79@router_b_a.get("/") 5 ctx1abcde

80def get_b_a(): 5 ctx1abcde

81 return "Hello B A" 5 ctx1abcde

82 

83 

84@router_b_a.get("/override", response_class=HTMLResponse) 5 ctx1abcde

85def get_b_a_path_override(): 5 ctx1abcde

86 return "Hello B A" 5 ctx1abcde

87 

88 

89@router_b_a_c_override.get("/") 5 ctx1abcde

90def get_b_a_c(): 5 ctx1abcde

91 return "Hello B A C" 5 ctx1abcde

92 

93 

94@router_b_a_c_override.get("/override", response_class=OverrideResponse) 5 ctx1abcde

95def get_b_a_c_path_override(): 5 ctx1abcde

96 return {"msg": "Hello B A C"} 5 ctx1abcde

97 

98 

99router_b_a.include_router( 5 ctx1abcde

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

101) 

102router_b_override.include_router(router_b_a, prefix="/a") 5 ctx1abcde

103router_a.include_router(router_a_a, prefix="/a") 5 ctx1abcde

104router_a.include_router( 5 ctx1abcde

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

106) 

107app.include_router(router_a, prefix="/a") 5 ctx1abcde

108app.include_router( 5 ctx1abcde

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

110) 

111 

112 

113client = TestClient(app) 5 ctx1abcde

114 

115orjson_type = "application/x-orjson" 5 ctx1abcde

116text_type = "text/plain; charset=utf-8" 5 ctx1abcde

117html_type = "text/html; charset=utf-8" 5 ctx1abcde

118override_type = "application/x-override" 5 ctx1abcde

119 

120 

121def test_app(): 5 ctx1abcde

122 with client: 5 ctx1abcde

123 response = client.get("/") 5 ctx1abcde

124 assert response.json() == {"msg": "Hello World"} 5 ctx1abcde

125 assert response.headers["content-type"] == orjson_type 5 ctx1abcde

126 

127 

128def test_app_override(): 5 ctx1abcde

129 with client: 5 ctx1abcde

130 response = client.get("/override") 5 ctx1abcde

131 assert response.content == b"Hello World" 5 ctx1abcde

132 assert response.headers["content-type"] == text_type 5 ctx1abcde

133 

134 

135def test_router_a(): 5 ctx1abcde

136 with client: 5 ctx1abcde

137 response = client.get("/a") 5 ctx1abcde

138 assert response.json() == {"msg": "Hello A"} 5 ctx1abcde

139 assert response.headers["content-type"] == orjson_type 5 ctx1abcde

140 

141 

142def test_router_a_override(): 5 ctx1abcde

143 with client: 5 ctx1abcde

144 response = client.get("/a/override") 5 ctx1abcde

145 assert response.content == b"Hello A" 5 ctx1abcde

146 assert response.headers["content-type"] == text_type 5 ctx1abcde

147 

148 

149def test_router_a_a(): 5 ctx1abcde

150 with client: 5 ctx1abcde

151 response = client.get("/a/a") 5 ctx1abcde

152 assert response.json() == {"msg": "Hello A A"} 5 ctx1abcde

153 assert response.headers["content-type"] == orjson_type 5 ctx1abcde

154 

155 

156def test_router_a_a_override(): 5 ctx1abcde

157 with client: 5 ctx1abcde

158 response = client.get("/a/a/override") 5 ctx1abcde

159 assert response.content == b"Hello A A" 5 ctx1abcde

160 assert response.headers["content-type"] == text_type 5 ctx1abcde

161 

162 

163def test_router_a_b(): 5 ctx1abcde

164 with client: 5 ctx1abcde

165 response = client.get("/a/b") 5 ctx1abcde

166 assert response.content == b"Hello A B" 5 ctx1abcde

167 assert response.headers["content-type"] == text_type 5 ctx1abcde

168 

169 

170def test_router_a_b_override(): 5 ctx1abcde

171 with client: 5 ctx1abcde

172 response = client.get("/a/b/override") 5 ctx1abcde

173 assert response.content == b"Hello A B" 5 ctx1abcde

174 assert response.headers["content-type"] == html_type 5 ctx1abcde

175 

176 

177def test_router_b(): 5 ctx1abcde

178 with client: 5 ctx1abcde

179 response = client.get("/b") 5 ctx1abcde

180 assert response.content == b"Hello B" 5 ctx1abcde

181 assert response.headers["content-type"] == text_type 5 ctx1abcde

182 

183 

184def test_router_b_override(): 5 ctx1abcde

185 with client: 5 ctx1abcde

186 response = client.get("/b/override") 5 ctx1abcde

187 assert response.content == b"Hello B" 5 ctx1abcde

188 assert response.headers["content-type"] == html_type 5 ctx1abcde

189 

190 

191def test_router_b_a(): 5 ctx1abcde

192 with client: 5 ctx1abcde

193 response = client.get("/b/a") 5 ctx1abcde

194 assert response.content == b"Hello B A" 5 ctx1abcde

195 assert response.headers["content-type"] == text_type 5 ctx1abcde

196 

197 

198def test_router_b_a_override(): 5 ctx1abcde

199 with client: 5 ctx1abcde

200 response = client.get("/b/a/override") 5 ctx1abcde

201 assert response.content == b"Hello B A" 5 ctx1abcde

202 assert response.headers["content-type"] == html_type 5 ctx1abcde

203 

204 

205def test_router_b_a_c(): 5 ctx1abcde

206 with client: 5 ctx1abcde

207 response = client.get("/b/a/c") 5 ctx1abcde

208 assert response.content == b"Hello B A C" 5 ctx1abcde

209 assert response.headers["content-type"] == html_type 5 ctx1abcde

210 

211 

212def test_router_b_a_c_override(): 5 ctx1abcde

213 with client: 5 ctx1abcde

214 response = client.get("/b/a/c/override") 5 ctx1abcde

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

216 assert response.headers["content-type"] == override_type 5 ctx1abcde