Coverage for tests / test_tutorial / test_request_files / test_tutorial002.py: 100%

1import importlib 4 ctx1abcd

2 

3import pytest 4 ctx1abcd

4from fastapi import FastAPI 4 ctx1abcd

5from fastapi.testclient import TestClient 4 ctx1abcd

6from inline_snapshot import snapshot 4 ctx1abcd

7 

8 

9@pytest.fixture( 4 ctx1abcd

10 name="app", 

11 params=[ 

12 "tutorial002_py310", 

13 "tutorial002_an_py310", 

14 ], 

15) 

16def get_app(request: pytest.FixtureRequest): 4 ctx1abcd

17 mod = importlib.import_module(f"docs_src.request_files.{request.param}") 38 ctx1,-uvwx./yz:;AB=?CDEF@[GH]^_`IJKL{|MN}~

18 

19 return mod.app 38 ctx1,-uvwx./yz:;AB=?CDEF@[GH]^_`IJKL{|MN}~

20 

21 

22@pytest.fixture(name="client") 4 ctx1abcd

23def get_client(app: FastAPI): 4 ctx1abcd

24 client = TestClient(app) 20 ctx1uvwxyzABCDEFGHIJKLMN

25 return client 20 ctx1uvwxyzABCDEFGHIJKLMN

26 

27 

28def test_post_form_no_body(client: TestClient): 4 ctx1abcd

29 response = client.post("/files/") 8 ctx1WXYZ0123

30 assert response.status_code == 422, response.text 8 ctx1WXYZ0123

31 assert response.json() == { 8 ctx1WXYZ0123

32 "detail": [ 

33 { 

34 "type": "missing", 

35 "loc": ["body", "files"], 

36 "msg": "Field required", 

37 "input": None, 

38 } 

39 ] 

40 } 

41 

42 

43def test_post_body_json(client: TestClient): 4 ctx1abcd

44 response = client.post("/files/", json={"file": "Foo"}) 8 ctx1456789!#

45 assert response.status_code == 422, response.text 8 ctx1456789!#

46 assert response.json() == { 8 ctx1456789!#

47 "detail": [ 

48 { 

49 "type": "missing", 

50 "loc": ["body", "files"], 

51 "msg": "Field required", 

52 "input": None, 

53 } 

54 ] 

55 } 

56 

57 

58def test_post_files(tmp_path, app: FastAPI): 4 ctx1abcd

59 path = tmp_path / "test.txt" 8 ctx1efghijkl

60 path.write_bytes(b"<file content>") 8 ctx1efghijkl

61 path2 = tmp_path / "test2.txt" 8 ctx1efghijkl

62 path2.write_bytes(b"<file content2>") 8 ctx1efghijkl

63 

64 client = TestClient(app) 8 ctx1efghijkl

65 with path.open("rb") as file, path2.open("rb") as file2: 8 ctx1efghijkl

66 response = client.post( 8 ctx1efghijkl

67 "/files/", 

68 files=( 

69 ("files", ("test.txt", file)), 

70 ("files", ("test2.txt", file2)), 

71 ), 

72 ) 

73 assert response.status_code == 200, response.text 8 ctx1efghijkl

74 assert response.json() == {"file_sizes": [14, 15]} 8 ctx1efghijkl

75 

76 

77def test_post_upload_file(tmp_path, app: FastAPI): 4 ctx1abcd

78 path = tmp_path / "test.txt" 8 ctx1mnopqrst

79 path.write_bytes(b"<file content>") 8 ctx1mnopqrst

80 path2 = tmp_path / "test2.txt" 8 ctx1mnopqrst

81 path2.write_bytes(b"<file content2>") 8 ctx1mnopqrst

82 

83 client = TestClient(app) 8 ctx1mnopqrst

84 with path.open("rb") as file, path2.open("rb") as file2: 8 ctx1mnopqrst

85 response = client.post( 8 ctx1mnopqrst

86 "/uploadfiles/", 

87 files=( 

88 ("files", ("test.txt", file)), 

89 ("files", ("test2.txt", file2)), 

90 ), 

91 ) 

92 assert response.status_code == 200, response.text 8 ctx1mnopqrst

93 assert response.json() == {"filenames": ["test.txt", "test2.txt"]} 8 ctx1mnopqrst

94 

95 

96def test_get_root(app: FastAPI): 4 ctx1abcd

97 client = TestClient(app) 8 ctx1OPQRSTUV

98 response = client.get("/") 8 ctx1OPQRSTUV

