OLD | NEW |
| (Empty) |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "base/message_loop/message_loop.h" | |
6 #include "base/test/test_simple_task_runner.h" | |
7 #include "chrome/browser/drive/drive_api_service.h" | |
8 #include "google_apis/drive/dummy_auth_service.h" | |
9 #include "google_apis/drive/request_sender.h" | |
10 #include "google_apis/drive/test_util.h" | |
11 #include "net/url_request/url_request_test_util.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 | |
14 namespace drive { | |
15 namespace { | |
16 const char kTestUserAgent[] = "test-user-agent"; | |
17 } | |
18 | |
19 class TestAuthService : public google_apis::DummyAuthService { | |
20 public: | |
21 void StartAuthentication( | |
22 const google_apis::AuthStatusCallback& callback) override { | |
23 callback_ = callback; | |
24 } | |
25 | |
26 bool HasAccessToken() const override { return false; } | |
27 | |
28 void SendHttpError() { | |
29 ASSERT_FALSE(callback_.is_null()); | |
30 callback_.Run(google_apis::HTTP_UNAUTHORIZED, ""); | |
31 } | |
32 | |
33 private: | |
34 google_apis::AuthStatusCallback callback_; | |
35 }; | |
36 | |
37 TEST(DriveAPIServiceTest, BatchRequestConfiguratorWithAuthFailure) { | |
38 const GURL test_base_url("http://localhost/"); | |
39 google_apis::DriveApiUrlGenerator url_generator(test_base_url, test_base_url); | |
40 scoped_refptr<base::TestSimpleTaskRunner> task_runner = | |
41 new base::TestSimpleTaskRunner(); | |
42 scoped_refptr<net::TestURLRequestContextGetter> request_context_getter = | |
43 new net::TestURLRequestContextGetter(task_runner.get()); | |
44 google_apis::RequestSender sender(new TestAuthService, | |
45 request_context_getter.get(), | |
46 task_runner.get(), kTestUserAgent); | |
47 google_apis::drive::BatchUploadRequest* const request = | |
48 new google_apis::drive::BatchUploadRequest(&sender, url_generator); | |
49 sender.StartRequestWithAuthRetry(request); | |
50 BatchRequestConfigurator configurator( | |
51 request->GetWeakPtrAsBatchUploadRequest(), task_runner.get(), | |
52 url_generator, google_apis::CancelCallback()); | |
53 static_cast<TestAuthService*>(sender.auth_service())->SendHttpError(); | |
54 | |
55 { | |
56 google_apis::DriveApiErrorCode error = google_apis::HTTP_SUCCESS; | |
57 scoped_ptr<google_apis::FileResource> file_resource; | |
58 configurator.MultipartUploadNewFile( | |
59 "text/plain", 10, "", "title", | |
60 base::FilePath(FILE_PATH_LITERAL("/file")), UploadNewFileOptions(), | |
61 google_apis::test_util::CreateCopyResultCallback(&error, | |
62 &file_resource), | |
63 google_apis::ProgressCallback()); | |
64 EXPECT_EQ(google_apis::DRIVE_OTHER_ERROR, error); | |
65 } | |
66 { | |
67 google_apis::DriveApiErrorCode error = google_apis::HTTP_SUCCESS; | |
68 scoped_ptr<google_apis::FileResource> file_resource; | |
69 configurator.MultipartUploadExistingFile( | |
70 "text/plain", 10, "resource_id", | |
71 base::FilePath(FILE_PATH_LITERAL("/file")), UploadExistingFileOptions(), | |
72 google_apis::test_util::CreateCopyResultCallback(&error, | |
73 &file_resource), | |
74 google_apis::ProgressCallback()); | |
75 EXPECT_EQ(google_apis::DRIVE_OTHER_ERROR, error); | |
76 } | |
77 } | |
78 | |
79 } // namespace drive | |
OLD | NEW |