Coverage for tests/test_default_response_class_router.py: 100%

1from fastapi import APIRouter, FastAPI 5 ctx1abcde

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

3from fastapi.testclient import TestClient 5 ctx1abcde

4 

5 

6class OverrideResponse(JSONResponse): 5 ctx1abcde

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

8 

9 

10app = FastAPI() 5 ctx1abcde

11router_a = APIRouter() 5 ctx1abcde

12router_a_a = APIRouter() 5 ctx1abcde

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

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

15router_b_a = APIRouter() 5 ctx1abcde

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

17 

18 

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

20def get_root(): 5 ctx1abcde

21 return {"msg": "Hello World"} 5 ctx1fghij

22 

23 

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

25def get_path_override(): 5 ctx1abcde

26 return "Hello World" 5 ctx1klmno

27 

28 

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

30def get_a(): 5 ctx1abcde

31 return {"msg": "Hello A"} 5 ctx1pqrst

32 

33 

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

35def get_a_path_override(): 5 ctx1abcde

36 return "Hello A" 5 ctx1uvwxy

37 

38 

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

40def get_a_a(): 5 ctx1abcde

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

42 

43 

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

45def get_a_a_path_override(): 5 ctx1abcde

46 return "Hello A A" 5 ctx1EFGHI

47 

48 

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

50def get_a_b(): 5 ctx1abcde

51 return "Hello A B" 5 ctx1JKLMN

52 

53 

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

55def get_a_b_path_override(): 5 ctx1abcde

56 return "Hello A B" 5 ctx1OPQRS

57 

58 

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

60def get_b(): 5 ctx1abcde

61 return "Hello B" 5 ctx1TUVWX

62 

63 

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

65def get_b_path_override(): 5 ctx1abcde

66 return "Hello B" 5 ctx1YZ012

67 

68 

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

70def get_b_a(): 5 ctx1abcde

71 return "Hello B A" 5 ctx134567

72 

73 

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

75def get_b_a_path_override(): 5 ctx1abcde

76 return "Hello B A" 5 ctx189!#$

77 

78 

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

80def get_b_a_c(): 5 ctx1abcde

81 return "Hello B A C" 5 ctx1%'()*

82 

83 

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

85def get_b_a_c_path_override(): 5 ctx1abcde

86 return {"msg": "Hello B A C"} 5 ctx1+,-./

87 

88 

89router_b_a.include_router( 5 ctx1abcde

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

91) 

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

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

94router_a.include_router( 5 ctx1abcde

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

96) 

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

98app.include_router( 5 ctx1abcde

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

100) 

101 

102 

103client = TestClient(app) 5 ctx1abcde

104 

105json_type = "application/json" 5 ctx1abcde

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

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

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

109 

110 

111def test_app(): 5 ctx1abcde

112 with client: 5 ctx1fghij

113 response = client.get("/") 5 ctx1fghij

114 assert response.json() == {"msg": "Hello World"} 5 ctx1fghij

115 assert response.headers["content-type"] == json_type 5 ctx1fghij

116 

117 

118def test_app_override(): 5 ctx1abcde

119 with client: 5 ctx1klmno

120 response = client.get("/override") 5 ctx1klmno

121 assert response.content == b"Hello World" 5 ctx1klmno

122 assert response.headers["content-type"] == text_type 5 ctx1klmno

123 

124 

125def test_router_a(): 5 ctx1abcde

126 with client: 5 ctx1pqrst

127 response = client.get("/a") 5 ctx1pqrst

128 assert response.json() == {"msg": "Hello A"} 5 ctx1pqrst

129 assert response.headers["content-type"] == json_type 5 ctx1pqrst

130 

131 

132def test_router_a_override(): 5 ctx1abcde

133 with client: 5 ctx1uvwxy

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

135 assert response.content == b"Hello A" 5 ctx1uvwxy

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

137 

138 

139def test_router_a_a(): 5 ctx1abcde

140 with client: 5 ctx1zABCD

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

142 assert response.json() == {"msg": "Hello A A"} 5 ctx1zABCD

143 assert response.headers["content-type"] == json_type 5 ctx1zABCD

144 

145 

146def test_router_a_a_override(): 5 ctx1abcde

147 with client: 5 ctx1EFGHI

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

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

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

151 

152 

153def test_router_a_b(): 5 ctx1abcde

154 with client: 5 ctx1JKLMN

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

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

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

158 

159 

160def test_router_a_b_override(): 5 ctx1abcde

161 with client: 5 ctx1OPQRS

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

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

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

165 

166 

167def test_router_b(): 5 ctx1abcde

168 with client: 5 ctx1TUVWX

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

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

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

172 

173 

174def test_router_b_override(): 5 ctx1abcde

175 with client: 5 ctx1YZ012

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

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

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

179 

180 

181def test_router_b_a(): 5 ctx1abcde

182 with client: 5 ctx134567

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

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

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

186 

187 

188def test_router_b_a_override(): 5 ctx1abcde

189 with client: 5 ctx189!#$

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

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

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

193 

194 

195def test_router_b_a_c(): 5 ctx1abcde

196 with client: 5 ctx1%'()*

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

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

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

200 

201 

202def test_router_b_a_c_override(): 5 ctx1abcde

203 with client: 5 ctx1+,-./

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

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

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