99 assert response.status_code == 200, response.text 8 ctx1OPQRSTUV

100 assert b"<form" in response.content 8 ctx1OPQRSTUV

101 

102 

103def test_openapi_schema(client: TestClient): 4 ctx1abcd

104 response = client.get("/openapi.json") 7 ctx1$%'()*+

105 assert response.status_code == 200, response.text 7 ctx1$%'()*+

106 assert response.json() == snapshot( 7 ctx1$%'()*+

107 { 

108 "openapi": "3.1.0", 

109 "info": {"title": "FastAPI", "version": "0.1.0"}, 

110 "paths": { 

111 "/files/": { 

112 "post": { 

113 "responses": { 

114 "200": { 

115 "description": "Successful Response", 

116 "content": {"application/json": {"schema": {}}}, 

117 }, 

118 "422": { 

119 "description": "Validation Error", 

120 "content": { 

121 "application/json": { 

122 "schema": { 

123 "$ref": "#/components/schemas/HTTPValidationError" 

124 } 

125 } 

126 }, 

127 }, 

128 }, 

129 "summary": "Create Files", 

130 "operationId": "create_files_files__post", 

131 "requestBody": { 

132 "content": { 

133 "multipart/form-data": { 

134 "schema": { 

135 "$ref": "#/components/schemas/Body_create_files_files__post" 

136 } 

137 } 

138 }, 

139 "required": True, 

140 }, 

141 } 

142 }, 

143 "/uploadfiles/": { 

144 "post": { 

145 "responses": { 

146 "200": { 

147 "description": "Successful Response", 

148 "content": {"application/json": {"schema": {}}}, 

149 }, 

150 "422": { 

151 "description": "Validation Error", 

152 "content": { 

153 "application/json": { 

154 "schema": { 

155 "$ref": "#/components/schemas/HTTPValidationError" 

156 } 

157 } 

158 }, 

159 }, 

160 }, 

161 "summary": "Create Upload Files", 

162 "operationId": "create_upload_files_uploadfiles__post", 

163 "requestBody": { 

164 "content": { 

165 "multipart/form-data": { 

166 "schema": { 

167 "$ref": "#/components/schemas/Body_create_upload_files_uploadfiles__post" 

168 } 

169 } 

170 }, 

171 "required": True, 

172 }, 

173 } 

174 }, 

175 "/": { 

176 "get": { 

177 "responses": { 

178 "200": { 

179 "description": "Successful Response", 

180 "content": {"application/json": {"schema": {}}}, 

181 } 

182 }, 

183 "summary": "Main", 

184 "operationId": "main__get", 

185 } 

186 }, 

187 }, 

188 "components": { 

189 "schemas": { 

190 "Body_create_upload_files_uploadfiles__post": { 

191 "title": "Body_create_upload_files_uploadfiles__post", 

192 "required": ["files"], 

193 "type": "object", 

194 "properties": { 

195 "files": { 

196 "title": "Files", 

197 "type": "array", 

198 "items": { 

199 "type": "string", 

200 "contentMediaType": "application/octet-stream", 

201 }, 

202 } 

203 }, 

204 }, 

205 "Body_create_files_files__post": { 

206 "title": "Body_create_files_files__post", 

207 "required": ["files"], 

208 "type": "object", 

209 "properties": { 

210 "files": { 

211 "title": "Files", 

212 "type": "array", 

213 "items": { 

214 "type": "string", 

215 "contentMediaType": "application/octet-stream", 

216 }, 

217 } 

218 }, 

219 }, 

220 "ValidationError": { 

221 "title": "ValidationError", 

222 "required": ["loc", "msg", "type"], 

223 "type": "object", 

224 "properties": { 

225 "loc": { 

226 "title": "Location", 

227 "type": "array", 

228 "items": { 

229 "anyOf": [{"type": "string"}, {"type": "integer"}] 

230 }, 

231 }, 

232 "msg": {"title": "Message", "type": "string"}, 

233 "type": {"title": "Error Type", "type": "string"}, 

234 "input": {"title": "Input"}, 

235 "ctx": {"title": "Context", "type": "object"}, 

236 }, 

237 }, 

238 "HTTPValidationError": { 

239 "title": "HTTPValidationError", 

240 "type": "object", 

241 "properties": { 

242 "detail": { 

243 "title": "Detail", 

244 "type": "array", 

245 "items": { 

246 "$ref": "#/components/schemas/ValidationError" 

247 }, 

248 } 

249 }, 

250 }, 

251 } 

252 }, 

253 } 

254 )