// 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/anchor_element_provider.h"

#include "base/test/bind.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/interaction/element_identifier.h"
#include "ui/base/interaction/element_test_util.h"

namespace user_education {

namespace {
DEFINE_LOCAL_ELEMENT_IDENTIFIER_VALUE(kTestElementId1);
DEFINE_LOCAL_ELEMENT_IDENTIFIER_VALUE(kTestElementId2);
constexpr ui::ElementContext kTestElementContext1 =
    ui::ElementContext::CreateFakeContextForTesting(1);
constexpr ui::ElementContext kTestElementContext2 =
    ui::ElementContext::CreateFakeContextForTesting(2);
}  // namespace

TEST(AnchorElementProviderTest,
     AnchorElementProviderCommonRetrievesSingleElement) {
  ui::test::TestElement el(kTestElementId1, kTestElementContext1);

  AnchorElementProviderCommon provider(kTestElementId1);

  // No visible element, so returns null.
  EXPECT_EQ(nullptr,
            provider.GetAnchorElement(kTestElementContext1,
                                      AnchorElementFilter(), std::nullopt));
  el.Show();

  // Finds visible element.
  EXPECT_EQ(&el,
            provider.GetAnchorElement(kTestElementContext1,
                                      AnchorElementFilter(), std::nullopt));

  // Does not find element in the wrong context.
  EXPECT_EQ(nullptr,
            provider.GetAnchorElement(kTestElementContext2,
                                      AnchorElementFilter(), std::nullopt));

  // Wrong ID does not find element.
  AnchorElementProviderCommon provider2(kTestElementId2);
  EXPECT_EQ(nullptr,
            provider2.GetAnchorElement(kTestElementContext1,
                                       AnchorElementFilter(), std::nullopt));

  // Finds an element matched by the filter.
  EXPECT_EQ(&el,
            provider.GetAnchorElement(
                kTestElementContext1,
                base::BindLambdaForTesting(
                    [&el](const ui::ElementTracker::ElementList& elements) {
                      return elements.front() == &el ? elements.front()
                                                     : nullptr;
                    }),
                std::nullopt));

  // Does not find an element rejected by the filter.
  EXPECT_EQ(nullptr,
            provider.GetAnchorElement(
                kTestElementContext1,
                base::BindLambdaForTesting(
                    [&el](const ui::ElementTracker::ElementList& elements) {
                      return elements.front() != &el ? elements.front()
                                                     : nullptr;
                    }),
                std::nullopt));
}

}  // namespace user_education