// 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 "ash/quick_insert/search/quick_insert_math_search.h"

#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace ash {
namespace {

using ::testing::AllOf;
using ::testing::Field;
using ::testing::IsEmpty;
using ::testing::Not;
using ::testing::Optional;
using ::testing::Property;
using ::testing::VariantWith;

TEST(QuickInsertMathSearchTest, NoResult) {
  EXPECT_FALSE(QuickInsertMathSearch(u"abc").has_value());
}

TEST(QuickInsertMathSearchTest, OnePlusOneEqualsTwo) {
  EXPECT_THAT(
      QuickInsertMathSearch(u"1 + 1"),
      Optional(AllOf(VariantWith<QuickInsertTextResult>(Field(
                         "text", &QuickInsertTextResult::primary_text, u"2")),
                     VariantWith<QuickInsertTextResult>(
                         Field("source", &QuickInsertTextResult::source,
                               QuickInsertTextResult::Source::kMath)))));
}

TEST(QuickInsertMathSearchTest, ReturnsExamples) {
  std::vector<QuickInsertSearchResult> results = QuickInsertMathExamples();
  EXPECT_THAT(results, Not(IsEmpty()));
  EXPECT_THAT(
      results,
      Each(VariantWith<QuickInsertSearchRequestResult>(AllOf(
          Field("primary_text", &QuickInsertSearchRequestResult::primary_text,
                Not(IsEmpty())),
          Field("secondary_text",
                &QuickInsertSearchRequestResult::secondary_text,
                Not(IsEmpty()))))));
}

}  // namespace
}  // namespace ash