Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(313)

Side by Side Diff: chrome/browser/net/safe_search_util_unittest.cc

Issue 2804023003: Network traffic annotation added to chrome/browser/ unittests. (Closed)
Patch Set: Another missing header added. Created 3 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698