Allow Chrome OS login profile to have different default pref values

chrome/browser/chromeos/policy/recommendation_restorer_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 "chrome/browser/chromeos/policy/recommendation_restorer.h"
 
 #include "ash/magnifier/magnifier_constants.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/prefs/pref_notifier_impl.h"
 #include "base/prefs/testing_pref_store.h"
@@ skipping 25 matching lines @@
   const int kRestoreDelayInMs = 60 * 1000;  // 1 minute.
 }  // namespace
 
 class RecommendationRestorerTest : public testing::Test {
  protected:
   RecommendationRestorerTest();
 
   // testing::Test:
   virtual void SetUp() OVERRIDE;
 
+  void RegisterUserProfilePrefs();
+  void RegisterLoginProfilePrefs();
+
   void SetRecommendedValues();
   void SetUserSettings();
 
   void CreateLoginProfile();
   void CreateUserProfile();
 
   void NotifyOfSessionStart();
   void NotifyOfUserActivity();
 
   void VerifyPrefFollowsUser(const char* pref_name,
@@ skipping 28 matching lines @@
           new TestingPrefStore,
           new TestingPrefStore,
           recommended_prefs_,
           new user_prefs::PrefRegistrySyncable,
           new PrefNotifierImpl)),
       restorer_(NULL),
       runner_(new base::TestSimpleTaskRunner),
       runner_handler_(runner_),
       prefs_owner_(prefs_),
       profile_manager_(TestingBrowserProcess::GetGlobal()) {
-  chrome::RegisterUserPrefs(prefs_->registry());
 }
 
 void RecommendationRestorerTest::SetUp() {
   testing::Test::SetUp();
   ASSERT_TRUE(profile_manager_.SetUp());
 }
 
