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

#ifndef COMPONENTS_AUTOFILL_CORE_BROWSER_SUGGESTIONS_SUGGESTION_TEST_HELPERS_H_
#define COMPONENTS_AUTOFILL_CORE_BROWSER_SUGGESTIONS_SUGGESTION_TEST_HELPERS_H_

#include "components/autofill/core/browser/suggestions/suggestion.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace autofill {

::testing::Matcher<Suggestion> EqualsSuggestion(SuggestionType id);

::testing::Matcher<Suggestion> EqualsSuggestion(
    SuggestionType id,
    const std::u16string& main_text,
    bool is_main_text_primary = true);

::testing::Matcher<Suggestion> EqualsSuggestion(
    SuggestionType id,
    const Suggestion::Payload& payload);

::testing::Matcher<Suggestion> EqualsSuggestion(SuggestionType id,
                                                const std::u16string& main_text,
                                                Suggestion::Icon icon);

::testing::Matcher<Suggestion> EqualsSuggestion(
    SuggestionType type,
    const std::u16string& main_text,
    Suggestion::Icon icon,
    const std::vector<std::vector<Suggestion::Text>>& labels);

::testing::Matcher<Suggestion> EqualsSuggestion(
    SuggestionType type,
    const std::u16string& main_text,
    bool is_main_text_primary,
    Suggestion::Icon icon,
    const std::vector<std::vector<Suggestion::Text>>& labels,
    const Suggestion::Payload& payload);

::testing::Matcher<Suggestion> EqualsSuggestion(
    SuggestionType id,
    const std::u16string& main_text,
    Suggestion::Icon icon,
    const Suggestion::Payload& payload);

::testing::Matcher<Suggestion> HasIcon(Suggestion::Icon icon);

::testing::Matcher<Suggestion> HasTrailingIcon(Suggestion::Icon icon);

::testing::Matcher<Suggestion> HasIphFeature(
    raw_ptr<const base::Feature> feature);

::testing::Matcher<Suggestion> HasNoIphFeature();

template <class... Matchers>
inline auto SuggestionVectorIdsAre(const Matchers&... matchers) {
  return ::testing::ElementsAre(
      ::testing::Field("type", &Suggestion::type, matchers)...);
}

template <class... Matchers>
inline auto SuggestionVectorMainTextsAre(const Matchers&... matchers) {
  return ::testing::ElementsAre(
      ::testing::Field("main_text", &Suggestion::main_text, matchers)...);
}

template <class... Matchers>
inline auto SuggestionAdditionalLabelsContains(const Matchers&... matchers) {
  return ::testing::Contains(::testing::Field(
      "additional_label", &Suggestion::additional_label, matchers)...);
}

template <class... Matchers>
inline auto SuggestionVectorIconsAre(const Matchers&... matchers) {
  return ::testing::ElementsAre(
      ::testing::Field("icon", &Suggestion::icon, matchers)...);
}

template <class... Matchers>
inline auto SuggestionVectorStoreIndicatorIconsAre(
    const Matchers&... matchers) {
  return ::testing::ElementsAre(
      ::testing::Field("icon", &Suggestion::trailing_icon, matchers)...);
}

}  // namespace autofill

#endif  // COMPONENTS_AUTOFILL_CORE_BROWSER_SUGGESTIONS_SUGGESTION_TEST_HELPERS_H_