Added a ForceYouTubeRestrict policy and deprecated the old ForceYouTubeSafetyMode policy

chrome/browser/net/safe_search_util_unittest.cc

Index: chrome/browser/net/safe_search_util_unittest.cc
diff --git a/chrome/browser/net/safe_search_util_unittest.cc b/chrome/browser/net/safe_search_util_unittest.cc
index f5e66ce9ad6fa7e183ac5c09705e5d62ece65a00..ada08660707901591396e6fc7a74667cc203c21a 100644
--- a/chrome/browser/net/safe_search_util_unittest.cc
+++ b/chrome/browser/net/safe_search_util_unittest.cc
@@ -143,28 +143,50 @@ TEST_F(SafeSearchUtilTest, AddGoogleSafeSearchParams) {
 TEST_F(SafeSearchUtilTest, SetYoutubeHeader) {
   std::unique_ptr<net::URLRequest> request = CreateYoutubeRequest();
   net::HttpRequestHeaders headers;
-  safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers);
+  safe_search_util::ForceYouTubeRestrict(request.get(), &headers,
+                                         safe_search_util::YTRM_MODERATE);
   std::string value;
-  EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value));
-  EXPECT_EQ("Active", value);
+  EXPECT_TRUE(headers.GetHeader("Youtube-Restrict", &value));
+  EXPECT_EQ("Moderate", value);
 }
 
 TEST_F(SafeSearchUtilTest, OverrideYoutubeHeader) {
   std::unique_ptr<net::URLRequest> request = CreateYoutubeRequest();
   net::HttpRequestHeaders headers;
-  headers.SetHeader("Youtube-Safety-Mode", "Off");
-  safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers);
+  headers.SetHeader("Youtube-Restrict", "Off");
+  safe_search_util::ForceYouTubeRestrict(request.get(), &headers,
+                                         safe_search_util::YTRM_MODERATE);
   std::string value;
-  EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value));
-  EXPECT_EQ("Active", value);
+  EXPECT_TRUE(headers.GetHeader("Youtube-Restrict", &value));
+  EXPECT_EQ("Moderate", value);
 }
 
 TEST_F(SafeSearchUtilTest, DoesntTouchNonYoutubeURL) {
   std::unique_ptr<net::URLRequest> request = CreateNonYoutubeRequest();
   net::HttpRequestHeaders headers;
-  headers.SetHeader("Youtube-Safety-Mode", "Off");
-  safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers);
+  headers.SetHeader("Youtube-Restrict", "Off");
+  safe_search_util::ForceYouTubeRestrict(request.get(), &headers,
+                                         safe_search_util::YTRM_MODERATE);
   std::string value;
-  EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value));
+  EXPECT_TRUE(headers.GetHeader("Youtube-Restrict", &value));
   EXPECT_EQ("Off", value);
 }
+
+TEST_F(SafeSearchUtilTest, SettingOffReturnsIfNothingIsSet) {
+  std::unique_ptr<net::URLRequest> request = CreateYoutubeRequest();
+  net::HttpRequestHeaders headers;
+  safe_search_util::ForceYouTubeRestrict(request.get(), &headers,
+                                         safe_search_util::YTRM_OFF);
+  std::string value;
+  EXPECT_FALSE(headers.GetHeader("Youtube-Restrict", &value));
+}
+
+TEST_F(SafeSearchUtilTest, SettingOffWipesIfSomethingIsSet) {
+  std::unique_ptr<net::URLRequest> request = CreateYoutubeRequest();
+  net::HttpRequestHeaders headers;
+  headers.SetHeader("Youtube-Restrict", "Moderate");
+  safe_search_util::ForceYouTubeRestrict(request.get(), &headers,
+                                         safe_search_util::YTRM_OFF);
+  std::string value;
+  EXPECT_FALSE(headers.GetHeader("Youtube-Restrict", &value));
+}