| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "google_apis/gaia/ubertoken_fetcher.h" |
| |
| #include "base/memory/ref_counted.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/message_loop/message_loop.h" |
| #include "base/thread_task_runner_handle.h" |
| #include "google_apis/gaia/fake_oauth2_token_service.h" |
| #include "google_apis/gaia/gaia_constants.h" |
| #include "net/url_request/test_url_fetcher_factory.h" |
| #include "net/url_request/url_request_test_util.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace { |
| |
| const char kTestAccountId[] = "test@gmail.com"; |
| |
| class MockUbertokenConsumer : public UbertokenConsumer { |
| public: |
| MockUbertokenConsumer() |
| : nb_correct_token_(0), |
| last_error_(GoogleServiceAuthError::AuthErrorNone()), |
| nb_error_(0) { |
| } |
| ~MockUbertokenConsumer() override {} |
| |
| void OnUbertokenSuccess(const std::string& token) override { |
| last_token_ = token; |
| ++ nb_correct_token_; |
| } |
| |
| void OnUbertokenFailure(const GoogleServiceAuthError& error) override { |
| last_error_ = error; |
| ++nb_error_; |
| } |
| |
| std::string last_token_; |
| int nb_correct_token_; |
| GoogleServiceAuthError last_error_; |
| int nb_error_; |
| }; |
| |
| } // namespace |
| |
| class UbertokenFetcherTest : public testing::Test { |
| public: |
| void SetUp() override { |
| request_context_getter_ = new net::TestURLRequestContextGetter( |
| base::ThreadTaskRunnerHandle::Get()); |
| fetcher_.reset(new UbertokenFetcher(&token_service_, |
| &consumer_, |
| GaiaConstants::kChromeSource, |
| request_context_getter_.get())); |
| } |
| |
| void TearDown() override { fetcher_.reset(); } |
| |
| protected: |
| base::MessageLoop message_loop_; |
| net::TestURLFetcherFactory factory_; |
| FakeOAuth2TokenService token_service_; |
| scoped_refptr<net::URLRequestContextGetter> request_context_getter_; |
| MockUbertokenConsumer consumer_; |
| scoped_ptr<UbertokenFetcher> fetcher_; |
| }; |
| |
| TEST_F(UbertokenFetcherTest, Basic) { |
| } |
| |
| TEST_F(UbertokenFetcherTest, Success) { |
| fetcher_->StartFetchingToken(kTestAccountId); |
| fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time()); |
| fetcher_->OnUberAuthTokenSuccess("uberToken"); |
| |
| EXPECT_EQ(0, consumer_.nb_error_); |
| EXPECT_EQ(1, consumer_.nb_correct_token_); |
| EXPECT_EQ("uberToken", consumer_.last_token_); |
| } |
| |
| TEST_F(UbertokenFetcherTest, NoRefreshToken) { |
| fetcher_->StartFetchingToken(kTestAccountId); |
| GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP); |
| fetcher_->OnGetTokenFailure(NULL, error); |
| |
| EXPECT_EQ(1, consumer_.nb_error_); |
| EXPECT_EQ(0, consumer_.nb_correct_token_); |
| } |
| |
| TEST_F(UbertokenFetcherTest, FailureToGetAccessToken) { |
| fetcher_->StartFetchingToken(kTestAccountId); |
| GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP); |
| fetcher_->OnGetTokenFailure(NULL, error); |
| |
| EXPECT_EQ(1, consumer_.nb_error_); |
| EXPECT_EQ(0, consumer_.nb_correct_token_); |
| EXPECT_EQ("", consumer_.last_token_); |
| } |
| |
| TEST_F(UbertokenFetcherTest, TransientFailureEventualFailure) { |
| fetcher_->StartFetchingToken(kTestAccountId); |
| GoogleServiceAuthError error(GoogleServiceAuthError::CONNECTION_FAILED); |
| fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time()); |
| |
| for (int i = 0; i < UbertokenFetcher::kMaxRetries; ++i) { |
| fetcher_->OnUberAuthTokenFailure(error); |
| EXPECT_EQ(0, consumer_.nb_error_); |
| EXPECT_EQ(0, consumer_.nb_correct_token_); |
| EXPECT_EQ("", consumer_.last_token_); |
| } |
| |
| fetcher_->OnUberAuthTokenFailure(error); |
| EXPECT_EQ(1, consumer_.nb_error_); |
| EXPECT_EQ(0, consumer_.nb_correct_token_); |
| EXPECT_EQ("", consumer_.last_token_); |
| } |
| |
| TEST_F(UbertokenFetcherTest, TransientFailureEventualSuccess) { |
| fetcher_->StartFetchingToken(kTestAccountId); |
| GoogleServiceAuthError error(GoogleServiceAuthError::CONNECTION_FAILED); |
| fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time()); |
| |
| for (int i = 0; i < UbertokenFetcher::kMaxRetries; ++i) { |
| fetcher_->OnUberAuthTokenFailure(error); |
| EXPECT_EQ(0, consumer_.nb_error_); |
| EXPECT_EQ(0, consumer_.nb_correct_token_); |
| EXPECT_EQ("", consumer_.last_token_); |
| } |
| |
| fetcher_->OnUberAuthTokenSuccess("uberToken"); |
| EXPECT_EQ(0, consumer_.nb_error_); |
| EXPECT_EQ(1, consumer_.nb_correct_token_); |
| EXPECT_EQ("uberToken", consumer_.last_token_); |
| } |
| |
| TEST_F(UbertokenFetcherTest, PermanentFailureEventualFailure) { |
| fetcher_->StartFetchingToken(kTestAccountId); |
| fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time()); |
| |
| GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP); |
| fetcher_->OnUberAuthTokenFailure(error); |
| EXPECT_EQ(0, consumer_.nb_error_); |
| EXPECT_EQ(0, consumer_.nb_correct_token_); |
| EXPECT_EQ("", consumer_.last_token_); |
| |
| fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time()); |
| fetcher_->OnUberAuthTokenFailure(error); |
| EXPECT_EQ(1, consumer_.nb_error_); |
| EXPECT_EQ(0, consumer_.nb_correct_token_); |
| EXPECT_EQ("", consumer_.last_token_); |
| } |
| |
| TEST_F(UbertokenFetcherTest, PermanentFailureEventualSuccess) { |
| fetcher_->StartFetchingToken(kTestAccountId); |
| GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP); |
| fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time()); |
| |
| fetcher_->OnUberAuthTokenFailure(error); |
| EXPECT_EQ(0, consumer_.nb_error_); |
| EXPECT_EQ(0, consumer_.nb_correct_token_); |
| EXPECT_EQ("", consumer_.last_token_); |
| |
| fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time()); |
| fetcher_->OnUberAuthTokenSuccess("uberToken"); |
| EXPECT_EQ(0, consumer_.nb_error_); |
| EXPECT_EQ(1, consumer_.nb_correct_token_); |
| EXPECT_EQ("uberToken", consumer_.last_token_); |
| } |