Added a ForceYouTubeRestrict policy and deprecated the old ForceYouTubeSafetyMode policy

chrome/browser/net/safe_search_util_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 "chrome/browser/net/safe_search_util.h"
 
 #include "base/message_loop/message_loop.h"
 #include "base/strings/string_piece.h"
 #include "chrome/common/url_constants.h"
 #include "net/http/http_request_headers.h"
@@ skipping 125 matching lines @@
 
   // Test with percent-encoded data (%26 is &)
   CheckAddedParameters("http://google.com/?q=%26%26%26&" + kSsuiParameter +
                        "&" + kSafeParameter + "&param=%26%26%26",
                        "q=%26%26%26&param=%26%26%26&" + kBothParameters);
 }
 
 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::YouTubeRestrictMode::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::YouTubeRestrictMode::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::YouTubeRestrictMode::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::YouTubeRestrictMode::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::YouTubeRestrictMode::OFF);
+  std::string value;
+  EXPECT_FALSE(headers.GetHeader("Youtube-Restrict", &value));
+}