chrome/browser/reputation/reputation_service_unittest.cc

// Copyright 2019 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/reputation/reputation_service.h"

#include <algorithm>
#include <string>
#include <utility>
#include <vector>

#include "chrome/test/base/chrome_render_view_host_test_harness.h"
#include "components/reputation/core/safety_tip_test_utils.h"
#include "components/reputation/core/safety_tips_config.h"
#include "components/security_state/core/security_state.h"
#include "testing/gtest/include/gtest/gtest.h"

class ReputationServiceTest : public ChromeRenderViewHostTestHarness {
 public:
  ReputationServiceTest(const ReputationServiceTest&) = delete;
  ReputationServiceTest& operator=(const ReputationServiceTest&) = delete;

 protected:
  ReputationServiceTest() {}
  ~ReputationServiceTest() override {}
};

// Test that the blocklist blocks patterns as expected.
TEST_F(ReputationServiceTest, BlocklistTest) {
  reputation::SetSafetyTipBadRepPatterns(
      {"domain.test/", "directory.test/foo/", "path.test/foo/bar.html",
       "query.test/foo/bar.html?baz=test", "sub.subdomain.test/"});

  const std::vector<std::pair<std::string, security_state::SafetyTipStatus>>
      kTests = {
          {"http://unblocked.test", security_state::SafetyTipStatus::kNone},
          {"http://unblocked.test/foo", security_state::SafetyTipStatus::kNone},
          {"http://unblocked.test/foo.html?bar=baz",
           security_state::SafetyTipStatus::kNone},

          {"http://sub.domain.test",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://domain.test",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://domain.test/foo",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://domain.test/foo/bar",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://domain.test/foo.html?bar=baz",
           security_state::SafetyTipStatus::kBadReputation},

          {"http://directory.test", security_state::SafetyTipStatus::kNone},
          {"http://directory.test/bar", security_state::SafetyTipStatus::kNone},
          {"http://directory.test/bar/foo.html",
           security_state::SafetyTipStatus::kNone},
          {"http://directory.test/foo", security_state::SafetyTipStatus::kNone},
          {"http://directory.test/foo/bar/",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://directory.test/foo/bar.html?bar=baz",
           security_state::SafetyTipStatus::kBadReputation},

          {"http://path.test", security_state::SafetyTipStatus::kNone},
          {"http://path.test/foo", security_state::SafetyTipStatus::kNone},
          {"http://path.test/foo/bar/", security_state::SafetyTipStatus::kNone},
          {"http://path.test/foo/bar.htm",
           security_state::SafetyTipStatus::kNone},
          {"http://path.test/foo/bar.html",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://path.test/foo/bar.html?bar=baz",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://path.test/bar/foo.html",
           security_state::SafetyTipStatus::kNone},

          {"http://query.test", security_state::SafetyTipStatus::kNone},
          {"http://query.test/foo", security_state::SafetyTipStatus::kNone},
          {"http://query.test/foo/bar/",
           security_state::SafetyTipStatus::kNone},
          {"http://query.test/foo/bar.html",
           security_state::SafetyTipStatus::kNone},
          {"http://query.test/foo/bar.html?baz=test",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://query.test/foo/bar.html?baz=test&a=1",
           security_state::SafetyTipStatus::kNone},
          {"http://query.test/foo/bar.html?baz=no",
           security_state::SafetyTipStatus::kNone},

          {"http://subdomain.test", security_state::SafetyTipStatus::kNone},
          {"http://sub.subdomain.test",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://sub.subdomain.test/foo/bar",
           security_state::SafetyTipStatus::kBadReputation},
          {"http://sub.subdomain.test/foo.html?bar=baz",
           security_state::SafetyTipStatus::kBadReputation},
      };

  for (auto test : kTests) {
    EXPECT_EQ(reputation::GetSafetyTipUrlBlockType(GURL(test.first)),
              test.second);
  }
}