Reflect email address changes in the Avatar Menu.

components/signin/core/browser/account_tracker_service_unittest.cc

 // 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 <algorithm>
 #include <vector>
 
 #include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/scoped_user_pref_update.h"
 #include "base/prefs/testing_pref_service.h"
@@ skipping 227 matching lines @@
           ? testing::AssertionSuccess()
           : (testing::AssertionFailure()
              << "Expected " << maybe_newline << Str(events) << ", "
              << maybe_newline << "Got " << maybe_newline << Str(events_)));
   events_.clear();
   return result;
 }
 
 }  // namespace
 
+class InfoMutatingFakeAccountFetcherService : public FakeAccountFetcherService {
+ public:
+  InfoMutatingFakeAccountFetcherService(AccountInfo new_info)
+      : info_(new_info) {}
+
+  void StartFetchingUserInfo(const std::string& account_id) override {
+    // It would make no sense to return an AccountInfo with a different ID.
+    EXPECT_EQ(info_.account_id, account_id);
+    // Immediately fake a response with the provided |info_|.
+    FakeUserInfoFetchSuccess(account_id,
+                             info_.email,
+                             info_.gaia,
+                             info_.hosted_domain,
+                             info_.full_name,
+                             info_.given_name,
+                             info_.locale,
+                             info_.picture_url);
+  }
+
+ private:
+  AccountInfo info_;
+};
+
 class AccountTrackerServiceTest : public testing::Test {
  public:
   AccountTrackerServiceTest() {}
 
   ~AccountTrackerServiceTest() override {}
 
   void SetUp() override {
     fake_oauth2_token_service_.reset(new FakeOAuth2TokenService());
 
     pref_service_.registry()->RegisterListPref(
@@ skipping 61 matching lines @@
 
   net::TestURLFetcherFactory* test_fetcher_factory() {
     return &test_fetcher_factory_;
   }
   AccountFetcherService* account_fetcher() { return account_fetcher_.get(); }
   AccountTrackerService* account_tracker() { return account_tracker_.get(); }
   OAuth2TokenService* token_service() {
     return fake_oauth2_token_service_.get();
   }
   SigninClient* signin_client() { return signin_client_.get(); }
+  PrefService* pref_service() { return &pref_service_; }
 
  private:
   void ReturnOAuthUrlFetchResults(int fetcher_id,
                                   net::HttpStatusCode response_code,
                                   const std::string& response_string);
 
   base::MessageLoopForIO message_loop_;
   net::TestURLFetcherFactory test_fetcher_factory_;
   scoped_ptr<FakeOAuth2TokenService> fake_oauth2_token_service_;
   TestingPrefServiceSimple pref_service_;
@@ skipping 825 matching lines @@
   ASSERT_FALSE(info.is_child_account);
   ASSERT_TRUE(observer.CheckEvents(TrackingEvent(UPDATED, child_id)));
 
   SimulateTokenRevoked(child_id);
   ASSERT_TRUE(observer.CheckEvents(TrackingEvent(REMOVED, child_id)));
 
   tracker.RemoveObserver(&observer);
   fetcher.Shutdown();
   tracker.Shutdown();
 }
+
+TEST_F(AccountTrackerServiceTest, RefreshOnTokensLoadedWhenNeeded) {
+  std::string account_id("account_id");
+  AccountInfo info;
+  {
+    AccountTrackerService tracker;
+    tracker.Initialize(signin_client());
+    FakeAccountFetcherService fetcher;
+    fetcher.Initialize(signin_client(), token_service(), &tracker, nullptr);
+    fetcher.EnableNetworkFetches();
+    fetcher.FakeUserInfoFetchSuccess(account_id,
+                                     "original@example.com",
+                                     account_id,
+                                     "",
+                                     "first",
+                                     "fi",
+                                     "en-US",
+                                     "");
+    info = tracker.GetAccountInfo(account_id);
+    EXPECT_TRUE(info.IsValid());
+    EXPECT_EQ("original@example.com", info.email);
+  }
+
+  // Set the last update to a long time ago to trigger a refresh when the AFS
+  // starts.
+  pref_service()->SetInt64(AccountFetcherService::kLastUpdatePref, 0);
+
+  {
+    // Prepare the new account info that will be fetched for |account_id|.
+    info.email = "mutated@example.com";
+    AccountTrackerService tracker;
+    tracker.Initialize(signin_client());
+    InfoMutatingFakeAccountFetcherService fetcher(info);
+    fetcher.Initialize(signin_client(), token_service(), &tracker, nullptr);
+    fetcher.EnableNetworkFetches();
+
+    info = tracker.GetAccountInfo(account_id);
+    // The persisted info should be returned since the refresh tokens aren't
+    // loaded yet.
+    EXPECT_TRUE(info.IsValid());
+    EXPECT_EQ("original@example.com", info.email);
+
+    // Trigger a OnRefreshTokensLoaded() notification.
+    static_cast<FakeOAuth2TokenService*>(token_service())->
+        GetFakeOAuth2TokenServiceDelegate()->LoadCredentials(account_id);
+
+    info = tracker.GetAccountInfo(account_id);
+    EXPECT_TRUE(info.IsValid());
+    EXPECT_EQ("mutated@example.com", info.email);
+  }
+}
+