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 ctx1abcde

22 

23 

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

25def get_path_override(): 5 ctx1abcde

26 return "Hello World" 5 ctx1abcde

27 

28 

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

30def get_a(): 5 ctx1abcde

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

32 

33 

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

35def get_a_path_override(): 5 ctx1abcde

36 return "Hello A" 5 ctx1abcde

37 

38 

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

40def get_a_a(): 5 ctx1abcde

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

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 ctx1abcde

47 

48 

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

50def get_a_b(): 5 ctx1abcde

51 return "Hello A B" 5 ctx1abcde

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 ctx1abcde

57 

58 

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

60def get_b(): 5 ctx1abcde

61 return "Hello B" 5 ctx1abcde

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 ctx1abcde

67 

68 

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

70def get_b_a(): 5 ctx1abcde

71 return "Hello B A" 5 ctx1abcde

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 ctx1abcde

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 ctx1abcde

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 ctx1abcde

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 ctx1abcde

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

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

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

116 

117 

118def test_app_override(): 5 ctx1abcde

119 with client: 5 ctx1abcde

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

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

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

123 

124 

125def test_router_a(): 5 ctx1abcde

126 with client: 5 ctx1abcde

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

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

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

130 

131 

132def test_router_a_override(): 5 ctx1abcde

133 with client: 5 ctx1abcde

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

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

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

137 

138 

139def test_router_a_a(): 5 ctx1abcde

140 with client: 5 ctx1abcde

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

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

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

144 

145 

146def test_router_a_a_override(): 5 ctx1abcde

147 with client: 5 ctx1abcde

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

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

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

151 

152 

153def test_router_a_b(): 5 ctx1abcde

154 with client: 5 ctx1abcde

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

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

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

158 

159 

160def test_router_a_b_override(): 5 ctx1abcde

161 with client: 5 ctx1abcde

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

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

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

165 

166 

167def test_router_b(): 5 ctx1abcde

168 with client: 5 ctx1abcde

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

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

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

172 

173 

174def test_router_b_override(): 5 ctx1abcde

175 with client: 5 ctx1abcde

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

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

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

179 

180 

181def test_router_b_a(): 5 ctx1abcde

182 with client: 5 ctx1abcde

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

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

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

186 

187 

188def test_router_b_a_override(): 5 ctx1abcde

189 with client: 5 ctx1abcde

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

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

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

193 

194 

195def test_router_b_a_c(): 5 ctx1abcde

196 with client: 5 ctx1abcde

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

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

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

200 

201 

202def test_router_b_a_c_override(): 5 ctx1abcde

203 with client: 5 ctx1abcde

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

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

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