| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/net/safe_search_util.h" | 5 #include "chrome/browser/net/safe_search_util.h" |
| 6 | 6 |
| 7 #include "base/message_loop/message_loop.h" | 7 #include "base/message_loop/message_loop.h" |
| 8 #include "base/strings/string_piece.h" | 8 #include "base/strings/string_piece.h" |
| 9 #include "chrome/common/url_constants.h" | 9 #include "chrome/common/url_constants.h" |
| 10 #include "net/http/http_request_headers.h" | 10 #include "net/http/http_request_headers.h" |
| 11 #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
| 11 #include "net/url_request/url_request_test_util.h" | 12 #include "net/url_request/url_request_test_util.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 13 #include "url/gurl.h" | 14 #include "url/gurl.h" |
| 14 | 15 |
| 15 class SafeSearchUtilTest : public ::testing::Test { | 16 class SafeSearchUtilTest : public ::testing::Test { |
| 16 protected: | 17 protected: |
| 17 SafeSearchUtilTest() {} | 18 SafeSearchUtilTest() {} |
| 18 ~SafeSearchUtilTest() override {} | 19 ~SafeSearchUtilTest() override {} |
| 19 | 20 |
| 20 std::unique_ptr<net::URLRequest> CreateRequest(const std::string& url) { | 21 std::unique_ptr<net::URLRequest> CreateRequest(const std::string& url) { |
| 21 return context_.CreateRequest(GURL(url), net::DEFAULT_PRIORITY, NULL); | 22 return context_.CreateRequest(GURL(url), net::DEFAULT_PRIORITY, NULL, |
| 23 TRAFFIC_ANNOTATION_FOR_TESTS); |
| 22 } | 24 } |
| 23 | 25 |
| 24 std::unique_ptr<net::URLRequest> CreateYoutubeRequest() { | 26 std::unique_ptr<net::URLRequest> CreateYoutubeRequest() { |
| 25 return CreateRequest("http://www.youtube.com"); | 27 return CreateRequest("http://www.youtube.com"); |
| 26 } | 28 } |
| 27 | 29 |
| 28 std::unique_ptr<net::URLRequest> CreateNonYoutubeRequest() { | 30 std::unique_ptr<net::URLRequest> CreateNonYoutubeRequest() { |
| 29 return CreateRequest("http://www.notyoutube.com"); | 31 return CreateRequest("http://www.notyoutube.com"); |
| 30 } | 32 } |
| 31 | 33 |
| (...skipping 132 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 164 TEST_F(SafeSearchUtilTest, DoesntTouchNonYoutubeURL) { | 166 TEST_F(SafeSearchUtilTest, DoesntTouchNonYoutubeURL) { |
| 165 std::unique_ptr<net::URLRequest> request = CreateNonYoutubeRequest(); | 167 std::unique_ptr<net::URLRequest> request = CreateNonYoutubeRequest(); |
| 166 net::HttpRequestHeaders headers; | 168 net::HttpRequestHeaders headers; |
| 167 headers.SetHeader("Youtube-Restrict", "Off"); | 169 headers.SetHeader("Youtube-Restrict", "Off"); |
| 168 safe_search_util::ForceYouTubeRestrict( | 170 safe_search_util::ForceYouTubeRestrict( |
| 169 request.get(), &headers, safe_search_util::YOUTUBE_RESTRICT_MODERATE); | 171 request.get(), &headers, safe_search_util::YOUTUBE_RESTRICT_MODERATE); |
| 170 std::string value; | 172 std::string value; |
| 171 EXPECT_TRUE(headers.GetHeader("Youtube-Restrict", &value)); | 173 EXPECT_TRUE(headers.GetHeader("Youtube-Restrict", &value)); |
| 172 EXPECT_EQ("Off", value); | 174 EXPECT_EQ("Off", value); |
| 173 } | 175 } |
| OLD | NEW |