1// Copyright (c) 2012 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 "chrome/browser/signin/ubertoken_fetcher.h"
6
7#include "base/memory/scoped_ptr.h"
8#include "chrome/browser/signin/fake_profile_oauth2_token_service.h"
9#include "chrome/browser/signin/fake_signin_manager.h"
10#include "chrome/browser/signin/profile_oauth2_token_service.h"
11#include "chrome/browser/signin/profile_oauth2_token_service_factory.h"
12#include "chrome/browser/signin/signin_manager_factory.h"
13#include "chrome/test/base/testing_profile.h"
14#include "content/public/test/test_browser_thread_bundle.h"
15#include "google_apis/gaia/gaia_constants.h"
16#include "net/url_request/test_url_fetcher_factory.h"
17#include "testing/gtest/include/gtest/gtest.h"
18
19namespace {
20
21const char kTestAccountId[] = "test@gmail.com";
22
23class MockUbertokenConsumer : public UbertokenConsumer {
24 public:
25  MockUbertokenConsumer()
26      : nb_correct_token_(0),
27        last_error_(GoogleServiceAuthError::AuthErrorNone()),
28        nb_error_(0) {
29  }
30  virtual ~MockUbertokenConsumer() {}
31
32  virtual void OnUbertokenSuccess(const std::string& token) OVERRIDE {
33    last_token_ = token;
34    ++ nb_correct_token_;
35  }
36
37  virtual void OnUbertokenFailure(const GoogleServiceAuthError& error)
38      OVERRIDE {
39    last_error_ = error;
40    ++nb_error_;
41  }
42
43  std::string last_token_;
44  int nb_correct_token_;
45  GoogleServiceAuthError last_error_;
46  int nb_error_;
47};
48
49}  // namespace
50
51class UbertokenFetcherTest : public testing::Test {
52 public:
53  virtual void SetUp() OVERRIDE {
54    profile_ = CreateProfile();
55    fetcher_.reset(new UbertokenFetcher(profile(), &consumer_));
56  }
57
58  scoped_ptr<TestingProfile> CreateProfile() {
59    TestingProfile::Builder builder;
60    builder.AddTestingFactory(ProfileOAuth2TokenServiceFactory::GetInstance(),
61                              &FakeProfileOAuth2TokenService::Build);
62    return builder.Build().Pass();
63  }
64
65  virtual void TearDown() OVERRIDE {
66    fetcher_.reset();
67  }
68
69  TestingProfile* profile() { return profile_.get(); }
70
71 protected:
72  content::TestBrowserThreadBundle thread_bundle_;
73  scoped_ptr<TestingProfile> profile_;
74  net::TestURLFetcherFactory factory_;
75  MockUbertokenConsumer consumer_;
76  scoped_ptr<UbertokenFetcher> fetcher_;
77};
78
79TEST_F(UbertokenFetcherTest, Basic) {
80}
81
82TEST_F(UbertokenFetcherTest, Success) {
83  ProfileOAuth2TokenServiceFactory::GetForProfile(profile())->
84      UpdateCredentials(kTestAccountId, "refreshToken");
85  fetcher_->StartFetchingToken();
86  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
87  fetcher_->OnUberAuthTokenSuccess("uberToken");
88  EXPECT_EQ(0, consumer_.nb_error_);
89  EXPECT_EQ(1, consumer_.nb_correct_token_);
90  EXPECT_EQ("uberToken", consumer_.last_token_);
91}
92
93TEST_F(UbertokenFetcherTest, NoRefreshToken) {
94  fetcher_->StartFetchingToken();
95  GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP);
96  fetcher_->OnGetTokenFailure(NULL, error);
97  EXPECT_EQ(1, consumer_.nb_error_);
98  EXPECT_EQ(0, consumer_.nb_correct_token_);
99}
100
101TEST_F(UbertokenFetcherTest, FailureToGetAccessToken) {
102  GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP);
103
104  ProfileOAuth2TokenServiceFactory::GetForProfile(profile())->
105      UpdateCredentials(kTestAccountId, "refreshToken");
106  fetcher_->StartFetchingToken();
107  fetcher_->OnGetTokenFailure(NULL, error);
108
109  EXPECT_EQ(1, consumer_.nb_error_);
110  EXPECT_EQ(0, consumer_.nb_correct_token_);
111  EXPECT_EQ("", consumer_.last_token_);
112}
113
114TEST_F(UbertokenFetcherTest, FailureToGetUberToken) {
115  GoogleServiceAuthError error(GoogleServiceAuthError::USER_NOT_SIGNED_UP);
116
117  ProfileOAuth2TokenServiceFactory::GetForProfile(profile())->
118      UpdateCredentials(kTestAccountId, "refreshToken");
119  fetcher_->StartFetchingToken();
120  fetcher_->OnGetTokenSuccess(NULL, "accessToken", base::Time());
121  fetcher_->OnUberAuthTokenFailure(error);
122
123  EXPECT_EQ(1, consumer_.nb_error_);
124  EXPECT_EQ(0, consumer_.nb_correct_token_);
125  EXPECT_EQ("", consumer_.last_token_);
126}
127