Coverage for tests/test_default_response_class_router.py: 100%

135 statements  

« prev     ^ index     » next       coverage.py v7.6.1, created at 2025-05-06 08:24 +0000

1from fastapi import APIRouter, FastAPI 1abcdef

2from fastapi.responses import HTMLResponse, JSONResponse, PlainTextResponse 1abcdef

3from fastapi.testclient import TestClient 1abcdef

4 

5 

6class OverrideResponse(JSONResponse): 1abcdef

7 media_type = "application/x-override" 1abcdef

8 

9 

10app = FastAPI() 1abcdef

11router_a = APIRouter() 1abcdef

12router_a_a = APIRouter() 1abcdef

13router_a_b_override = APIRouter() # Overrides default class 1abcdef

14router_b_override = APIRouter() # Overrides default class 1abcdef

15router_b_a = APIRouter() 1abcdef

16router_b_a_c_override = APIRouter() # Overrides default class again 1abcdef

17 

18 

19@app.get("/") 1abcdef

20def get_root(): 1abcdef

21 return {"msg": "Hello World"} 1ghijkl

22 

23 

24@app.get("/override", response_class=PlainTextResponse) 1abcdef

25def get_path_override(): 1abcdef

26 return "Hello World" 1mnopqr

27 

28 

29@router_a.get("/") 1abcdef

30def get_a(): 1abcdef

31 return {"msg": "Hello A"} 1stuvwx

32 

33 

34@router_a.get("/override", response_class=PlainTextResponse) 1abcdef

35def get_a_path_override(): 1abcdef

36 return "Hello A" 1yzABCD

37 

38 

39@router_a_a.get("/") 1abcdef

40def get_a_a(): 1abcdef

41 return {"msg": "Hello A A"} 1EFGHIJ

42 

43 

44@router_a_a.get("/override", response_class=PlainTextResponse) 1abcdef

45def get_a_a_path_override(): 1abcdef

46 return "Hello A A" 1KLMNOP

47 

48 

49@router_a_b_override.get("/") 1abcdef

50def get_a_b(): 1abcdef

51 return "Hello A B" 1QRSTUV

52 

53 

54@router_a_b_override.get("/override", response_class=HTMLResponse) 1abcdef

55def get_a_b_path_override(): 1abcdef

56 return "Hello A B" 1WXYZ01

57 

58 

59@router_b_override.get("/") 1abcdef

60def get_b(): 1abcdef

61 return "Hello B" 1234567

62 

63 

64@router_b_override.get("/override", response_class=HTMLResponse) 1abcdef

65def get_b_path_override(): 1abcdef

66 return "Hello B" 189!#$%

67 

68 

69@router_b_a.get("/") 1abcdef

70def get_b_a(): 1abcdef

71 return "Hello B A" 1'()*+,

72 

73 

74@router_b_a.get("/override", response_class=HTMLResponse) 1abcdef

75def get_b_a_path_override(): 1abcdef

76 return "Hello B A" 1-./:;=

77 

78 

79@router_b_a_c_override.get("/") 1abcdef

80def get_b_a_c(): 1abcdef

81 return "Hello B A C" 1?@[]^_

82 

83 

84@router_b_a_c_override.get("/override", response_class=OverrideResponse) 1abcdef

85def get_b_a_c_path_override(): 1abcdef

