Fix issues with token refresh in AccountReconcilor

chrome/browser/signin/google_auto_login_helper_unittest.cc

+// Copyright 2013 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 <algorithm>
+#include <string>
+#include <vector>
+
+#include "base/message_loop/message_loop.h"
+#include "chrome/browser/signin/google_auto_login_helper.h"
+#include "chrome/test/base/testing_profile.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+class MockUbertokenFetcher : public UbertokenFetcher {
+ public:
+  MockUbertokenFetcher(Profile* profile, UbertokenConsumer* consumer) :
+      UbertokenFetcher(profile, consumer), account_id("") {}
+
+  void completePendingRequest() {
+    OnUberAuthTokenSuccess("mock token: " + account_id);
+  }
+
+  virtual void StartFetchingToken(const std::string& id) OVERRIDE {
+    account_id = id;
+  }
+
+  std::string account_id;
+};
+
+// Counts number of InstrumentedGoogleAutoLoginHelper created.
+// We can EXPECT_* to be zero at the end of our unit tests
+// to make sure everything is properly deleted.
+
+int total = 0;
+
+class InstrumentedGoogleAutoLoginHelper : public GoogleAutoLoginHelper {
+ public:
+  std::vector<MockUbertokenFetcher*>* mock_uber_fetchers_;
+  TestingProfile profile_;
+
+  InstrumentedGoogleAutoLoginHelper(
+    std::vector<MockUbertokenFetcher*>* fetchers) :
+    GoogleAutoLoginHelper(0), mock_uber_fetchers_(fetchers) {
+    total++;
+  }
+
+  virtual ~InstrumentedGoogleAutoLoginHelper() {
+    total--;
+  }
+
+  virtual void OnUbertokenSuccess(const std::string& token) OVERRIDE {
+    OnMergeSessionSuccess(token);
+  }
+
+  virtual UbertokenFetcher* CreateNewUbertokenFetcher() OVERRIDE {
+    MockUbertokenFetcher* m = new MockUbertokenFetcher(&profile_, this);
+    mock_uber_fetchers_->push_back(m);
+    return m;
+  }
+
+  void completePendingFetchers() {
+    while (!mock_uber_fetchers_->empty()) {
+      MockUbertokenFetcher* fetcher = mock_uber_fetchers_->back();
+      mock_uber_fetchers_->pop_back();
+      fetcher->completePendingRequest();
+    }
+  }
+
+  MOCK_METHOD1(OnMergeSessionSuccess, void(const std::string& uber_token));
+
+  void OnMergeSessionSuccessConcrete(const std::string& uber_token) {
+    GoogleAutoLoginHelper::OnMergeSessionSuccess(uber_token);
+  }
+};
+
+class GoogleAutoLoginHelperTest : public testing::Test {
+ public:
+  std::vector<MockUbertokenFetcher*> mock_uber_fetchers_;
+  base::MessageLoop message_loop_;
+};
+
+} // namespace
+
+using ::testing::Invoke;
+using ::testing::_;
+
+TEST_F(GoogleAutoLoginHelperTest, AllRequestsInOneGo) {
+  total = 0;
+
+  InstrumentedGoogleAutoLoginHelper* helper =
+      new InstrumentedGoogleAutoLoginHelper(&mock_uber_fetchers_);
+  EXPECT_CALL(*helper, OnMergeSessionSuccess("mock token: acc1@gmail.com"));
+  EXPECT_CALL(*helper, OnMergeSessionSuccess("mock token: acc2@gmail.com"));
+  EXPECT_CALL(*helper, OnMergeSessionSuccess("mock token: acc3@gmail.com"));
+  EXPECT_CALL(*helper, OnMergeSessionSuccess("mock token: acc4@gmail.com"));
+  EXPECT_CALL(*helper, OnMergeSessionSuccess("mock token: acc5@gmail.com"));
+
+  // Don't completely mock out OnMergeSessionSuccess, let GoogleAutoLoginHelper
+  // actually call OnMergeSessionSuccess to clean up it's work queue because
+  // it'll delete itself once the work queue becomes empty.
+  ON_CALL(*helper, OnMergeSessionSuccess(_)).WillByDefault(Invoke(helper,
+          &InstrumentedGoogleAutoLoginHelper::OnMergeSessionSuccessConcrete));
+  helper->LogIn("acc1@gmail.com");
+  helper->LogIn("acc2@gmail.com");
+  helper->LogIn("acc3@gmail.com");
+  helper->LogIn("acc4@gmail.com");
+  helper->LogIn("acc5@gmail.com");
+  helper->completePendingFetchers();
+
+  // Make sure GoogleAutoLoginHelper cleans itself up after everything is done.
+  message_loop_.RunUntilIdle();
+  EXPECT_EQ(0, total);
+}