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

Unified Diff: chrome/browser/net/safe_search_util_unittest.cc

Issue 354183002: Enforce SafetyMode for YouTube if prefs::kForceSafeSearch is on. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: revert string_util.h change Created 6 years, 5 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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « chrome/browser/net/safe_search_util.cc ('k') | chrome/chrome_browser.gypi » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
new file mode 100644
index 0000000000000000000000000000000000000000..112c96137ffc85ad41fbf3d3ced73b5feec54229
--- /dev/null
+++ b/chrome/browser/net/safe_search_util_unittest.cc
@@ -0,0 +1,90 @@
+// Copyright (c) 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/net/safe_search_util.h"
+
+#include "base/message_loop/message_loop.h"
+#include "base/strings/string_piece.h"
+#include "net/http/http_request_headers.h"
+#include "net/url_request/url_request_test_util.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "url/gurl.h"
+
+class SafeSearchUtilTest : public ::testing::Test {
+ protected:
+ SafeSearchUtilTest() {}
+ virtual ~SafeSearchUtilTest() {}
+
+ scoped_ptr<net::URLRequest> CreateYoutubeRequest() {
+ return context_.CreateRequest(GURL("http://www.youtube.com"),
+ net::DEFAULT_PRIORITY,
+ NULL,
+ NULL);
+ }
+
+ scoped_ptr<net::URLRequest> CreateNonYoutubeRequest() {
+ return context_.CreateRequest(GURL("http://www.notyoutube.com"),
+ net::DEFAULT_PRIORITY,
+ NULL,
+ NULL);
+ }
+
+ static void SetCookie(net::HttpRequestHeaders* headers,
+ const std::string& value) {
+ headers->SetHeader(base::StringPiece(net::HttpRequestHeaders::kCookie),
+ base::StringPiece(value));
+ }
+
+ static void CheckHeaders(net::URLRequest* request,
+ const std::string& header_string_original,
+ const std::string& header_string_expected) {
+ net::HttpRequestHeaders headers;
+ SetCookie(&headers, header_string_original);
+ safe_search_util::ForceYouTubeSafetyMode(request, &headers);
+
+ net::HttpRequestHeaders headers_expected;
+ SetCookie(&headers_expected, header_string_expected);
+ EXPECT_EQ(headers_expected.ToString(), headers.ToString());
+ }
+
+ base::MessageLoop message_loop_;
+ net::TestURLRequestContext context_;
+};
+
+// ForceGoogleSafeSearch is already tested quite extensively in
+// ChromeNetworkDelegateSafeSearchTest (in chrome_network_delegate_unittest.cc),
+// so we won't test it again here.
+
+TEST_F(SafeSearchUtilTest, CreateYoutubePrefCookie) {
+ scoped_ptr<net::URLRequest> request = CreateYoutubeRequest();
+ CheckHeaders(request.get(),
+ "OtherCookie=value",
+ "OtherCookie=value; PREF=f2=8000000");
+}
+
+TEST_F(SafeSearchUtilTest, ModifyYoutubePrefCookie) {
+ scoped_ptr<net::URLRequest> request = CreateYoutubeRequest();
+ CheckHeaders(request.get(),
+ "PREF=f1=123; OtherCookie=value",
+ "PREF=f1=123&f2=8000000; OtherCookie=value");
+ CheckHeaders(request.get(),
+ "PREF=",
+ "PREF=f2=8000000");
+ CheckHeaders(request.get(),
+ "PREF=\"\"",
+ "PREF=\"f2=8000000\"");
+ CheckHeaders(request.get(),
+ "PREF=f1=123&f2=4321&foo=bar",
+ "PREF=f1=123&f2=8004321&foo=bar");
+ CheckHeaders(request.get(),
+ "PREF=\"f1=1&f2=4321\"; OtherCookie=value",
+ "PREF=\"f1=1&f2=8004321\"; OtherCookie=value");
+}
+
+TEST_F(SafeSearchUtilTest, DoesntTouchNonYoutubeURL) {
+ scoped_ptr<net::URLRequest> request = CreateNonYoutubeRequest();
+ CheckHeaders(request.get(),
+ "PREF=f2=0",
+ "PREF=f2=0");
+}
« no previous file with comments | « chrome/browser/net/safe_search_util.cc ('k') | chrome/chrome_browser.gypi » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698