components/subresource_filter/content/browser/content_activation_list_utils_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 "components/subresource_filter/content/browser/content_activation_list_utils.h"
#include "base/test/scoped_feature_list.h"
#include "components/subresource_filter/core/browser/subresource_filter_features.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace subresource_filter {

enum class AdBlockOnAbusiveSitesTest { kEnabled, kDisabled };

TEST(ContentActivationListUtilsTest, GetListForThreatTypeAndMetadata) {
  typedef safe_browsing::SubresourceFilterType Type;
  typedef safe_browsing::SubresourceFilterLevel Level;
  const struct {
    std::string test_id;
    safe_browsing::SBThreatType sb_threat_type;
    safe_browsing::ThreatPatternType threat_pattern_type;
    safe_browsing::SubresourceFilterMatch subresource_filter_match;
    AdBlockOnAbusiveSitesTest adblock_on_abusive_sites;
    ActivationList expected_activation_list;
    bool expected_warning;
  } kTestCases[]{
      {"Phishing_With_SocialEngineeringAds",
       safe_browsing::SB_THREAT_TYPE_URL_PHISHING,
       safe_browsing::ThreatPatternType::SOCIAL_ENGINEERING_ADS,
       {},
       AdBlockOnAbusiveSitesTest::kDisabled,
       ActivationList::SOCIAL_ENG_ADS_INTERSTITIAL,
       false},
      {"Phishing_Without_SocialEngineeringAds",
       safe_browsing::SB_THREAT_TYPE_URL_PHISHING,
       safe_browsing::ThreatPatternType::NONE,
       {},
       AdBlockOnAbusiveSitesTest::kDisabled,
       ActivationList::PHISHING_INTERSTITIAL,
       false},
      {"Empty_SubresourceFilterMatch",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {},
       AdBlockOnAbusiveSitesTest::kDisabled,
       ActivationList::SUBRESOURCE_FILTER,
       false},
      {"BetterAds_Warn_DisableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::BETTER_ADS, Level::WARN}}, base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kDisabled,
       ActivationList::BETTER_ADS,
       true},
      {"BetterAds_Enforce_DisableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::BETTER_ADS, Level::ENFORCE}}, base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kDisabled,
       ActivationList::BETTER_ADS,
       false},
      {"Abusive_Warn_DisableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::ABUSIVE, Level::WARN}}, base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kDisabled,
       ActivationList::NONE,
       false},
      {"Abusive_Enforce_DisableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::ABUSIVE, Level::ENFORCE}}, base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kDisabled,
       ActivationList::NONE,
       false},
      {"BetterAds_Warn_EnableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::BETTER_ADS, Level::WARN}}, base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kEnabled,
       ActivationList::BETTER_ADS,
       true},
      {"BetterAds_Enforce_EnableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::BETTER_ADS, Level::ENFORCE}}, base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kEnabled,
       ActivationList::BETTER_ADS,
       false},
      {"Abusive_Warn_EnableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::ABUSIVE, Level::WARN}}, base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kEnabled,
       ActivationList::ABUSIVE,
       true},
      {"Abusive_Enforce_EnableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::ABUSIVE, Level::ENFORCE}}, base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kEnabled,
       ActivationList::ABUSIVE,
       false},
      {"BetterAds_Warn_Abusive_Warn_EnableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::ABUSIVE, Level::WARN}, {Type::BETTER_ADS, Level::WARN}},
        base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kDisabled,
       ActivationList::BETTER_ADS,
       true},
      {"BetterAds_Warn_Abusive_Enforce_EnableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::ABUSIVE, Level::ENFORCE}, {Type::BETTER_ADS, Level::WARN}},
        base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kEnabled,
       ActivationList::ABUSIVE,
       false},
      {"BetterAds_Enforce_Abusive_Warn_EnableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::ABUSIVE, Level::WARN}, {Type::BETTER_ADS, Level::ENFORCE}},
        base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kEnabled,
       ActivationList::BETTER_ADS,
       false},
      {"BetterAds_Enforce_Abusive_Enforce_EnableAdBlockOnAbusiveSites",
       safe_browsing::SB_THREAT_TYPE_SUBRESOURCE_FILTER,
       safe_browsing::ThreatPatternType::NONE,
       {{{Type::ABUSIVE, Level::ENFORCE}, {Type::BETTER_ADS, Level::ENFORCE}},
        base::KEEP_FIRST_OF_DUPES},
       AdBlockOnAbusiveSitesTest::kEnabled,
       ActivationList::BETTER_ADS,
       false}};

  for (const auto& test_case : kTestCases) {
    SCOPED_TRACE(::testing::Message() << "test_id = " << test_case.test_id);
    base::test::ScopedFeatureList scoped_feature_list;
    scoped_feature_list.InitWithFeatureState(
        kFilterAdsOnAbusiveSites, test_case.adblock_on_abusive_sites ==
                                      AdBlockOnAbusiveSitesTest::kEnabled);
    safe_browsing::ThreatMetadata metadata;
    metadata.threat_pattern_type = test_case.threat_pattern_type;
    metadata.subresource_filter_match = test_case.subresource_filter_match;
    bool warning = false;
    EXPECT_EQ(test_case.expected_activation_list,
              GetListForThreatTypeAndMetadata(test_case.sb_threat_type,
                                              metadata, &warning));
    EXPECT_EQ(test_case.expected_warning, warning);
  }
}

}  // namespace subresource_filter