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

#include "components/user_education/common/feature_promo/feature_promo_specification.h"

#include "base/feature_list.h"
#include "base/test/bind.h"
#include "base/test/scoped_feature_list.h"
#include "components/strings/grit/components_strings.h"
#include "components/user_education/common/anchor_element_provider.h"
#include "components/user_education/common/feature_promo/feature_promo_handle.h"
#include "components/user_education/common/help_bubble/help_bubble_params.h"
#include "components/user_education/common/user_education_features.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/interaction/element_identifier.h"
#include "ui/base/interaction/element_test_util.h"
#include "ui/base/interaction/element_tracker.h"

namespace user_education {

namespace {
BASE_FEATURE(kTestRotatingPromo,
             "TEST_RotatingPromo",
             base::FEATURE_DISABLED_BY_DEFAULT);
BASE_FEATURE(kTestSimplePromo,
             "TEST_SimplePromo",
             base::FEATURE_DISABLED_BY_DEFAULT);
DEFINE_LOCAL_ELEMENT_IDENTIFIER_VALUE(kTestAnchorElement);
DEFINE_LOCAL_ELEMENT_IDENTIFIER_VALUE(kTestAnchorElement2);
constexpr ui::ElementContext kTestContext =
    ui::ElementContext::CreateFakeContextForTesting(1);
}  // namespace

TEST(FeaturePromoSpecificationTest, HelpBubbleArrow) {
  ui::test::TestElement el(kTestAnchorElement, kTestContext);
  FeaturePromoSpecification spec =
      std::move(FeaturePromoSpecification::CreateForToastPromo(
                    kTestSimplePromo, kTestAnchorElement, IDS_CLOSE_PROMO, 0,
                    FeaturePromoSpecification::AcceleratorInfo())
                    .SetBubbleArrow(HelpBubbleArrow::kBottomLeft));
  EXPECT_EQ(HelpBubbleArrow::kBottomLeft, spec.GetBubbleArrow(&el));
}

TEST(FeaturePromoSpecificationTest, HelpBubbleArrowCallback) {
  ui::test::TestElement el(kTestAnchorElement, kTestContext);
  FeaturePromoSpecification spec =
      std::move(FeaturePromoSpecification::CreateForToastPromo(
                    kTestSimplePromo, kTestAnchorElement, IDS_CLOSE_PROMO, 0,
                    FeaturePromoSpecification::AcceleratorInfo())
                    .SetBubbleArrowCallback(base::BindLambdaForTesting(
                        [&](const ui::TrackedElement* anchor_element) {
                          EXPECT_EQ(&el, anchor_element);
                          return HelpBubbleArrow::kBottomLeft;
                        })));
  EXPECT_EQ(HelpBubbleArrow::kBottomLeft, spec.GetBubbleArrow(&el));
}

TEST(FeaturePromoSpecificationTest, RotatingPromoOverrideFocusOnShow) {
  FeaturePromoSpecification::RotatingPromos promos(
      FeaturePromoSpecification::CreateForToastPromo(
          kTestRotatingPromo, kTestAnchorElement, IDS_CLOSE_PROMO, 0,
          FeaturePromoSpecification::AcceleratorInfo()),
      std::nullopt,
      FeaturePromoSpecification::CreateForToastPromo(
          kTestRotatingPromo, kTestAnchorElement, IDS_CLOSE_PROMO, 0,
          FeaturePromoSpecification::AcceleratorInfo()));

  promos.at(2)->OverrideFocusOnShow(false);

  auto spec = FeaturePromoSpecification::CreateRotatingPromoForTesting(
      kTestRotatingPromo, std::move(promos));

  spec.OverrideFocusOnShow(true);

  EXPECT_EQ(true, spec.rotating_promos().at(0)->focus_on_show_override());
  EXPECT_EQ(std::nullopt, spec.rotating_promos().at(1));
  EXPECT_EQ(false, spec.rotating_promos().at(2)->focus_on_show_override());
}

TEST(FeaturePromoSpecificationTest, GetAnchorElementFromRotatingPromo) {
  ui::test::TestElement el1(kTestAnchorElement, kTestContext);
  ui::test::TestElement el2(kTestAnchorElement2, kTestContext);
  el1.Show();
  el2.Show();

  FeaturePromoSpecification::RotatingPromos promos(
      FeaturePromoSpecification::CreateForToastPromo(
          kTestRotatingPromo, kTestAnchorElement, IDS_CLOSE_PROMO, 0,
          FeaturePromoSpecification::AcceleratorInfo()),
      std::nullopt,
      FeaturePromoSpecification::CreateForToastPromo(
          kTestRotatingPromo, kTestAnchorElement2, IDS_CLOSE_PROMO, 0,
          FeaturePromoSpecification::AcceleratorInfo()));

  auto spec = FeaturePromoSpecification::CreateRotatingPromoForTesting(
      kTestRotatingPromo, std::move(promos));

  EXPECT_EQ(&el1,
            spec.GetAnchorElement(kTestContext, AnchorElementFilter(), 0));
  EXPECT_EQ(&el2,
            spec.GetAnchorElement(kTestContext, AnchorElementFilter(), 2));
}

TEST(FeaturePromoSpecificationTest, CustomActionCaptionLazyLoad) {
  // kLazilySetCustomActionCaption is disabled (eager evaluation).
  {
    base::test::ScopedFeatureList feature_list;
    feature_list.InitAndDisableFeature(features::kLazilySetCustomActionCaption);

    FeaturePromoSpecification spec =
        FeaturePromoSpecification::CreateForCustomAction(
            kTestSimplePromo, kTestAnchorElement, IDS_CLOSE_PROMO,
            IDS_CLOSE_PROMO, base::DoNothing());

    EXPECT_EQ(l10n_util::GetStringUTF16(IDS_CLOSE_PROMO),
              spec.custom_action_caption());
  }

  // kLazilySetCustomActionCaption is enabled (lazy evaluation).
  {
    base::test::ScopedFeatureList feature_list;
    feature_list.InitAndEnableFeature(features::kLazilySetCustomActionCaption);

    FeaturePromoSpecification spec =
        FeaturePromoSpecification::CreateForCustomAction(
            kTestSimplePromo, kTestAnchorElement, IDS_CLOSE_PROMO,
            IDS_CLOSE_PROMO, base::DoNothing());

    EXPECT_EQ(l10n_util::GetStringUTF16(IDS_CLOSE_PROMO),
              spec.custom_action_caption());
  }
}

}  // namespace user_education