Added a ForceYouTubeRestrict policy and deprecated the old ForceYouTubeSafetyMode policy

chrome/browser/net/chrome_network_delegate_unittest.cc

Index: chrome/browser/net/chrome_network_delegate_unittest.cc
diff --git a/chrome/browser/net/chrome_network_delegate_unittest.cc b/chrome/browser/net/chrome_network_delegate_unittest.cc
index 394b23b1220d6a4e98708de8dd00e85e5e421955..c406de5acac7c7884e300bf59b39e15b5485260e 100644
--- a/chrome/browser/net/chrome_network_delegate_unittest.cc
+++ b/chrome/browser/net/chrome_network_delegate_unittest.cc
@@ -387,7 +387,7 @@ class ChromeNetworkDelegateSafeSearchTest :
         &enable_referrers_,
         nullptr,
         &force_google_safe_search_,
-        &force_youtube_safety_mode_,
+        &force_youtube_restrict_,
         nullptr,
         profile_.GetTestingPrefService());
   }
@@ -398,23 +398,20 @@ class ChromeNetworkDelegateSafeSearchTest :
         new ChromeNetworkDelegate(forwarder(), &enable_referrers_,
                                   metrics::UpdateUsagePrefCallbackType()));
     network_delegate->set_force_google_safe_search(&force_google_safe_search_);
-    network_delegate->set_force_youtube_safety_mode(
-        &force_youtube_safety_mode_);
+    network_delegate->set_force_youtube_restrict(&force_youtube_restrict_);
     return std::move(network_delegate);
   }
 
-  void SetSafeSearch(bool google_safe_search,
-                     bool youtube_safety_mode) {
+  void SetSafeSearch(bool google_safe_search, int youtube_restrict) {
     force_google_safe_search_.SetValue(google_safe_search);
-    force_youtube_safety_mode_.SetValue(youtube_safety_mode);
+    force_youtube_restrict_.SetValue(youtube_restrict);
   }
 
   // Does a request to an arbitrary URL and verifies that the SafeSearch
   // enforcement utility functions were called/not called as expected.
-  void QueryURL(bool expect_google_safe_search,
-                bool expect_youtube_safety_mode) {
+  void QueryURL(bool expect_google_safe_search, bool expect_youtube_restrict) {
     safe_search_util::ClearForceGoogleSafeSearchCountForTesting();
-    safe_search_util::ClearForceYouTubeSafetyModeCountForTesting();
+    safe_search_util::ClearForceYouTubeRestrictCountForTesting();
 
     std::unique_ptr<net::URLRequest> request(context_.CreateRequest(
         GURL("http://anyurl.com"), net::DEFAULT_PRIORITY, &delegate_));
@@ -423,14 +420,14 @@ class ChromeNetworkDelegateSafeSearchTest :
     base::RunLoop().RunUntilIdle();
 
     EXPECT_EQ(expect_google_safe_search ? 1 : 0,
-        safe_search_util::GetForceGoogleSafeSearchCountForTesting());
-    EXPECT_EQ(expect_youtube_safety_mode ? 1 : 0,
-        safe_search_util::GetForceYouTubeSafetyModeCountForTesting());
+              safe_search_util::GetForceGoogleSafeSearchCountForTesting());
+    EXPECT_EQ(expect_youtube_restrict ? 1 : 0,
+              safe_search_util::GetForceYouTubeRestrictCountForTesting());
   }
 
  private:
   BooleanPrefMember force_google_safe_search_;
-  BooleanPrefMember force_youtube_safety_mode_;
+  IntegerPrefMember force_youtube_restrict_;
 };
 
 TEST_F(ChromeNetworkDelegateSafeSearchTest, SafeSearch) {
@@ -438,12 +435,16 @@ TEST_F(ChromeNetworkDelegateSafeSearchTest, SafeSearch) {
   SetDelegate(delegate.get());
 
   // Loop over all combinations of the two policies.
-  for (int i = 0; i < 4; i++) {
-    bool google_safe_search = i % 2;
-    bool youtube_safety_mode = i / 2;
-    SetSafeSearch(google_safe_search, youtube_safety_mode);
-
-    QueryURL(google_safe_search, youtube_safety_mode);
+  for (int i = 0; i < 6; i++) {
+    static_assert(safe_search_util::YOUTUBE_RESTRICT_OFF      == 0 &&
+                  safe_search_util::YOUTUBE_RESTRICT_MODERATE == 1 &&
+                  safe_search_util::YOUTUBE_RESTRICT_STRICT   == 2,
+                  "Adjust youtube_restrict to match YouTubeRestrictMode enum");
+    bool google_safe_search = (i / 3) != 0;
+    int youtube_restrict = i % 3;
+    SetSafeSearch(google_safe_search, youtube_restrict);
+
+    QueryURL(google_safe_search, youtube_restrict != 0);
   }
 }