chrome/browser/signin/google_auto_login_helper_unittest.cc

   1 // Copyright 2013 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 <algorithm>
   6 #include <string>
   7 #include <vector>
   8
   9 #include "base/message_loop/message_loop.h"
  10 #include "base/strings/stringprintf.h"
  11 #include "chrome/browser/signin/google_auto_login_helper.h"
  12 #include "chrome/test/base/testing_profile.h"
  13 #include "content/public/test/test_browser_thread_bundle.h"
  14 #include "testing/gmock/include/gmock/gmock.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 namespace {
  18
  19 class MockObserver : public GoogleAutoLoginHelper::Observer {
  20  public:
  21   explicit MockObserver(GoogleAutoLoginHelper* helper) : helper_(helper) {
  22     helper_->AddObserver(this);
  23   }
  24
  25   ~MockObserver() {
  26     helper_->RemoveObserver(this);
  27   }
  28
  29   MOCK_METHOD2(MergeSessionCompleted,
  30                void(const std::string&,
  31                     const GoogleServiceAuthError& ));
  32  private:
  33   GoogleAutoLoginHelper* helper_;
  34
  35   DISALLOW_COPY_AND_ASSIGN(MockObserver);
  36 };
  37
  38 // Counts number of InstrumentedGoogleAutoLoginHelper created.
  39 // We can EXPECT_* to be zero at the end of our unit tests
  40 // to make sure everything is properly deleted.
  41
  42 int total = 0;
  43
  44 class InstrumentedGoogleAutoLoginHelper : public GoogleAutoLoginHelper {
  45  public:
  46   explicit InstrumentedGoogleAutoLoginHelper(Profile* profile) :
  47     GoogleAutoLoginHelper(profile, NULL) {
  48     total++;
  49   }
  50
  51   virtual ~InstrumentedGoogleAutoLoginHelper() {
  52     total--;
  53   }
  54
  55   MOCK_METHOD0(StartFetching, void());
  56
  57  private:
  58   DISALLOW_COPY_AND_ASSIGN(InstrumentedGoogleAutoLoginHelper);
  59 };
  60
  61 class GoogleAutoLoginHelperTest : public testing::Test {
  62  public:
  63    GoogleAutoLoginHelperTest()
  64        : no_error_(GoogleServiceAuthError::NONE),
  65          error_(GoogleServiceAuthError::SERVICE_ERROR) {}
  66
  67   TestingProfile* profile() { return &profile_; }
  68
  69   void SimulateUbertokenFailure(UbertokenConsumer* consumer,
  70                                 const GoogleServiceAuthError& error) {
  71     consumer->OnUbertokenFailure(error);
  72   }
  73
  74   void SimulateMergeSessionSuccess(GaiaAuthConsumer* consumer,
  75                                    const std::string& data) {
  76     consumer->OnMergeSessionSuccess(data);
  77   }
  78
  79   void SimulateMergeSessionFailure(GaiaAuthConsumer* consumer,
  80                                    const GoogleServiceAuthError& error) {
  81     consumer->OnMergeSessionFailure(error);
  82   }
  83
  84   const GoogleServiceAuthError& no_error() { return no_error_; }
  85   const GoogleServiceAuthError& error() { return error_; }
  86
  87  private:
  88   content::TestBrowserThreadBundle thread_bundle_;
  89   TestingProfile profile_;
  90   GoogleServiceAuthError no_error_;
  91   GoogleServiceAuthError error_;
  92 };
  93
  94 } // namespace
  95
  96 using ::testing::_;
  97
  98 TEST_F(GoogleAutoLoginHelperTest, Success) {
  99   InstrumentedGoogleAutoLoginHelper helper(profile());
 100   MockObserver observer(&helper);
 101
 102   EXPECT_CALL(helper, StartFetching());
 103   EXPECT_CALL(observer, MergeSessionCompleted("acc1@gmail.com", no_error()));
 104
 105   helper.LogIn("acc1@gmail.com");
 106   SimulateMergeSessionSuccess(&helper, "token");
 107 }
 108
 109 TEST_F(GoogleAutoLoginHelperTest, FailedMergeSession) {
 110   InstrumentedGoogleAutoLoginHelper helper(profile());
 111   MockObserver observer(&helper);
 112
 113   EXPECT_CALL(helper, StartFetching());
 114   EXPECT_CALL(observer, MergeSessionCompleted("acc1@gmail.com", error()));
 115
 116   helper.LogIn("acc1@gmail.com");
 117   SimulateMergeSessionFailure(&helper, error());
 118 }
 119
 120 TEST_F(GoogleAutoLoginHelperTest, FailedUbertoken) {
 121   InstrumentedGoogleAutoLoginHelper helper(profile());
 122   MockObserver observer(&helper);
 123
 124   EXPECT_CALL(helper, StartFetching());
 125   EXPECT_CALL(observer, MergeSessionCompleted("acc1@gmail.com", error()));
 126
 127   helper.LogIn("acc1@gmail.com");
 128   SimulateUbertokenFailure(&helper, error());
 129 }
 130
 131 TEST_F(GoogleAutoLoginHelperTest, ContinueAfterSuccess) {
 132   InstrumentedGoogleAutoLoginHelper helper(profile());
 133   MockObserver observer(&helper);
 134
 135   EXPECT_CALL(helper, StartFetching()).Times(2);
 136   EXPECT_CALL(observer, MergeSessionCompleted("acc1@gmail.com", no_error()));
 137   EXPECT_CALL(observer, MergeSessionCompleted("acc2@gmail.com", no_error()));
 138
 139   helper.LogIn("acc1@gmail.com");
 140   helper.LogIn("acc2@gmail.com");
 141   SimulateMergeSessionSuccess(&helper, "token1");
 142   SimulateMergeSessionSuccess(&helper, "token2");
 143 }
 144
 145 TEST_F(GoogleAutoLoginHelperTest, ContinueAfterFailure1) {
 146   InstrumentedGoogleAutoLoginHelper helper(profile());
 147   MockObserver observer(&helper);
 148
 149   EXPECT_CALL(helper, StartFetching()).Times(2);
 150   EXPECT_CALL(observer, MergeSessionCompleted("acc1@gmail.com", error()));
 151   EXPECT_CALL(observer, MergeSessionCompleted("acc2@gmail.com", no_error()));
 152
 153   helper.LogIn("acc1@gmail.com");
 154   helper.LogIn("acc2@gmail.com");
 155   SimulateMergeSessionFailure(&helper, error());
 156   SimulateMergeSessionSuccess(&helper, "token2");
 157 }
 158
 159 TEST_F(GoogleAutoLoginHelperTest, ContinueAfterFailure2) {
 160   InstrumentedGoogleAutoLoginHelper helper(profile());
 161   MockObserver observer(&helper);
 162
 163   EXPECT_CALL(helper, StartFetching()).Times(2);
 164   EXPECT_CALL(observer, MergeSessionCompleted("acc1@gmail.com", error()));
 165   EXPECT_CALL(observer, MergeSessionCompleted("acc2@gmail.com", no_error()));
 166
 167   helper.LogIn("acc1@gmail.com");
 168   helper.LogIn("acc2@gmail.com");
 169   SimulateUbertokenFailure(&helper, error());
 170   SimulateMergeSessionSuccess(&helper, "token2");
 171 }
 172
 173 TEST_F(GoogleAutoLoginHelperTest, AllRequestsInMultipleGoes) {
 174   InstrumentedGoogleAutoLoginHelper helper(profile());
 175   MockObserver observer(&helper);
 176
 177   EXPECT_CALL(helper, StartFetching()).Times(4);
 178   EXPECT_CALL(observer, MergeSessionCompleted(_, no_error())).Times(4);
 179
 180   helper.LogIn("acc1@gmail.com");
 181   helper.LogIn("acc2@gmail.com");
 182
 183   SimulateMergeSessionSuccess(&helper, "token1");
 184
 185   helper.LogIn("acc3@gmail.com");
 186
 187   SimulateMergeSessionSuccess(&helper, "token2");
 188   SimulateMergeSessionSuccess(&helper, "token3");
 189
 190   helper.LogIn("acc4@gmail.com");
 191
 192   SimulateMergeSessionSuccess(&helper, "token4");
 193 }