OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/net/safe_search_util.h" | |
6 | |
7 #include "base/message_loop/message_loop.h" | |
8 #include "base/strings/string_piece.h" | |
9 #include "net/http/http_request_headers.h" | |
10 #include "net/url_request/url_request_test_util.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 #include "url/gurl.h" | |
13 | |
14 class SafeSearchUtilTest : public ::testing::Test { | |
15 protected: | |
16 SafeSearchUtilTest() {} | |
17 virtual ~SafeSearchUtilTest() {} | |
18 | |
19 scoped_ptr<net::URLRequest> CreateYoutubeRequest() { | |
20 return context_.CreateRequest(GURL("http://www.youtube.com"), | |
21 net::DEFAULT_PRIORITY, | |
22 NULL, | |
23 NULL); | |
24 } | |
25 | |
26 scoped_ptr<net::URLRequest> CreateNonYoutubeRequest() { | |
27 return context_.CreateRequest(GURL("http://www.notyoutube.com"), | |
28 net::DEFAULT_PRIORITY, | |
29 NULL, | |
30 NULL); | |
31 } | |
32 | |
33 static void SetCookie(net::HttpRequestHeaders* headers, | |
34 const std::string& value) { | |
35 headers->SetHeader(base::StringPiece(net::HttpRequestHeaders::kCookie), | |
36 base::StringPiece(value)); | |
37 } | |
38 | |
39 static void CheckHeaders(net::URLRequest* request, | |
40 const std::string& header_string_original, | |
41 const std::string& header_string_expected) { | |
42 net::HttpRequestHeaders headers; | |
43 SetCookie(&headers, header_string_original); | |
44 safe_search_util::ForceYouTubeSafetyMode(request, &headers); | |
45 | |
46 net::HttpRequestHeaders headers_expected; | |
47 SetCookie(&headers_expected, header_string_expected); | |
48 EXPECT_EQ(headers_expected.ToString(), headers.ToString()); | |
49 } | |
50 | |
51 base::MessageLoop message_loop_; | |
52 net::TestURLRequestContext context_; | |
53 }; | |
54 | |
55 // ForceGoogleSafeSearch is already tested quite extensively in | |
56 // ChromeNetworkDelegateSafeSearchTest (in chrome_network_delegate_unittest.cc), | |
57 // so we won't test it again here. | |
58 | |
59 TEST_F(SafeSearchUtilTest, CreateYoutubePrefCookie) { | |
60 scoped_ptr<net::URLRequest> request = CreateYoutubeRequest(); | |
61 CheckHeaders(request.get(), | |
62 "OtherCookie=value", | |
63 "OtherCookie=value; PREF=f2=8000000"); | |
64 } | |
65 | |
66 TEST_F(SafeSearchUtilTest, ModifyYoutubePrefCookie) { | |
67 scoped_ptr<net::URLRequest> request = CreateYoutubeRequest(); | |
68 CheckHeaders(request.get(), | |
69 "PREF=f1=123; OtherCookie=value", | |
70 "PREF=f1=123&f2=8000000; OtherCookie=value"); | |
71 CheckHeaders(request.get(), | |
72 "PREF=", | |
73 "PREF=f2=8000000"); | |
74 CheckHeaders(request.get(), | |
75 "PREF=\"\"", | |
76 "PREF=\"f2=8000000\""); | |
77 CheckHeaders(request.get(), | |
78 "PREF=f1=123&f2=4321&foo=bar", | |
79 "PREF=f1=123&f2=8004321&foo=bar"); | |
80 CheckHeaders(request.get(), | |
81 "PREF=\"f1=1&f2=4321\"; OtherCookie=value", | |
82 "PREF=\"f1=1&f2=8004321\"; OtherCookie=value"); | |
83 } | |
84 | |
85 TEST_F(SafeSearchUtilTest, DoesntTouchNonYoutubeURL) { | |
86 scoped_ptr<net::URLRequest> request = CreateNonYoutubeRequest(); | |
87 CheckHeaders(request.get(), | |
88 "PREF=f2=0", | |
89 "PREF=f2=0"); | |
90 } | |
OLD | NEW |