chrome/browser/chromeos/policy/recommendation_restorer_unittest.cc - chromium/src.git - Git at Google

 // 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 <memory>
 #include <utility>

 #include "ash/public/cpp/ash_pref_names.h"
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/test_simple_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "base/values.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/chromeos/policy/recommendation_restorer_factory.h"
 #include "chrome/browser/prefs/browser_prefs.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chrome/test/base/testing_profile_manager.h"
 #include "components/pref_registry/pref_registry_syncable.h"
 #include "components/prefs/pref_notifier_impl.h"
 #include "components/prefs/testing_pref_store.h"
 #include "components/sync_preferences/pref_service_syncable.h"
 #include "components/sync_preferences/testing_pref_service_syncable.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "extensions/browser/quota_service.h"
 #include "testing/gtest/include/gtest/gtest.h"

 namespace policy {

 namespace {
   // The amount of idle time after which recommended values are restored.
   const int kRestoreDelayInMs = 60 * 1000;  // 1 minute.
 }  // namespace

 class RecommendationRestorerTest : public testing::Test {
  protected:
   RecommendationRestorerTest();

   // testing::Test:
   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,
                              const base::Value& expected_value) const;
   void VerifyPrefsFollowUser() const;
   void VerifyPrefFollowsRecommendation(const char* pref_name,
                                        const base::Value& expected_value) const;
   void VerifyPrefsFollowRecommendations() const;

   void VerifyNotListeningForNotifications() const;
   void VerifyTimerIsStopped() const;
   void VerifyTimerIsRunning() const;

   content::TestBrowserThreadBundle thread_bundle_;
   extensions::QuotaService::ScopedDisablePurgeForTesting
       disable_purge_for_testing_;

   TestingPrefStore* recommended_prefs_;  // Not owned.
   sync_preferences::TestingPrefServiceSyncable* prefs_;  // Not owned.
   RecommendationRestorer* restorer_;     // Not owned.

   scoped_refptr<base::TestSimpleTaskRunner> runner_;

  private:
   std::unique_ptr<sync_preferences::PrefServiceSyncable> prefs_owner_;

   TestingProfileManager profile_manager_;

   DISALLOW_COPY_AND_ASSIGN(RecommendationRestorerTest);
 };

 RecommendationRestorerTest::RecommendationRestorerTest()
     : recommended_prefs_(new TestingPrefStore),
       prefs_(new sync_preferences::TestingPrefServiceSyncable(
           new TestingPrefStore,
           new TestingPrefStore,
           new TestingPrefStore,
           recommended_prefs_,
           new user_prefs::PrefRegistrySyncable,
           new PrefNotifierImpl)),
       restorer_(NULL),
       runner_(new base::TestSimpleTaskRunner),
       prefs_owner_(prefs_),
       profile_manager_(TestingBrowserProcess::GetGlobal()) {}

 void RecommendationRestorerTest::SetUp() {
   testing::Test::SetUp();
   base::MessageLoop::current()->SetTaskRunner(runner_);
   ASSERT_TRUE(profile_manager_.SetUp());
 }

 void RecommendationRestorerTest::RegisterUserProfilePrefs() {
   ::RegisterUserProfilePrefs(prefs_->registry());
 }

 void RecommendationRestorerTest::RegisterLoginProfilePrefs() {
   ::RegisterLoginProfilePrefs(prefs_->registry());
 }

 void RecommendationRestorerTest::SetRecommendedValues() {
   recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityLargeCursorEnabled,
                                  false);
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilitySpokenFeedbackEnabled, false);
   recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled,
                                  false);
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilityScreenMagnifierEnabled, false);
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilityVirtualKeyboardEnabled, false);
 }

 void RecommendationRestorerTest::SetUserSettings() {
   prefs_->SetBoolean(ash::prefs::kAccessibilityLargeCursorEnabled, true);
   prefs_->SetBoolean(ash::prefs::kAccessibilitySpokenFeedbackEnabled, true);
   prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled, true);
   prefs_->SetBoolean(ash::prefs::kAccessibilityScreenMagnifierEnabled, true);
   prefs_->SetBoolean(ash::prefs::kAccessibilityVirtualKeyboardEnabled, true);
 }

 void RecommendationRestorerTest::CreateLoginProfile() {
   ASSERT_FALSE(restorer_);
   TestingProfile* profile = profile_manager_.CreateTestingProfile(
       chrome::kInitialProfile, std::move(prefs_owner_),
       base::UTF8ToUTF16(chrome::kInitialProfile), 0, std::string(),
       TestingProfile::TestingFactories());
   restorer_ = RecommendationRestorerFactory::GetForProfile(profile);
   EXPECT_TRUE(restorer_);
 }

 void RecommendationRestorerTest::CreateUserProfile() {
   ASSERT_FALSE(restorer_);
   TestingProfile* profile = profile_manager_.CreateTestingProfile(
       "user", std::move(prefs_owner_), base::UTF8ToUTF16("user"), 0,
       std::string(), TestingProfile::TestingFactories());
   restorer_ = RecommendationRestorerFactory::GetForProfile(profile);
   EXPECT_TRUE(restorer_);
 }

 void RecommendationRestorerTest::NotifyOfSessionStart() {
   ASSERT_TRUE(restorer_);
   restorer_->Observe(chrome::NOTIFICATION_LOGIN_USER_CHANGED,
                      content::Source<RecommendationRestorerTest>(this),
                      content::NotificationService::NoDetails());
 }

 void RecommendationRestorerTest::NotifyOfUserActivity() {
   ASSERT_TRUE(restorer_);
   restorer_->OnUserActivity(NULL);
 }

 void RecommendationRestorerTest::VerifyPrefFollowsUser(
     const char* pref_name,
     const base::Value& expected_value) const {
   const sync_preferences::PrefServiceSyncable::Preference* pref =
       prefs_->FindPreference(pref_name);
   ASSERT_TRUE(pref);
   EXPECT_TRUE(pref->HasUserSetting());
   const base::Value* value = pref->GetValue();
   ASSERT_TRUE(value);
   EXPECT_TRUE(expected_value.Equals(value));
 }

 void RecommendationRestorerTest::VerifyPrefsFollowUser() const {
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
                         base::Value(true));
   VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
                         base::Value(true));
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
                         base::Value(true));
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
                         base::Value(true));
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
                         base::Value(true));
 }

 void RecommendationRestorerTest::VerifyPrefFollowsRecommendation(
     const char* pref_name,
     const base::Value& expected_value) const {
   const sync_preferences::PrefServiceSyncable::Preference* pref =
       prefs_->FindPreference(pref_name);
   ASSERT_TRUE(pref);
   EXPECT_TRUE(pref->IsRecommended());
   EXPECT_FALSE(pref->HasUserSetting());
   const base::Value* value = pref->GetValue();
   ASSERT_TRUE(value);
   EXPECT_TRUE(expected_value.Equals(value));
 }

 void RecommendationRestorerTest::VerifyPrefsFollowRecommendations() const {
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
                                   base::Value(false));
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilitySpokenFeedbackEnabled, base::Value(false));
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityHighContrastEnabled,
                                   base::Value(false));
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilityScreenMagnifierEnabled, base::Value(false));
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilityVirtualKeyboardEnabled, base::Value(false));
 }

 void RecommendationRestorerTest::VerifyNotListeningForNotifications() const {
   ASSERT_TRUE(restorer_);
   EXPECT_TRUE(restorer_->pref_change_registrar_.IsEmpty());
   EXPECT_TRUE(restorer_->notification_registrar_.IsEmpty());
 }

 void RecommendationRestorerTest::VerifyTimerIsStopped() const {
   ASSERT_TRUE(restorer_);
   EXPECT_FALSE(restorer_->restore_timer_.IsRunning());
 }

 void RecommendationRestorerTest::VerifyTimerIsRunning() const {
   ASSERT_TRUE(restorer_);
   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(ash::prefs::kAccessibilityLargeCursorEnabled,
                                  false);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
                                   base::Value(false));

   VerifyTimerIsStopped();
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilitySpokenFeedbackEnabled, false);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilitySpokenFeedbackEnabled, base::Value(false));

   VerifyTimerIsStopped();
   recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled,
                                  false);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityHighContrastEnabled,
                                   base::Value(false));

   VerifyTimerIsStopped();
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilityScreenMagnifierEnabled, false);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilityScreenMagnifierEnabled, base::Value(false));
   VerifyTimerIsStopped();
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilityVirtualKeyboardEnabled, false);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilityVirtualKeyboardEnabled, base::Value(false));
   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(ash::prefs::kAccessibilityLargeCursorEnabled,
                         base::Value(true));
   recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityLargeCursorEnabled,
                                  false);
   VerifyTimerIsStopped();
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
                                   base::Value(false));

   VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
                         base::Value(true));
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilitySpokenFeedbackEnabled, false);
   VerifyTimerIsStopped();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilitySpokenFeedbackEnabled, base::Value(false));

   VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
                         base::Value(true));
   recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled,
                                  false);
   VerifyTimerIsStopped();
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityHighContrastEnabled,
                                   base::Value(false));

   VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
                         base::Value(true));
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilityScreenMagnifierEnabled, false);
   VerifyTimerIsStopped();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilityScreenMagnifierEnabled, base::Value(false));

   VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
                         base::Value(true));
   recommended_prefs_->SetBoolean(
       ash::prefs::kAccessibilityVirtualKeyboardEnabled, false);
   VerifyTimerIsStopped();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilityVirtualKeyboardEnabled, base::Value(false));
 }

 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(ash::prefs::kAccessibilityLargeCursorEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
                         base::Value(true));
   VerifyTimerIsStopped();

   prefs_->SetBoolean(ash::prefs::kAccessibilitySpokenFeedbackEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
                         base::Value(true));
   VerifyTimerIsStopped();

   prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
                         base::Value(true));
   VerifyTimerIsStopped();

   prefs_->SetBoolean(ash::prefs::kAccessibilityScreenMagnifierEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
                         base::Value(true));
   VerifyTimerIsStopped();

   prefs_->SetBoolean(ash::prefs::kAccessibilityScreenMagnifierEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
                         base::Value(true));
   VerifyTimerIsStopped();

   prefs_->SetBoolean(ash::prefs::kAccessibilityVirtualKeyboardEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
                         base::Value(true));
   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(ash::prefs::kAccessibilityLargeCursorEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
                                   base::Value(false));

   VerifyTimerIsStopped();
   prefs_->SetBoolean(ash::prefs::kAccessibilitySpokenFeedbackEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilitySpokenFeedbackEnabled, base::Value(false));

   VerifyTimerIsStopped();
   prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityHighContrastEnabled,
                                   base::Value(false));

   VerifyTimerIsStopped();
   prefs_->SetBoolean(ash::prefs::kAccessibilityScreenMagnifierEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilityScreenMagnifierEnabled, base::Value(false));

   VerifyTimerIsStopped();
   prefs_->SetBoolean(ash::prefs::kAccessibilityVirtualKeyboardEnabled, true);
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
                         base::Value(true));
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(
       ash::prefs::kAccessibilityVirtualKeyboardEnabled, base::Value(false));

   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(ash::prefs::kAccessibilityLargeCursorEnabled,
                                  false);

   CreateLoginProfile();

   prefs_->SetBoolean(ash::prefs::kAccessibilityLargeCursorEnabled, true);
   VerifyTimerIsRunning();

   // Notify that there is user activity, then fast forward until the originally
   // set timer fires.
   NotifyOfUserActivity();
   runner_->RunPendingTasks();
   VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
                         base::Value(true));

   // Fast forward until the reset timer fires.
   VerifyTimerIsRunning();
   runner_->RunUntilIdle();
   VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
                                   base::Value(false));
   VerifyTimerIsStopped();
 }

 }  // namespace policy
	// 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 <memory>
	#include <utility>

	#include "ash/public/cpp/ash_pref_names.h"
	#include "base/macros.h"
	#include "base/message_loop/message_loop.h"
	#include "base/strings/utf_string_conversions.h"
	#include "base/test/test_simple_task_runner.h"
	#include "base/threading/thread_task_runner_handle.h"
	#include "base/time/time.h"
	#include "base/values.h"
	#include "chrome/browser/chrome_notification_types.h"
	#include "chrome/browser/chromeos/policy/recommendation_restorer_factory.h"
	#include "chrome/browser/prefs/browser_prefs.h"
	#include "chrome/common/chrome_constants.h"
	#include "chrome/test/base/testing_browser_process.h"
	#include "chrome/test/base/testing_profile.h"
	#include "chrome/test/base/testing_profile_manager.h"
	#include "components/pref_registry/pref_registry_syncable.h"
	#include "components/prefs/pref_notifier_impl.h"
	#include "components/prefs/testing_pref_store.h"
	#include "components/sync_preferences/pref_service_syncable.h"
	#include "components/sync_preferences/testing_pref_service_syncable.h"
	#include "content/public/browser/notification_details.h"
	#include "content/public/browser/notification_service.h"
	#include "content/public/browser/notification_source.h"
	#include "content/public/test/test_browser_thread_bundle.h"
	#include "extensions/browser/quota_service.h"
	#include "testing/gtest/include/gtest/gtest.h"

	namespace policy {

	namespace {
	// The amount of idle time after which recommended values are restored.
	const int kRestoreDelayInMs = 60 * 1000; // 1 minute.
	} // namespace

	class RecommendationRestorerTest : public testing::Test {
	protected:
	RecommendationRestorerTest();

	// testing::Test:
	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,
	const base::Value& expected_value) const;
	void VerifyPrefsFollowUser() const;
	void VerifyPrefFollowsRecommendation(const char* pref_name,
	const base::Value& expected_value) const;
	void VerifyPrefsFollowRecommendations() const;

	void VerifyNotListeningForNotifications() const;
	void VerifyTimerIsStopped() const;
	void VerifyTimerIsRunning() const;

	content::TestBrowserThreadBundle thread_bundle_;
	extensions::QuotaService::ScopedDisablePurgeForTesting
	disable_purge_for_testing_;

	TestingPrefStore* recommended_prefs_; // Not owned.
	sync_preferences::TestingPrefServiceSyncable* prefs_; // Not owned.
	RecommendationRestorer* restorer_; // Not owned.

	scoped_refptr<base::TestSimpleTaskRunner> runner_;

	private:
	std::unique_ptr<sync_preferences::PrefServiceSyncable> prefs_owner_;

	TestingProfileManager profile_manager_;

	DISALLOW_COPY_AND_ASSIGN(RecommendationRestorerTest);
	};

	RecommendationRestorerTest::RecommendationRestorerTest()
	: recommended_prefs_(new TestingPrefStore),
	prefs_(new sync_preferences::TestingPrefServiceSyncable(
	new TestingPrefStore,
	new TestingPrefStore,
	new TestingPrefStore,
	recommended_prefs_,
	new user_prefs::PrefRegistrySyncable,
	new PrefNotifierImpl)),
	restorer_(NULL),
	runner_(new base::TestSimpleTaskRunner),
	prefs_owner_(prefs_),
	profile_manager_(TestingBrowserProcess::GetGlobal()) {}

	void RecommendationRestorerTest::SetUp() {
	testing::Test::SetUp();
	base::MessageLoop::current()->SetTaskRunner(runner_);
	ASSERT_TRUE(profile_manager_.SetUp());
	}

	void RecommendationRestorerTest::RegisterUserProfilePrefs() {
	::RegisterUserProfilePrefs(prefs_->registry());
	}

	void RecommendationRestorerTest::RegisterLoginProfilePrefs() {
	::RegisterLoginProfilePrefs(prefs_->registry());
	}

	void RecommendationRestorerTest::SetRecommendedValues() {
	recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityLargeCursorEnabled,
	false);
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilitySpokenFeedbackEnabled, false);
	recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled,
	false);
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilityScreenMagnifierEnabled, false);
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilityVirtualKeyboardEnabled, false);
	}

	void RecommendationRestorerTest::SetUserSettings() {
	prefs_->SetBoolean(ash::prefs::kAccessibilityLargeCursorEnabled, true);
	prefs_->SetBoolean(ash::prefs::kAccessibilitySpokenFeedbackEnabled, true);
	prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled, true);
	prefs_->SetBoolean(ash::prefs::kAccessibilityScreenMagnifierEnabled, true);
	prefs_->SetBoolean(ash::prefs::kAccessibilityVirtualKeyboardEnabled, true);
	}

	void RecommendationRestorerTest::CreateLoginProfile() {
	ASSERT_FALSE(restorer_);
	TestingProfile* profile = profile_manager_.CreateTestingProfile(
	chrome::kInitialProfile, std::move(prefs_owner_),
	base::UTF8ToUTF16(chrome::kInitialProfile), 0, std::string(),
	TestingProfile::TestingFactories());
	restorer_ = RecommendationRestorerFactory::GetForProfile(profile);
	EXPECT_TRUE(restorer_);
	}

	void RecommendationRestorerTest::CreateUserProfile() {
	ASSERT_FALSE(restorer_);
	TestingProfile* profile = profile_manager_.CreateTestingProfile(
	"user", std::move(prefs_owner_), base::UTF8ToUTF16("user"), 0,
	std::string(), TestingProfile::TestingFactories());
	restorer_ = RecommendationRestorerFactory::GetForProfile(profile);
	EXPECT_TRUE(restorer_);
	}

	void RecommendationRestorerTest::NotifyOfSessionStart() {
	ASSERT_TRUE(restorer_);
	restorer_->Observe(chrome::NOTIFICATION_LOGIN_USER_CHANGED,
	content::Source<RecommendationRestorerTest>(this),
	content::NotificationService::NoDetails());
	}

	void RecommendationRestorerTest::NotifyOfUserActivity() {
	ASSERT_TRUE(restorer_);
	restorer_->OnUserActivity(NULL);
	}

	void RecommendationRestorerTest::VerifyPrefFollowsUser(
	const char* pref_name,
	const base::Value& expected_value) const {
	const sync_preferences::PrefServiceSyncable::Preference* pref =
	prefs_->FindPreference(pref_name);
	ASSERT_TRUE(pref);
	EXPECT_TRUE(pref->HasUserSetting());
	const base::Value* value = pref->GetValue();
	ASSERT_TRUE(value);
	EXPECT_TRUE(expected_value.Equals(value));
	}

	void RecommendationRestorerTest::VerifyPrefsFollowUser() const {
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(true));
	VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
	base::Value(true));
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(true));
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
	base::Value(true));
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
	base::Value(true));
	}

	void RecommendationRestorerTest::VerifyPrefFollowsRecommendation(
	const char* pref_name,
	const base::Value& expected_value) const {
	const sync_preferences::PrefServiceSyncable::Preference* pref =
	prefs_->FindPreference(pref_name);
	ASSERT_TRUE(pref);
	EXPECT_TRUE(pref->IsRecommended());
	EXPECT_FALSE(pref->HasUserSetting());
	const base::Value* value = pref->GetValue();
	ASSERT_TRUE(value);
	EXPECT_TRUE(expected_value.Equals(value));
	}

	void RecommendationRestorerTest::VerifyPrefsFollowRecommendations() const {
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(false));
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilitySpokenFeedbackEnabled, base::Value(false));
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(false));
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilityScreenMagnifierEnabled, base::Value(false));
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilityVirtualKeyboardEnabled, base::Value(false));
	}

	void RecommendationRestorerTest::VerifyNotListeningForNotifications() const {
	ASSERT_TRUE(restorer_);
	EXPECT_TRUE(restorer_->pref_change_registrar_.IsEmpty());
	EXPECT_TRUE(restorer_->notification_registrar_.IsEmpty());
	}

	void RecommendationRestorerTest::VerifyTimerIsStopped() const {
	ASSERT_TRUE(restorer_);
	EXPECT_FALSE(restorer_->restore_timer_.IsRunning());
	}

	void RecommendationRestorerTest::VerifyTimerIsRunning() const {
	ASSERT_TRUE(restorer_);
	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(ash::prefs::kAccessibilityLargeCursorEnabled,
	false);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(false));

	VerifyTimerIsStopped();
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilitySpokenFeedbackEnabled, false);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilitySpokenFeedbackEnabled, base::Value(false));

	VerifyTimerIsStopped();
	recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled,
	false);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(false));

	VerifyTimerIsStopped();
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilityScreenMagnifierEnabled, false);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilityScreenMagnifierEnabled, base::Value(false));
	VerifyTimerIsStopped();
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilityVirtualKeyboardEnabled, false);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilityVirtualKeyboardEnabled, base::Value(false));
	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(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(true));
	recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityLargeCursorEnabled,
	false);
	VerifyTimerIsStopped();
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(false));

	VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
	base::Value(true));
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilitySpokenFeedbackEnabled, false);
	VerifyTimerIsStopped();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilitySpokenFeedbackEnabled, base::Value(false));

	VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(true));
	recommended_prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled,
	false);
	VerifyTimerIsStopped();
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(false));

	VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
	base::Value(true));
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilityScreenMagnifierEnabled, false);
	VerifyTimerIsStopped();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilityScreenMagnifierEnabled, base::Value(false));

	VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
	base::Value(true));
	recommended_prefs_->SetBoolean(
	ash::prefs::kAccessibilityVirtualKeyboardEnabled, false);
	VerifyTimerIsStopped();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilityVirtualKeyboardEnabled, base::Value(false));
	}

	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(ash::prefs::kAccessibilityLargeCursorEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(true));
	VerifyTimerIsStopped();

	prefs_->SetBoolean(ash::prefs::kAccessibilitySpokenFeedbackEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
	base::Value(true));
	VerifyTimerIsStopped();

	prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(true));
	VerifyTimerIsStopped();

	prefs_->SetBoolean(ash::prefs::kAccessibilityScreenMagnifierEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
	base::Value(true));
	VerifyTimerIsStopped();

	prefs_->SetBoolean(ash::prefs::kAccessibilityScreenMagnifierEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
	base::Value(true));
	VerifyTimerIsStopped();

	prefs_->SetBoolean(ash::prefs::kAccessibilityVirtualKeyboardEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
	base::Value(true));
	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(ash::prefs::kAccessibilityLargeCursorEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(false));

	VerifyTimerIsStopped();
	prefs_->SetBoolean(ash::prefs::kAccessibilitySpokenFeedbackEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilitySpokenFeedbackEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilitySpokenFeedbackEnabled, base::Value(false));

	VerifyTimerIsStopped();
	prefs_->SetBoolean(ash::prefs::kAccessibilityHighContrastEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityHighContrastEnabled,
	base::Value(false));

	VerifyTimerIsStopped();
	prefs_->SetBoolean(ash::prefs::kAccessibilityScreenMagnifierEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityScreenMagnifierEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilityScreenMagnifierEnabled, base::Value(false));

	VerifyTimerIsStopped();
	prefs_->SetBoolean(ash::prefs::kAccessibilityVirtualKeyboardEnabled, true);
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityVirtualKeyboardEnabled,
	base::Value(true));
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(
	ash::prefs::kAccessibilityVirtualKeyboardEnabled, base::Value(false));

	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(ash::prefs::kAccessibilityLargeCursorEnabled,
	false);

	CreateLoginProfile();

	prefs_->SetBoolean(ash::prefs::kAccessibilityLargeCursorEnabled, true);
	VerifyTimerIsRunning();

	// Notify that there is user activity, then fast forward until the originally
	// set timer fires.
	NotifyOfUserActivity();
	runner_->RunPendingTasks();
	VerifyPrefFollowsUser(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(true));

	// Fast forward until the reset timer fires.
	VerifyTimerIsRunning();
	runner_->RunUntilIdle();
	VerifyPrefFollowsRecommendation(ash::prefs::kAccessibilityLargeCursorEnabled,
	base::Value(false));
	VerifyTimerIsStopped();
	}

	} // namespace policy