| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "sync/internal_api/public/attachments/attachment_downloader.h" | 5 #include "sync/internal_api/public/attachments/attachment_downloader.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/memory/weak_ptr.h" | 8 #include "base/memory/weak_ptr.h" |
| 9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
| (...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 92 const ScopeSet& scopes, | 92 const ScopeSet& scopes, |
| 93 const std::string& access_token) { | 93 const std::string& access_token) { |
| 94 ++num_invalidate_token_; | 94 ++num_invalidate_token_; |
| 95 } | 95 } |
| 96 | 96 |
| 97 class TokenServiceProvider | 97 class TokenServiceProvider |
| 98 : public OAuth2TokenServiceRequest::TokenServiceProvider, | 98 : public OAuth2TokenServiceRequest::TokenServiceProvider, |
| 99 base::NonThreadSafe { | 99 base::NonThreadSafe { |
| 100 public: | 100 public: |
| 101 TokenServiceProvider(OAuth2TokenService* token_service); | 101 TokenServiceProvider(OAuth2TokenService* token_service); |
| 102 virtual ~TokenServiceProvider(); | |
| 103 | 102 |
| 104 // OAuth2TokenService::TokenServiceProvider implementation. | 103 // OAuth2TokenService::TokenServiceProvider implementation. |
| 105 virtual scoped_refptr<base::SingleThreadTaskRunner> | 104 virtual scoped_refptr<base::SingleThreadTaskRunner> |
| 106 GetTokenServiceTaskRunner() OVERRIDE; | 105 GetTokenServiceTaskRunner() OVERRIDE; |
| 107 virtual OAuth2TokenService* GetTokenService() OVERRIDE; | 106 virtual OAuth2TokenService* GetTokenService() OVERRIDE; |
| 108 | 107 |
| 109 private: | 108 private: |
| 109 virtual ~TokenServiceProvider(); |
| 110 |
| 110 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 111 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
| 111 OAuth2TokenService* token_service_; | 112 OAuth2TokenService* token_service_; |
| 112 }; | 113 }; |
| 113 | 114 |
| 114 TokenServiceProvider::TokenServiceProvider(OAuth2TokenService* token_service) | 115 TokenServiceProvider::TokenServiceProvider(OAuth2TokenService* token_service) |
| 115 : task_runner_(base::ThreadTaskRunnerHandle::Get()), | 116 : task_runner_(base::ThreadTaskRunnerHandle::Get()), |
| 116 token_service_(token_service) { | 117 token_service_(token_service) { |
| 117 DCHECK(token_service_); | 118 DCHECK(token_service_); |
| 118 } | 119 } |
| 119 | 120 |
| (...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 175 ResultsMap download_results_; | 176 ResultsMap download_results_; |
| 176 int num_completed_downloads_; | 177 int num_completed_downloads_; |
| 177 }; | 178 }; |
| 178 | 179 |
| 179 void AttachmentDownloaderImplTest::SetUp() { | 180 void AttachmentDownloaderImplTest::SetUp() { |
| 180 url_request_context_getter_ = | 181 url_request_context_getter_ = |
| 181 new net::TestURLRequestContextGetter(message_loop_.message_loop_proxy()); | 182 new net::TestURLRequestContextGetter(message_loop_.message_loop_proxy()); |
| 182 url_fetcher_factory_.set_remove_fetcher_on_delete(true); | 183 url_fetcher_factory_.set_remove_fetcher_on_delete(true); |
| 183 token_service_.reset(new MockOAuth2TokenService()); | 184 token_service_.reset(new MockOAuth2TokenService()); |
| 184 token_service_->AddAccount(kAccountId); | 185 token_service_->AddAccount(kAccountId); |
| 185 scoped_ptr<OAuth2TokenServiceRequest::TokenServiceProvider> | 186 scoped_refptr<OAuth2TokenServiceRequest::TokenServiceProvider> |
| 186 token_service_provider(new TokenServiceProvider(token_service_.get())); | 187 token_service_provider(new TokenServiceProvider(token_service_.get())); |
| 187 | 188 |
| 188 OAuth2TokenService::ScopeSet scopes; | 189 OAuth2TokenService::ScopeSet scopes; |
| 189 scopes.insert(GaiaConstants::kChromeSyncOAuth2Scope); | 190 scopes.insert(GaiaConstants::kChromeSyncOAuth2Scope); |
| 190 attachment_downloader_ = | 191 attachment_downloader_ = |
| 191 AttachmentDownloader::Create(GURL(kAttachmentServerUrl), | 192 AttachmentDownloader::Create(GURL(kAttachmentServerUrl), |
| 192 url_request_context_getter_, | 193 url_request_context_getter_, |
| 193 kAccountId, | 194 kAccountId, |
| 194 scopes, | 195 scopes, |
| 195 token_service_provider.Pass()); | 196 token_service_provider); |
| 196 } | 197 } |
| 197 | 198 |
| 198 void AttachmentDownloaderImplTest::TearDown() { | 199 void AttachmentDownloaderImplTest::TearDown() { |
| 199 RunMessageLoop(); | 200 RunMessageLoop(); |
| 200 } | 201 } |
| 201 | 202 |
| 202 void AttachmentDownloaderImplTest::CompleteDownload(int response_code) { | 203 void AttachmentDownloaderImplTest::CompleteDownload(int response_code) { |
| 203 // TestURLFetcherFactory remembers last active URLFetcher. | 204 // TestURLFetcherFactory remembers last active URLFetcher. |
| 204 net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0); | 205 net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0); |
| 205 // There should be outstanding url fetch request. | 206 // There should be outstanding url fetch request. |
| (...skipping 142 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 348 GoogleServiceAuthError::AuthErrorNone()); | 349 GoogleServiceAuthError::AuthErrorNone()); |
| 349 RunMessageLoop(); | 350 RunMessageLoop(); |
| 350 // Fail URLFetcher. This should trigger download failure. Access token | 351 // Fail URLFetcher. This should trigger download failure. Access token |
| 351 // shouldn't be invalidated. | 352 // shouldn't be invalidated. |
| 352 CompleteDownload(net::HTTP_SERVICE_UNAVAILABLE); | 353 CompleteDownload(net::HTTP_SERVICE_UNAVAILABLE); |
| 353 EXPECT_EQ(0, token_service()->num_invalidate_token()); | 354 EXPECT_EQ(0, token_service()->num_invalidate_token()); |
| 354 VerifyDownloadResult(id1, AttachmentDownloader::DOWNLOAD_UNSPECIFIED_ERROR); | 355 VerifyDownloadResult(id1, AttachmentDownloader::DOWNLOAD_UNSPECIFIED_ERROR); |
| 355 } | 356 } |
| 356 | 357 |
| 357 } // namespace syncer | 358 } // namespace syncer |
| OLD | NEW |