+void RecommendationRestorerTest::RegisterUserProfilePrefs() {
+  chrome::RegisterUserProfilePrefs(prefs_->registry());
+}
+
+void RecommendationRestorerTest::RegisterLoginProfilePrefs() {
+  chrome::RegisterLoginProfilePrefs(prefs_->registry());
+}
+
 void RecommendationRestorerTest::SetRecommendedValues() {
   recommended_prefs_->SetBoolean(prefs::kLargeCursorEnabled, false);
   recommended_prefs_->SetBoolean(prefs::kSpokenFeedbackEnabled, false);
   recommended_prefs_->SetBoolean(prefs::kHighContrastEnabled, false);
   recommended_prefs_->SetBoolean(prefs::kScreenMagnifierEnabled, false);
   recommended_prefs_->SetInteger(prefs::kScreenMagnifierType, 0);
 }
 
 void RecommendationRestorerTest::SetUserSettings() {
   prefs_->SetBoolean(prefs::kLargeCursorEnabled, true);
@@ skipping 99 matching lines @@
   EXPECT_TRUE(restorer_->restore_timer_.IsRunning());
   EXPECT_EQ(base::TimeDelta::FromMilliseconds(kRestoreDelayInMs),
             restorer_->restore_timer_.GetCurrentDelay());
 }
 
 TEST_F(RecommendationRestorerTest, CreateForUserProfile) {
   // Verifies that when a RecommendationRestorer is created for a user profile,
   // it does not start listening for any notifications, does not clear user
   // settings on initialization and does not start a timer that will clear user
   // settings eventually.
+  RegisterUserProfilePrefs();
   SetRecommendedValues();
   SetUserSettings();
 
   CreateUserProfile();
   VerifyNotListeningForNotifications();
   VerifyPrefsFollowUser();
   VerifyTimerIsStopped();
 }
 
 TEST_F(RecommendationRestorerTest, NoRecommendations) {
   // Verifies that when no recommended values have been set and a
   // RecommendationRestorer is created for the login profile, it does not clear
   // user settings on initialization and does not start a timer that will clear
   // user settings eventually.
+  RegisterLoginProfilePrefs();
   SetUserSettings();
 
   CreateLoginProfile();
   VerifyPrefsFollowUser();
   VerifyTimerIsStopped();
 }
 
 TEST_F(RecommendationRestorerTest, RestoreOnStartup) {
   // Verifies that when recommended values have been set and a
   // RecommendationRestorer is created for the login profile, it clears user
   // settings on initialization.
+  RegisterLoginProfilePrefs();
   SetRecommendedValues();
   SetUserSettings();
 
   CreateLoginProfile();
   VerifyPrefsFollowRecommendations();
   VerifyTimerIsStopped();
 }
 
 TEST_F(RecommendationRestorerTest, RestoreOnRecommendationChangeOnLoginScreen) {
   // Verifies that if recommended values change while the login screen is being
   // shown, a timer is started that will clear user settings eventually.
+  RegisterLoginProfilePrefs();
   SetUserSettings();
 
   CreateLoginProfile();
 
   VerifyTimerIsStopped();
   recommended_prefs_->SetBoolean(prefs::kLargeCursorEnabled, false);
   VerifyPrefFollowsUser(prefs::kLargeCursorEnabled,
                         base::FundamentalValue(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
@@ skipping 31 matching lines @@
                                   base::FundamentalValue(false));
   VerifyPrefFollowsRecommendation(prefs::kScreenMagnifierType,
                                   base::FundamentalValue(0));
 
   VerifyTimerIsStopped();
 }
 
 TEST_F(RecommendationRestorerTest, RestoreOnRecommendationChangeInUserSession) {
   // Verifies that if recommended values change while a user session is in
   // progress, user settings are cleared immediately.
+  RegisterLoginProfilePrefs();
   SetUserSettings();
 
   CreateLoginProfile();
   NotifyOfSessionStart();
 
   VerifyPrefFollowsUser(prefs::kLargeCursorEnabled,
                         base::FundamentalValue(true));
   recommended_prefs_->SetBoolean(prefs::kLargeCursorEnabled, false);
   VerifyTimerIsStopped();
   VerifyPrefFollowsRecommendation(prefs::kLargeCursorEnabled,
@@ skipping 23 matching lines @@
   VerifyPrefFollowsRecommendation(prefs::kScreenMagnifierEnabled,
                                   base::FundamentalValue(false));
   VerifyPrefFollowsRecommendation(prefs::kScreenMagnifierType,
                                   base::FundamentalValue(0));
 }
 
 TEST_F(RecommendationRestorerTest, DoNothingOnUserChange) {
   // Verifies that if no recommended values have been set and user settings
   // change, the user settings are not cleared immediately and no timer is
   // started that will clear the user settings eventually.
+  RegisterLoginProfilePrefs();
   CreateLoginProfile();
 
   prefs_->SetBoolean(prefs::kLargeCursorEnabled, true);
   VerifyPrefFollowsUser(prefs::kLargeCursorEnabled,
                         base::FundamentalValue(true));
   VerifyTimerIsStopped();
 
   prefs_->SetBoolean(prefs::kSpokenFeedbackEnabled, true);
   VerifyPrefFollowsUser(prefs::kSpokenFeedbackEnabled,
                         base::FundamentalValue(true));
@@ skipping 15 matching lines @@
                         base::FundamentalValue(true));
   VerifyPrefFollowsUser(prefs::kScreenMagnifierType,
                         base::FundamentalValue(ash::MAGNIFIER_FULL));
   VerifyTimerIsStopped();
 }
 
 TEST_F(RecommendationRestorerTest, RestoreOnUserChange) {
   // Verifies that if recommended values have been set and user settings change
   // while the login screen is being shown, a timer is started that will clear
   // the user settings eventually.
+  RegisterLoginProfilePrefs();
   SetRecommendedValues();
 
   CreateLoginProfile();
 
   VerifyTimerIsStopped();
   prefs_->SetBoolean(prefs::kLargeCursorEnabled, true);
   VerifyPrefFollowsUser(prefs::kLargeCursorEnabled,
                         base::FundamentalValue(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
@@ skipping 33 matching lines @@
                                   base::FundamentalValue(0));
 
   VerifyTimerIsStopped();
 }
 
 TEST_F(RecommendationRestorerTest, RestoreOnSessionStart) {
   // Verifies that if recommended values have been set, user settings have
   // changed and a session is then started, the user settings are cleared
   // immediately and the timer that would have cleared them eventually on the
   // login screen is stopped.
+  RegisterLoginProfilePrefs();
   SetRecommendedValues();
 
   CreateLoginProfile();
   SetUserSettings();
 
   NotifyOfSessionStart();
   VerifyPrefsFollowRecommendations();
   VerifyTimerIsStopped();
 }
 
 TEST_F(RecommendationRestorerTest, DoNothingOnSessionStart) {
   // Verifies that if recommended values have not been set, user settings have
   // changed and a session is then started, the user settings are not cleared
   // immediately.
+  RegisterLoginProfilePrefs();
   CreateLoginProfile();
   SetUserSettings();
 
   NotifyOfSessionStart();
   VerifyPrefsFollowUser();
   VerifyTimerIsStopped();
 }
 
 TEST_F(RecommendationRestorerTest, UserActivityResetsTimer) {
   // Verifies that user activity resets the timer which clears user settings.
+  RegisterLoginProfilePrefs();
+
   recommended_prefs_->SetBoolean(prefs::kLargeCursorEnabled, false);
 
   CreateLoginProfile();
 
   prefs_->SetBoolean(prefs::kLargeCursorEnabled, true);
   VerifyTimerIsRunning();
 
   // Notify that there is user activity, then fast forward until the originally
   // set timer fires.
   NotifyOfUserActivity();
   runner_->RunPendingTasks();
   VerifyPrefFollowsUser(prefs::kLargeCursorEnabled,
                         base::FundamentalValue(true));
 
   // Fast forward until the reset timer fires.
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(prefs::kLargeCursorEnabled,
                                   base::FundamentalValue(false));
   VerifyTimerIsStopped();
 }
 
 }  // namespace policy