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 b0d9722b7be5eff8604253ae6ffe869b202178f4..f5e66ce9ad6fa7e183ac5c09705e5d62ece65a00 100644 |
--- a/chrome/browser/net/safe_search_util_unittest.cc |
+++ b/chrome/browser/net/safe_search_util_unittest.cc |
@@ -143,31 +143,28 @@ TEST_F(SafeSearchUtilTest, AddGoogleSafeSearchParams) { |
TEST_F(SafeSearchUtilTest, SetYoutubeHeader) { |
std::unique_ptr<net::URLRequest> request = CreateYoutubeRequest(); |
net::HttpRequestHeaders headers; |
- safe_search_util::ForceYouTubeRestrict( |
- request.get(), &headers, safe_search_util::YOUTUBE_RESTRICT_MODERATE); |
+ safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers); |
std::string value; |
- EXPECT_TRUE(headers.GetHeader("Youtube-Restrict", &value)); |
- EXPECT_EQ("Moderate", value); |
+ EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value)); |
+ EXPECT_EQ("Active", value); |
} |
TEST_F(SafeSearchUtilTest, OverrideYoutubeHeader) { |
std::unique_ptr<net::URLRequest> request = CreateYoutubeRequest(); |
net::HttpRequestHeaders headers; |
- headers.SetHeader("Youtube-Restrict", "Off"); |
- safe_search_util::ForceYouTubeRestrict( |
- request.get(), &headers, safe_search_util::YOUTUBE_RESTRICT_MODERATE); |
+ headers.SetHeader("Youtube-Safety-Mode", "Off"); |
+ safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers); |
std::string value; |
- EXPECT_TRUE(headers.GetHeader("Youtube-Restrict", &value)); |
- EXPECT_EQ("Moderate", value); |
+ EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value)); |
+ EXPECT_EQ("Active", value); |
} |
TEST_F(SafeSearchUtilTest, DoesntTouchNonYoutubeURL) { |
std::unique_ptr<net::URLRequest> request = CreateNonYoutubeRequest(); |
net::HttpRequestHeaders headers; |
- headers.SetHeader("Youtube-Restrict", "Off"); |
- safe_search_util::ForceYouTubeRestrict( |
- request.get(), &headers, safe_search_util::YOUTUBE_RESTRICT_MODERATE); |
+ headers.SetHeader("Youtube-Safety-Mode", "Off"); |
+ safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers); |
std::string value; |
- EXPECT_TRUE(headers.GetHeader("Youtube-Restrict", &value)); |
+ EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value)); |
EXPECT_EQ("Off", value); |
} |