chrome/browser/browser_switcher/browser_switcher_sitelist_unittest.cc

// Copyright 2018 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/browser_switcher/browser_switcher_sitelist.h"

#include "base/values.h"
#include "chrome/browser/browser_switcher/browser_switcher_prefs.h"
#include "chrome/browser/browser_switcher/ieem_sitelist_parser.h"
#include "components/prefs/pref_registry_simple.h"
#include "components/prefs/testing_pref_service.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

namespace browser_switcher {

namespace {

std::unique_ptr<base::Value> StringArrayToValue(
    const std::vector<const char*>& strings) {
  std::vector<base::Value> values(strings.size());
  std::transform(strings.begin(), strings.end(), values.begin(),
                 [](const char* s) { return base::Value(s); });
  return std::make_unique<base::Value>(values);
}

}  // namespace

class BrowserSwitcherSitelistTest : public testing::Test {
 public:
  void Initialize(const std::vector<const char*>& url_list,
                  const std::vector<const char*>& url_greylist) {
    prefs_.registry()->RegisterListPref(prefs::kUrlList);
    prefs_.registry()->RegisterListPref(prefs::kUrlGreylist);
    prefs_.SetManagedPref(prefs::kUrlList, StringArrayToValue(url_list));
    prefs_.SetManagedPref(prefs::kUrlGreylist,
                          StringArrayToValue(url_greylist));
    sitelist_ = std::make_unique<BrowserSwitcherSitelistImpl>(&prefs_);
  }

  TestingPrefServiceSimple* prefs() { return &prefs_; }
  BrowserSwitcherSitelist* sitelist() { return sitelist_.get(); }

 private:
  TestingPrefServiceSimple prefs_;
  std::unique_ptr<BrowserSwitcherSitelist> sitelist_;
};

TEST_F(BrowserSwitcherSitelistTest, ShouldRedirectWildcard) {
  // A "*" by itself means everything matches.
  Initialize({"*"}, {});
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("https://example.com/foobar/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/foobar/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://google.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldRedirectHost) {
  // A string without slashes means compare the URL's host (case-insensitive).
  Initialize({"example.com"}, {});
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("https://example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://subdomain.example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/foobar/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://google.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://example.ca/")));

  // For backwards compatibility, this should also match, even if it's not the
  // same host.
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("https://notexample.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldRedirectHostNotLowerCase) {
  // Host is not in lowercase form, but we compare ignoring case.
  Initialize({"eXaMpLe.CoM"}, {});
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldRedirectWrongScheme) {
  Initialize({"example.com"}, {});
  // Scheme is not one of 'http', 'https' or 'file'.
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("ftp://example.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldRedirectPrefix) {
  // A string with slashes means check if it's a prefix (case-sensitive).
  Initialize({"http://example.com/foobar"}, {});
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/foobar")));
  EXPECT_TRUE(
      sitelist()->ShouldSwitch(GURL("http://example.com/foobar/subroute/")));
  EXPECT_TRUE(
      sitelist()->ShouldSwitch(GURL("http://example.com/foobar#fragment")));
  EXPECT_TRUE(
      sitelist()->ShouldSwitch(GURL("http://example.com/foobar?query=param")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("https://example.com/foobar")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("HTTP://EXAMPLE.COM/FOOBAR")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://subdomain.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://google.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldRedirectInvertedMatch) {
  // The most specific (i.e., longest string) rule should have priority.
  Initialize({"!subdomain.example.com", "example.com"}, {});
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://subdomain.example.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldRedirectGreylist) {
  // The most specific (i.e., longest string) rule should have priority.
  Initialize({"example.com"}, {"http://example.com/login/"});
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://example.com/login/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldRedirectGreylistWildcard) {
  Initialize({"*"}, {"*"});
  // If both are wildcards, prefer the greylist.
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldMatchAnySchema) {
  // URLs formatted like these don't include a schema, so should match both HTTP
  // and HTTPS.
  Initialize({"//example.com", "reddit.com/r/funny"}, {});
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/something")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("https://example.com/something")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("file://example.com/foobar/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("https://foo.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("mailto://example.com")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://bad.com/example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://bad.com//example.com/")));
  EXPECT_FALSE(
      sitelist()->ShouldSwitch(GURL("http://bad.com/hackme.html?example.com")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://reddit.com/r/funny")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("https://reddit.com/r/funny")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://reddit.com/r/pics")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("https://reddit.com/r/pics")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldPickUpPrefChanges) {
  Initialize({}, {});
  prefs()->SetManagedPref(prefs::kUrlList, StringArrayToValue({"example.com"}));
  prefs()->SetManagedPref(prefs::kUrlGreylist,
                          StringArrayToValue({"foo.example.com"}));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://bar.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://foo.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://google.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, ShouldIgnoreNonManagedPrefs) {
  Initialize({}, {});

  prefs()->Set(prefs::kUrlList, *StringArrayToValue({"example.com"}));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://example.com/")));

  prefs()->SetManagedPref(prefs::kUrlList, StringArrayToValue({"example.com"}));
  prefs()->Set(prefs::kUrlGreylist,
               *StringArrayToValue({"morespecific.example.com"}));
  EXPECT_TRUE(
      sitelist()->ShouldSwitch(GURL("http://morespecific.example.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, SetIeemSitelist) {
  Initialize({}, {});
  ParsedXml ieem;
  ieem.sitelist = {"example.com"};
  ieem.greylist = {"foo.example.com"};
  sitelist()->SetIeemSitelist(std::move(ieem));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://bar.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://foo.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://google.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, SetExternalSitelist) {
  Initialize({}, {});
  ParsedXml external;
  external.sitelist = {"example.com"};
  external.greylist = {"foo.example.com"};
  sitelist()->SetExternalSitelist(std::move(external));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://bar.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://foo.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://google.com/")));
}

TEST_F(BrowserSwitcherSitelistTest, All3Sources) {
  Initialize({"google.com"}, {"mail.google.com"});
  ParsedXml ieem;
  ieem.sitelist = {"example.com"};
  ieem.greylist = {"foo.example.com"};
  sitelist()->SetIeemSitelist(std::move(ieem));
  ParsedXml external;
  external.sitelist = {"yahoo.com"};
  external.greylist = {"finance.yahoo.com"};
  sitelist()->SetExternalSitelist(std::move(external));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://google.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://drive.google.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://mail.google.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://bar.example.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://foo.example.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://yahoo.com/")));
  EXPECT_TRUE(sitelist()->ShouldSwitch(GURL("http://news.yahoo.com/")));
  EXPECT_FALSE(sitelist()->ShouldSwitch(GURL("http://finance.yahoo.com/")));
}

}  // namespace browser_switcher