// Copyright 2022 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/lookalikes/lookalike_test_helper.h"

#include "base/memory/raw_ptr.h"
#include "components/lookalikes/core/lookalike_url_util.h"
#include "components/lookalikes/core/safety_tip_test_utils.h"
#include "components/lookalikes/core/safety_tips_config.h"
#include "components/url_formatter/spoof_checks/idn_spoof_checker.h"
#include "components/url_formatter/spoof_checks/top_domains/test_domains_trie.h"
#include "components/url_formatter/spoof_checks/top_domains/test_top_bucket_domains.h"
#include "services/metrics/public/cpp/ukm_builders.h"
#include "testing/gtest/include/gtest/gtest.h"

LookalikeTestHelper::LookalikeTestHelper(ukm::TestUkmRecorder* ukm_recorder)
    : ukm_recorder_(ukm_recorder) {}

// static
void LookalikeTestHelper::SetUpLookalikeTestParams() {
  // Use test top domain lists instead of the actual list.
  url_formatter::IDNSpoofChecker::HuffmanTrieParams trie_params{
      url_formatter::kTestTopDomainsHuffmanTree,
      url_formatter::kTestTopDomainsTrie,
      url_formatter::kTestTopDomainsTrieBits,
      url_formatter::kTestTopDomainsRootPosition};
  url_formatter::IDNSpoofChecker::SetTrieParamsForTesting(trie_params);

  // Use test top bucket domain skeletons instead of the actual list.
  lookalikes::TopBucketDomainsParams top_bucket_params{
      test_top_bucket_domains::kTopBucketEditDistanceSkeletons,
      test_top_bucket_domains::kNumTopBucketEditDistanceSkeletons};
  lookalikes::SetTopBucketDomainsParamsForTesting(top_bucket_params);

  lookalikes::InitializeSafetyTipConfig();
}

// static
void LookalikeTestHelper::TearDownLookalikeTestParams() {
  url_formatter::IDNSpoofChecker::RestoreTrieParamsForTesting();
  lookalikes::ResetTopBucketDomainsParamsForTesting();
}

void LookalikeTestHelper::CheckSafetyTipUkmCount(
    size_t expected_event_count) const {
  std::vector<raw_ptr<const ukm::mojom::UkmEntry, VectorExperimental>> entries =
      ukm_recorder_->GetEntriesByName(
          ukm::builders::Security_SafetyTip::kEntryName);
  ASSERT_EQ(expected_event_count, entries.size());
}

void LookalikeTestHelper::CheckInterstitialUkmCount(
    size_t expected_event_count) const {
  std::vector<raw_ptr<const ukm::mojom::UkmEntry, VectorExperimental>> entries =
      ukm_recorder_->GetEntriesByName(
          ukm::builders::LookalikeUrl_NavigationSuggestion::kEntryName);
  ASSERT_EQ(expected_event_count, entries.size());
}

void LookalikeTestHelper::CheckNoLookalikeUkm() const {
  CheckSafetyTipUkmCount(0);
  CheckInterstitialUkmCount(0);
}