chromium/chrome/browser/new_tab_page/modules/safe_browsing/safe_browsing_handler_unittest.cc

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/new_tab_page/modules/safe_browsing/safe_browsing_handler.h"
#include "base/memory/raw_ptr.h"
#include "base/strings/string_number_conversions.h"
#include "base/test/mock_callback.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "base/time/time.h"
#include "chrome/browser/safe_browsing/safe_browsing_metrics_collector_factory.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/testing_profile.h"
#include "chrome/test/base/testing_profile_manager.h"
#include "components/prefs/pref_registry_simple.h"
#include "components/safe_browsing/core/browser/safe_browsing_metrics_collector.h"
#include "components/safe_browsing/core/common/safe_browsing_policy_handler.h"
#include "components/safe_browsing/core/common/safe_browsing_prefs.h"
#include "components/search/ntp_features.h"
#include "components/sync_preferences/testing_pref_service_syncable.h"
#include "content/public/browser/browser_context.h"
#include "content/public/test/browser_task_environment.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "testing/gtest/include/gtest/gtest.h"

SafeBrowsingMetricsCollector;
SafeBrowsingMetricsCollectorFactory;
SafeBrowsingState;

namespace ntp {

class MockSafeBrowsingMetricsCollector : public SafeBrowsingMetricsCollector { … };

class SafeBrowsingHandlerTest : public ::testing::Test { … };

TEST_F(SafeBrowsingHandlerTest, CanShowModule_DoesNotShowForManaged) { … }

TEST_F(SafeBrowsingHandlerTest, CanShowModule_DoesNotShowForEnhanced) { … }

TEST_F(SafeBrowsingHandlerTest, CanShowModule_Succeeds) { … }

TEST_F(SafeBrowsingHandlerTest, ProcessModuleClick_DisablesModule) { … }

TEST_F(SafeBrowsingHandlerTest, DismissRestore_DisablesModuleInBetween) { … }

TEST_F(SafeBrowsingHandlerTest, Cooldown_DisablesModuleInBetween) { … }

}  // namespace ntp