OLD | NEW |
1 // Copyright (c) 2015 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/chromeos/policy/upload_job.h" | 5 #include "chrome/browser/chromeos/policy/upload_job.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 | 8 |
9 #include <queue> | 9 #include <queue> |
10 #include <set> | 10 #include <set> |
(...skipping 155 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
166 void OnFailure(UploadJob::ErrorCode error_code) override { | 166 void OnFailure(UploadJob::ErrorCode error_code) override { |
167 if (expected_error_ && *expected_error_.get() == error_code) | 167 if (expected_error_ && *expected_error_.get() == error_code) |
168 run_loop_.Quit(); | 168 run_loop_.Quit(); |
169 else | 169 else |
170 FAIL(); | 170 FAIL(); |
171 } | 171 } |
172 | 172 |
173 const GURL GetServerURL() const { return test_server_.GetURL(kUploadPath); } | 173 const GURL GetServerURL() const { return test_server_.GetURL(kUploadPath); } |
174 | 174 |
175 void SetExpectedError(scoped_ptr<UploadJob::ErrorCode> expected_error) { | 175 void SetExpectedError(scoped_ptr<UploadJob::ErrorCode> expected_error) { |
176 expected_error_ = expected_error.Pass(); | 176 expected_error_ = std::move(expected_error); |
177 } | 177 } |
178 | 178 |
179 // testing::Test: | 179 // testing::Test: |
180 void SetUp() override { | 180 void SetUp() override { |
181 request_context_getter_ = new net::TestURLRequestContextGetter( | 181 request_context_getter_ = new net::TestURLRequestContextGetter( |
182 base::ThreadTaskRunnerHandle::Get()); | 182 base::ThreadTaskRunnerHandle::Get()); |
183 oauth2_service_.AddAccount("robot@gmail.com"); | 183 oauth2_service_.AddAccount("robot@gmail.com"); |
184 ASSERT_TRUE(test_server_.Start()); | 184 ASSERT_TRUE(test_server_.Start()); |
185 } | 185 } |
186 | 186 |
187 // testing::Test: | 187 // testing::Test: |
188 void TearDown() override { | 188 void TearDown() override { |
189 ASSERT_TRUE(test_server_.ShutdownAndWaitUntilComplete()); | 189 ASSERT_TRUE(test_server_.ShutdownAndWaitUntilComplete()); |
190 } | 190 } |
191 | 191 |
192 protected: | 192 protected: |
193 scoped_ptr<UploadJob> PrepareUploadJob(scoped_ptr< | 193 scoped_ptr<UploadJob> PrepareUploadJob(scoped_ptr< |
194 UploadJobImpl::MimeBoundaryGenerator> mime_boundary_generator) { | 194 UploadJobImpl::MimeBoundaryGenerator> mime_boundary_generator) { |
195 scoped_ptr<UploadJob> upload_job(new UploadJobImpl( | 195 scoped_ptr<UploadJob> upload_job( |
196 GetServerURL(), kRobotAccountId, &oauth2_service_, | 196 new UploadJobImpl(GetServerURL(), kRobotAccountId, &oauth2_service_, |
197 request_context_getter_.get(), this, mime_boundary_generator.Pass())); | 197 request_context_getter_.get(), this, |
| 198 std::move(mime_boundary_generator))); |
198 | 199 |
199 std::map<std::string, std::string> header_entries; | 200 std::map<std::string, std::string> header_entries; |
200 header_entries.insert(std::make_pair(kCustomField1, "CUSTOM1")); | 201 header_entries.insert(std::make_pair(kCustomField1, "CUSTOM1")); |
201 scoped_ptr<std::string> data(new std::string(kTestPayload1)); | 202 scoped_ptr<std::string> data(new std::string(kTestPayload1)); |
202 upload_job->AddDataSegment("Name1", "file1.ext", header_entries, | 203 upload_job->AddDataSegment("Name1", "file1.ext", header_entries, |
203 data.Pass()); | 204 std::move(data)); |
204 | 205 |
205 header_entries.insert(std::make_pair(kCustomField2, "CUSTOM2")); | 206 header_entries.insert(std::make_pair(kCustomField2, "CUSTOM2")); |
206 scoped_ptr<std::string> data2(new std::string(kTestPayload2)); | 207 scoped_ptr<std::string> data2(new std::string(kTestPayload2)); |
207 upload_job->AddDataSegment("Name2", "", header_entries, data2.Pass()); | 208 upload_job->AddDataSegment("Name2", "", header_entries, std::move(data2)); |
208 return upload_job.Pass(); | 209 return upload_job; |
209 } | 210 } |
210 | 211 |
211 content::TestBrowserThreadBundle test_browser_thread_bundle_; | 212 content::TestBrowserThreadBundle test_browser_thread_bundle_; |
212 base::RunLoop run_loop_; | 213 base::RunLoop run_loop_; |
213 net::EmbeddedTestServer test_server_; | 214 net::EmbeddedTestServer test_server_; |
214 scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_; | 215 scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_; |
215 MockOAuth2TokenService oauth2_service_; | 216 MockOAuth2TokenService oauth2_service_; |
216 | 217 |
217 scoped_ptr<UploadJob::ErrorCode> expected_error_; | 218 scoped_ptr<UploadJob::ErrorCode> expected_error_; |
218 }; | 219 }; |
(...skipping 12 matching lines...) Expand all Loading... |
231 scoped_ptr<net::test_server::HttpResponse> HandlePostRequest( | 232 scoped_ptr<net::test_server::HttpResponse> HandlePostRequest( |
232 const net::test_server::HttpRequest& request) { | 233 const net::test_server::HttpRequest& request) { |
233 EXPECT_TRUE(request.headers.find("Authorization") != request.headers.end()); | 234 EXPECT_TRUE(request.headers.find("Authorization") != request.headers.end()); |
234 const std::string authorization_header = | 235 const std::string authorization_header = |
235 request.headers.at("Authorization"); | 236 request.headers.at("Authorization"); |
236 scoped_ptr<net::test_server::BasicHttpResponse> response( | 237 scoped_ptr<net::test_server::BasicHttpResponse> response( |
237 new net::test_server::BasicHttpResponse); | 238 new net::test_server::BasicHttpResponse); |
238 const size_t pos = authorization_header.find(" "); | 239 const size_t pos = authorization_header.find(" "); |
239 if (pos == std::string::npos) { | 240 if (pos == std::string::npos) { |
240 response->set_code(net::HTTP_UNAUTHORIZED); | 241 response->set_code(net::HTTP_UNAUTHORIZED); |
241 return response.Pass(); | 242 return std::move(response); |
242 } | 243 } |
243 | 244 |
244 const std::string token = authorization_header.substr(pos + 1); | 245 const std::string token = authorization_header.substr(pos + 1); |
245 response->set_code(oauth2_service_.IsTokenValid(token) | 246 response->set_code(oauth2_service_.IsTokenValid(token) |
246 ? net::HTTP_OK | 247 ? net::HTTP_OK |
247 : net::HTTP_UNAUTHORIZED); | 248 : net::HTTP_UNAUTHORIZED); |
248 return response.Pass(); | 249 return std::move(response); |
249 } | 250 } |
250 }; | 251 }; |
251 | 252 |
252 TEST_F(UploadFlowTest, SuccessfulUpload) { | 253 TEST_F(UploadFlowTest, SuccessfulUpload) { |
253 oauth2_service_.SetTokenValid(kTokenValid); | 254 oauth2_service_.SetTokenValid(kTokenValid); |
254 oauth2_service_.AddTokenToQueue(kTokenValid); | 255 oauth2_service_.AddTokenToQueue(kTokenValid); |
255 scoped_ptr<UploadJob> upload_job = PrepareUploadJob( | 256 scoped_ptr<UploadJob> upload_job = PrepareUploadJob( |
256 make_scoped_ptr(new UploadJobImpl::RandomMimeBoundaryGenerator)); | 257 make_scoped_ptr(new UploadJobImpl::RandomMimeBoundaryGenerator)); |
257 upload_job->Start(); | 258 upload_job->Start(); |
258 run_loop_.Run(); | 259 run_loop_.Run(); |
(...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
300 test_server_.RegisterRequestHandler(base::Bind( | 301 test_server_.RegisterRequestHandler(base::Bind( |
301 &UploadRequestTest::HandlePostRequest, base::Unretained(this))); | 302 &UploadRequestTest::HandlePostRequest, base::Unretained(this))); |
302 } | 303 } |
303 | 304 |
304 scoped_ptr<net::test_server::HttpResponse> HandlePostRequest( | 305 scoped_ptr<net::test_server::HttpResponse> HandlePostRequest( |
305 const net::test_server::HttpRequest& request) { | 306 const net::test_server::HttpRequest& request) { |
306 scoped_ptr<net::test_server::BasicHttpResponse> response( | 307 scoped_ptr<net::test_server::BasicHttpResponse> response( |
307 new net::test_server::BasicHttpResponse); | 308 new net::test_server::BasicHttpResponse); |
308 response->set_code(net::HTTP_OK); | 309 response->set_code(net::HTTP_OK); |
309 EXPECT_EQ(expected_content_, request.content); | 310 EXPECT_EQ(expected_content_, request.content); |
310 return response.Pass(); | 311 return std::move(response); |
311 } | 312 } |
312 | 313 |
313 void SetExpectedRequestContent(const std::string& expected_content) { | 314 void SetExpectedRequestContent(const std::string& expected_content) { |
314 expected_content_ = expected_content; | 315 expected_content_ = expected_content; |
315 } | 316 } |
316 | 317 |
317 protected: | 318 protected: |
318 std::string expected_content_; | 319 std::string expected_content_; |
319 }; | 320 }; |
320 | 321 |
(...skipping 15 matching lines...) Expand all Loading... |
336 "customfield2: CUSTOM2\r\n" | 337 "customfield2: CUSTOM2\r\n" |
337 "\r\n" | 338 "\r\n" |
338 "**||--||PAYLOAD2||--||**\r\n--" | 339 "**||--||PAYLOAD2||--||**\r\n--" |
339 "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA--\r\n"); | 340 "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA--\r\n"); |
340 | 341 |
341 upload_job->Start(); | 342 upload_job->Start(); |
342 run_loop_.Run(); | 343 run_loop_.Run(); |
343 } | 344 } |
344 | 345 |
345 } // namespace policy | 346 } // namespace policy |
OLD | NEW |