ubertoken_fetcher_unittest.cc revision 1320f92c476a1ad9d19dba2a48c72b75566198e9 
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
3// found in the LICENSE file.
4
5#include "google_apis/gaia/ubertoken_fetcher.h"
6
7#include "base/memory/ref_counted.h"
8#include "base/memory/scoped_ptr.h"
9#include "base/message_loop/message_loop.h"
10#include "google_apis/gaia/fake_oauth2_token_service.h"
11#include "google_apis/gaia/gaia_constants.h"
12#include "net/url_request/test_url_fetcher_factory.h"
13#include "net/url_request/url_request_test_util.h"
14#include "testing/gtest/include/gtest/gtest.h"
15
16namespace {
17
18const char kTestAccountId[] = "test@gmail.com";
19
20class MockUbertokenConsumer : public UbertokenConsumer {
21 public:
22  MockUbertokenConsumer()
23      : nb_correct_token_(0),
24        last_error_(GoogleServiceAuthError::AuthErrorNone()),
25        nb_error_(0) {
26  }
27  virtual ~MockUbertokenConsumer() {}
28
29  virtual void OnUbertokenSuccess(const std::string& token) OVERRIDE {
30    last_token_ = token;
31    ++ nb_correct_token_;
32  }
33
34  virtual void OnUbertokenFailure(const GoogleServiceAuthError& error)
35      OVERRIDE {
36    last_error_ = error;
37    ++nb_error_;
38  }
39
40  std::string last_token_;
41  int nb_correct_token_;
42  GoogleServiceAuthError last_error_;
43  int nb_error_;
44};
45
46}  // namespace
47
48class UbertokenFetcherTest : public testing::Test {
49 public:
50  virtual void SetUp() OVERRIDE {
51    request_context_getter_ = new net::TestURLRequestContextGetter(
52        base::MessageLoopProxy::current());
53    fetcher_.reset(new UbertokenFetcher(&token_service_,
54                                        &consumer_,
55                                        request_context_getter_.get()));
56  }
57
58  virtual void TearDown() OVERRIDE {
59    fetcher_.reset();
60  }
61
62 protected:
63  base::MessageLoop message_loop_;
64  net::TestURLFetcherFactory factory_;
65  FakeOAuth2TokenService token_service_;
66  scoped_refptr<net::URLRequestContextGetter> request_context_getter_;
67  MockUbertokenConsumer consumer_;
68  scoped_ptr<UbertokenFetcher> fetcher_;
69};
70
71TEST_F(UbertokenFetcherTest, Basic) {
72}
73
74TEST_F(UbertokenFetcherTest, Success) {
75  fetcher_->StartFetchingToken(kTestAccountId);
76  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
77  fetcher_->OnUberAuthTokenSuccess("uberToken");
78
79  EXPECT_EQ(0, consumer_.nb_error_);
80  EXPECT_EQ(1, consumer_.nb_correct_token_);
81  EXPECT_EQ("uberToken", consumer_.last_token_);
82}
83
84TEST_F(UbertokenFetcherTest, NoRefreshToken) {
85  fetcher_->StartFetchingToken(kTestAccountId);
86  GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP);
87  fetcher_->OnGetTokenFailure(NULL, error);
88
89  EXPECT_EQ(1, consumer_.nb_error_);
90  EXPECT_EQ(0, consumer_.nb_correct_token_);
91}
92
93TEST_F(UbertokenFetcherTest, FailureToGetAccessToken) {
94  fetcher_->StartFetchingToken(kTestAccountId);
95  GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP);
96  fetcher_->OnGetTokenFailure(NULL, error);
97
98  EXPECT_EQ(1, consumer_.nb_error_);
99  EXPECT_EQ(0, consumer_.nb_correct_token_);
100  EXPECT_EQ("", consumer_.last_token_);
101}
102
103TEST_F(UbertokenFetcherTest, TransientFailureEventualFailure) {
104  fetcher_->StartFetchingToken(kTestAccountId);
105  GoogleServiceAuthError error(GoogleServiceAuthError::CONNECTION_FAILED);
106  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
107
108  for (int i = 0; i < UbertokenFetcher::kMaxRetries; ++i) {
109    fetcher_->OnUberAuthTokenFailure(error);
110    EXPECT_EQ(0, consumer_.nb_error_);
111    EXPECT_EQ(0, consumer_.nb_correct_token_);
112    EXPECT_EQ("", consumer_.last_token_);
113  }
114
115  fetcher_->OnUberAuthTokenFailure(error);
116  EXPECT_EQ(1, consumer_.nb_error_);
117  EXPECT_EQ(0, consumer_.nb_correct_token_);
118  EXPECT_EQ("", consumer_.last_token_);
119}
120
121TEST_F(UbertokenFetcherTest, TransientFailureEventualSuccess) {
122  fetcher_->StartFetchingToken(kTestAccountId);
123  GoogleServiceAuthError error(GoogleServiceAuthError::CONNECTION_FAILED);
124  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
125
126  for (int i = 0; i < UbertokenFetcher::kMaxRetries; ++i) {
127    fetcher_->OnUberAuthTokenFailure(error);
128    EXPECT_EQ(0, consumer_.nb_error_);
129    EXPECT_EQ(0, consumer_.nb_correct_token_);
130    EXPECT_EQ("", consumer_.last_token_);
131  }
132
133  fetcher_->OnUberAuthTokenSuccess("uberToken");
134  EXPECT_EQ(0, consumer_.nb_error_);
135  EXPECT_EQ(1, consumer_.nb_correct_token_);
136  EXPECT_EQ("uberToken", consumer_.last_token_);
137}
138
139TEST_F(UbertokenFetcherTest, PermanentFailureEventualFailure) {
140  fetcher_->StartFetchingToken(kTestAccountId);
141  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
142
143  GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP);
144  fetcher_->OnUberAuthTokenFailure(error);
145  EXPECT_EQ(0, consumer_.nb_error_);
146  EXPECT_EQ(0, consumer_.nb_correct_token_);
147  EXPECT_EQ("", consumer_.last_token_);
148
149  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
150  fetcher_->OnUberAuthTokenFailure(error);
151  EXPECT_EQ(1, consumer_.nb_error_);
152  EXPECT_EQ(0, consumer_.nb_correct_token_);
153  EXPECT_EQ("", consumer_.last_token_);
154}
155
156TEST_F(UbertokenFetcherTest, PermanentFailureEventualSuccess) {
157  fetcher_->StartFetchingToken(kTestAccountId);
158  GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP);
159  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
160
161  fetcher_->OnUberAuthTokenFailure(error);
162  EXPECT_EQ(0, consumer_.nb_error_);
163  EXPECT_EQ(0, consumer_.nb_correct_token_);
164  EXPECT_EQ("", consumer_.last_token_);
165
166  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
167  fetcher_->OnUberAuthTokenSuccess("uberToken");
168  EXPECT_EQ(0, consumer_.nb_error_);
169  EXPECT_EQ(1, consumer_.nb_correct_token_);
170  EXPECT_EQ("uberToken", consumer_.last_token_);
171}
172