86 return {"msg": "Hello B A C"} 2` { | } ~ ab

87 

88 

89router_b_a.include_router( 1abcdef

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

91) 

92router_b_override.include_router(router_b_a, prefix="/a") 1abcdef

93router_a.include_router(router_a_a, prefix="/a") 1abcdef

94router_a.include_router( 1abcdef

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

96) 

97app.include_router(router_a, prefix="/a") 1abcdef

98app.include_router( 1abcdef

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

100) 

101 

102 

103client = TestClient(app) 1abcdef

104 

105json_type = "application/json" 1abcdef

106text_type = "text/plain; charset=utf-8" 1abcdef

107html_type = "text/html; charset=utf-8" 1abcdef

108override_type = "application/x-override" 1abcdef

109 

110 

111def test_app(): 1abcdef

112 with client: 1ghijkl

113 response = client.get("/") 1ghijkl

114 assert response.json() == {"msg": "Hello World"} 1ghijkl

115 assert response.headers["content-type"] == json_type 1ghijkl

116 

117 

118def test_app_override(): 1abcdef

119 with client: 1mnopqr

120 response = client.get("/override") 1mnopqr

121 assert response.content == b"Hello World" 1mnopqr

122 assert response.headers["content-type"] == text_type 1mnopqr

123 

124 

125def test_router_a(): 1abcdef

126 with client: 1stuvwx

127 response = client.get("/a") 1stuvwx

128 assert response.json() == {"msg": "Hello A"} 1stuvwx

129 assert response.headers["content-type"] == json_type 1stuvwx

130 

131 

132def test_router_a_override(): 1abcdef

133 with client: 1yzABCD

134 response = client.get("/a/override") 1yzABCD

135 assert response.content == b"Hello A" 1yzABCD

136 assert response.headers["content-type"] == text_type 1yzABCD

137 

138 

139def test_router_a_a(): 1abcdef

140 with client: 1EFGHIJ

141 response = client.get("/a/a") 1EFGHIJ

142 assert response.json() == {"msg": "Hello A A"} 1EFGHIJ

143 assert response.headers["content-type"] == json_type 1EFGHIJ

144 

145 

146def test_router_a_a_override(): 1abcdef

147 with client: 1KLMNOP

148 response = client.get("/a/a/override") 1KLMNOP

149 assert response.content == b"Hello A A" 1KLMNOP

150 assert response.headers["content-type"] == text_type 1KLMNOP

151 

152 

153def test_router_a_b(): 1abcdef

154 with client: 1QRSTUV

155 response = client.get("/a/b") 1QRSTUV

156 assert response.content == b"Hello A B" 1QRSTUV

157 assert response.headers["content-type"] == text_type 1QRSTUV

158 

159 

160def test_router_a_b_override(): 1abcdef

161 with client: 1WXYZ01

162 response = client.get("/a/b/override") 1WXYZ01

163 assert response.content == b"Hello A B" 1WXYZ01

164 assert response.headers["content-type"] == html_type 1WXYZ01

165 

166 

167def test_router_b(): 1abcdef

168 with client: 1234567

169 response = client.get("/b") 1234567

170 assert response.content == b"Hello B" 1234567

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

172 

173 

174def test_router_b_override(): 1abcdef

175 with client: 189!#$%

176 response = client.get("/b/override") 189!#$%

177 assert response.content == b"Hello B" 189!#$%

178 assert response.headers["content-type"] == html_type 189!#$%

179 

180 

181def test_router_b_a(): 1abcdef

182 with client: 1'()*+,

183 response = client.get("/b/a") 1'()*+,

184 assert response.content == b"Hello B A" 1'()*+,

185 assert response.headers["content-type"] == text_type 1'()*+,

186 

187 

188def test_router_b_a_override(): 1abcdef

189 with client: 1-./:;=

190 response = client.get("/b/a/override") 1-./:;=

191 assert response.content == b"Hello B A" 1-./:;=

192 assert response.headers["content-type"] == html_type 1-./:;=

193 

194 

195def test_router_b_a_c(): 1abcdef

196 with client: 1?@[]^_

197 response = client.get("/b/a/c") 1?@[]^_

198 assert response.content == b"Hello B A C" 1?@[]^_

199 assert response.headers["content-type"] == html_type 1?@[]^_

200 

201 

202def test_router_b_a_c_override(): 1abcdef

203 with client: 2` { | } ~ ab

204 response = client.get("/b/a/c/override") 2` { | } ~ ab

205 assert response.json() == {"msg": "Hello B A C"} 2` { | } ~ ab

206 assert response.headers["content-type"] == override_type 2` { | } ~ ab