chrome/browser/net/safe_search_util_unittest.cc

   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 }