| 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);
|
| }
|
|
|