chrome/browser/page_info/merchant_trust_service_delegate_unittest.cc

// Copyright 2025 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/page_info/merchant_trust_service_delegate.h"

#include "base/test/bind.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "chrome/browser/ui/hats/hats_service_factory.h"
#include "chrome/browser/ui/hats/mock_hats_service.h"
#include "chrome/test/base/testing_profile.h"
#include "components/page_info/core/features.h"
#include "components/site_engagement/content/site_engagement_service.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

using testing::_;

class MerchantTrustServiceDelegateTest : public ::testing::Test {
 public:
  void SetUp() override {
    mock_hats_service_ = static_cast<MockHatsService*>(
        HatsServiceFactory::GetInstance()->SetTestingFactoryAndUse(
            profile(), base::BindRepeating(&BuildMockHatsService)));
    delegate_ = std::make_unique<MerchantTrustServiceDelegate>(profile());
  }

  TestingProfile* profile() { return &profile_; }
  MerchantTrustServiceDelegate* delegate() { return delegate_.get(); }
  MockHatsService* mock_hats_service() { return mock_hats_service_; }

 private:
  content::BrowserTaskEnvironment task_environment_;
  TestingProfile profile_;
  std::unique_ptr<MerchantTrustServiceDelegate> delegate_;
  raw_ptr<MockHatsService> mock_hats_service_;
};

TEST_F(MerchantTrustServiceDelegateTest, ControlSurvey) {
  base::test::ScopedFeatureList feature_list;
  EXPECT_CALL(
      *mock_hats_service(),
      LaunchSurvey(kHatsSurveyTriggerMerchantTrustEvaluationControlSurvey, _, _,
                   _, _, _, _));

  feature_list.InitWithFeaturesAndParameters(
      {{page_info::kMerchantTrustEvaluationControlSurvey, {}}},
      {page_info::kMerchantTrust});

  delegate()->ShowEvaluationSurvey();
}

TEST_F(MerchantTrustServiceDelegateTest, ExperimentSurvey) {
  base::test::ScopedFeatureList feature_list;
  EXPECT_CALL(
      *mock_hats_service(),
      LaunchSurvey(kHatsSurveyTriggerMerchantTrustEvaluationExperimentSurvey, _,
                   _, _, _, _, _));

  std::vector<base::test::FeatureRefAndParams> enabled_features = {
      {page_info::kMerchantTrust,
       {{page_info::kMerchantTrustForceShowUIForTestingName, "true"}}},
      {page_info::kMerchantTrustEvaluationExperimentSurvey, {}}};
  feature_list.InitWithFeaturesAndParameters(enabled_features, {});

  delegate()->ShowEvaluationSurvey();
}

TEST_F(MerchantTrustServiceDelegateTest, GetSiteEngagementScore) {
  site_engagement::SiteEngagementService::Get(profile())->ResetBaseScoreForURL(
      GURL("https://highengagement.com"), 20);
  EXPECT_EQ(
      delegate()->GetSiteEngagementScore(GURL("https://highengagement.com")),
      20);
}