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) 15 ctx1fghijklmnopqrst

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 ctx1filor

32 

33 

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

35def get_path_override(): 5 ctx1abcde

36 return "Hello World" 5 ctx1uvwxy

37 

38 

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

40def get_a(): 5 ctx1abcde

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

42 

43 

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

45def get_a_path_override(): 5 ctx1abcde

46 return "Hello A" 5 ctx1zABCD

47 

48 

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

50def get_a_a(): 5 ctx1abcde

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

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 ctx1EFGHI

57 

58 

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

60def get_a_b(): 5 ctx1abcde

61 return "Hello A B" 5 ctx1JKLMN

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 ctx1OPQRS

67 

68 

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

70def get_b(): 5 ctx1abcde

71 return "Hello B" 5 ctx1TUVWX

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 ctx1YZ012

77 

78 

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

80def get_b_a(): 5 ctx1abcde

81 return "Hello B A" 5 ctx134567

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 ctx189!#$

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 ctx1%'()*

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 ctx1+,-./

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 ctx1filor

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

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

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

126 

127 

128def test_app_override(): 5 ctx1abcde

129 with client: 5 ctx1uvwxy

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

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

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

133 

134 

135def test_router_a(): 5 ctx1abcde

136 with client: 5 ctx1gjmps

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

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

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

140 

141 

142def test_router_a_override(): 5 ctx1abcde

143 with client: 5 ctx1zABCD

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

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

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

147 

148 

149def test_router_a_a(): 5 ctx1abcde

150 with client: 5 ctx1hknqt

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

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

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

154 

155 

156def test_router_a_a_override(): 5 ctx1abcde

157 with client: 5 ctx1EFGHI

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

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

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

161 

162 

163def test_router_a_b(): 5 ctx1abcde

164 with client: 5 ctx1JKLMN

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

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

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

168 

169 

170def test_router_a_b_override(): 5 ctx1abcde

171 with client: 5 ctx1OPQRS

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

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

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

175 

176 

177def test_router_b(): 5 ctx1abcde

178 with client: 5 ctx1TUVWX

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

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

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

182 

183 

184def test_router_b_override(): 5 ctx1abcde

185 with client: 5 ctx1YZ012

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

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

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

189 

190 

191def test_router_b_a(): 5 ctx1abcde

192 with client: 5 ctx134567

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

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

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

196 

197 

198def test_router_b_a_override(): 5 ctx1abcde

199 with client: 5 ctx189!#$

200 response = client.get("/b/a/override") 5 ctx189!#$

201 assert response.content == b"Hello B A" 5 ctx189!#$

202 assert response.headers["content-type"] == html_type 5 ctx189!#$

203 

204 

205def test_router_b_a_c(): 5 ctx1abcde

206 with client: 5 ctx1%'()*

207 response = client.get("/b/a/c") 5 ctx1%'()*

208 assert response.content == b"Hello B A C" 5 ctx1%'()*

209 assert response.headers["content-type"] == html_type 5 ctx1%'()*

210 

211 

212def test_router_b_a_c_override(): 5 ctx1abcde

213 with client: 5 ctx1+,-./

214 response = client.get("/b/a/c/override") 5 ctx1+,-./

215 assert response.json() == {"msg": "Hello B A C"} 5 ctx1+,-./

216 assert response.headers["content-type"] == override_type 5 ctx1+,-./