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   ~SafeSearchUtilTest() override {}
  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   base::MessageLoop message_loop_;
  34   net::TestURLRequestContext context_;
  35 };
  36
  37 // ForceGoogleSafeSearch is already tested quite extensively in
  38 // ChromeNetworkDelegateSafeSearchTest (in chrome_network_delegate_unittest.cc),
  39 // so we won't test it again here.
  40
  41 TEST_F(SafeSearchUtilTest, SetYoutubeHeader) {
  42   scoped_ptr<net::URLRequest> request = CreateYoutubeRequest();
  43   net::HttpRequestHeaders headers;
  44   safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers);
  45   std::string value;
  46   EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value));
  47   EXPECT_EQ("Active", value);
  48 }
  49
  50 TEST_F(SafeSearchUtilTest, OverrideYoutubeHeader) {
  51   scoped_ptr<net::URLRequest> request = CreateYoutubeRequest();
  52   net::HttpRequestHeaders headers;
  53   headers.SetHeader("Youtube-Safety-Mode", "Off");
  54   safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers);
  55   std::string value;
  56   EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value));
  57   EXPECT_EQ("Active", value);
  58 }
  59
  60 TEST_F(SafeSearchUtilTest, DoesntTouchNonYoutubeURL) {
  61   scoped_ptr<net::URLRequest> request = CreateNonYoutubeRequest();
  62   net::HttpRequestHeaders headers;
  63   headers.SetHeader("Youtube-Safety-Mode", "Off");
  64   safe_search_util::ForceYouTubeSafetyMode(request.get(), &headers);
  65   std::string value;
  66   EXPECT_TRUE(headers.GetHeader("Youtube-Safety-Mode", &value));
  67   EXPECT_EQ("Off", value);
  68 }