Added a ForceYouTubeRestrict policy and deprecated the old ForceYouTubeSafetyMode policy

chrome/browser/supervised_user/supervised_user_pref_store_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 <set>
 #include <string>
 
 #include "base/memory/ref_counted.h"
 #include "base/values.h"
+#include "chrome/browser/net/safe_search_util.h"
 #include "chrome/browser/supervised_user/supervised_user_constants.h"
 #include "chrome/browser/supervised_user/supervised_user_pref_store.h"
 #include "chrome/browser/supervised_user/supervised_user_settings_service.h"
 #include "chrome/common/pref_names.h"
 #include "components/prefs/testing_pref_store.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::DictionaryValue;
 using base::Value;
 
@@ skipping 86 matching lines @@
   bool allow_deleting_browser_history = true;
   EXPECT_TRUE(fixture.changed_prefs()->GetBoolean(
       prefs::kAllowDeletingBrowserHistory, &allow_deleting_browser_history));
   EXPECT_FALSE(allow_deleting_browser_history);
 
   // kSupervisedModeManualHosts does not have a hardcoded value.
   base::DictionaryValue* manual_hosts = NULL;
   EXPECT_FALSE(fixture.changed_prefs()->GetDictionary(
       prefs::kSupervisedUserManualHosts, &manual_hosts));
 
-  // kForceGoogleSafeSearch and kForceYouTubeSafetyMode default to true for
-  // supervised users.
+  // kForceGoogleSafeSearch defaults to true and kForceYouTubeRestrict defaults
+  // to Moderate for supervised users.
   bool force_google_safesearch = false;
-  bool force_youtube_safety_mode = false;
+  int force_youtube_restrict =
+      static_cast<int>(safe_search_util::YouTubeRestrictMode::OFF);
   EXPECT_TRUE(fixture.changed_prefs()->GetBoolean(prefs::kForceGoogleSafeSearch,
                                                   &force_google_safesearch));
-  EXPECT_TRUE(fixture.changed_prefs()->GetBoolean(
-      prefs::kForceYouTubeSafetyMode, &force_youtube_safety_mode));
+  EXPECT_TRUE(fixture.changed_prefs()->GetInteger(prefs::kForceYouTubeRestrict,
+                                                  &force_youtube_restrict));
   EXPECT_TRUE(force_google_safesearch);
-  EXPECT_TRUE(force_youtube_safety_mode);
+  EXPECT_EQ(force_youtube_restrict,
+            static_cast<int>(safe_search_util::YouTubeRestrictMode::MODERATE));
 
   // Activating the service again should not change anything.
   fixture.changed_prefs()->Clear();
   service_.SetActive(true);
   EXPECT_EQ(0u, fixture.changed_prefs()->size());
 
   // kSupervisedModeManualHosts can be configured by the custodian.
   std::unique_ptr<base::DictionaryValue> dict(new base::DictionaryValue);
   dict->SetBoolean("example.com", true);
   dict->SetBoolean("moose.org", false);
   service_.SetLocalSetting(supervised_users::kContentPackManualBehaviorHosts,
                            std::unique_ptr<base::Value>(dict->DeepCopy()));
   EXPECT_EQ(1u, fixture.changed_prefs()->size());
   ASSERT_TRUE(fixture.changed_prefs()->GetDictionary(
       prefs::kSupervisedUserManualHosts, &manual_hosts));
   EXPECT_TRUE(manual_hosts->Equals(dict.get()));
 
-  // kForceGoogleSafeSearch and kForceYouTubeSafetyMode can be configured by the
+  // kForceGoogleSafeSearch and kForceYouTubeRestrict can be configured by the
   // custodian, overriding the hardcoded default.
   fixture.changed_prefs()->Clear();
   service_.SetLocalSetting(
       supervised_users::kForceSafeSearch,
       std::unique_ptr<base::Value>(new base::FundamentalValue(false)));
   EXPECT_EQ(1u, fixture.changed_prefs()->size());
   EXPECT_TRUE(fixture.changed_prefs()->GetBoolean(prefs::kForceGoogleSafeSearch,
                                                   &force_google_safesearch));
-  EXPECT_TRUE(fixture.changed_prefs()->GetBoolean(
-      prefs::kForceYouTubeSafetyMode, &force_youtube_safety_mode));
-  EXPECT_FALSE(force_youtube_safety_mode);
+  EXPECT_TRUE(fixture.changed_prefs()->GetInteger(prefs::kForceYouTubeRestrict,
+                                                  &force_youtube_restrict));
   EXPECT_FALSE(force_google_safesearch);
+  EXPECT_EQ(force_youtube_restrict,
+            static_cast<int>(safe_search_util::YouTubeRestrictMode::OFF));
 }
 
 TEST_F(SupervisedUserPrefStoreTest, ActivateSettingsBeforeInitialization) {
   SupervisedUserPrefStoreFixture fixture(&service_);
   EXPECT_FALSE(fixture.initialization_completed());
 
   service_.SetActive(true);
   EXPECT_FALSE(fixture.initialization_completed());
   EXPECT_EQ(0u, fixture.changed_prefs()->size());
 
   pref_store_->SetInitializationCompleted();
   EXPECT_TRUE(fixture.initialization_completed());
   EXPECT_EQ(0u, fixture.changed_prefs()->size());
 }
 
 TEST_F(SupervisedUserPrefStoreTest, CreatePrefStoreAfterInitialization) {
   pref_store_->SetInitializationCompleted();
   service_.SetActive(true);
 
   SupervisedUserPrefStoreFixture fixture(&service_);
   EXPECT_TRUE(fixture.initialization_completed());
   EXPECT_EQ(0u, fixture.changed_prefs()->size());
 }