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

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

namespace contextual_tasks {

TEST(GetMatchingWordsCountTest, QueryOrCandidateEmpty) {
  EXPECT_EQ(GetMatchingWordsCount("", "Page Title"), 0);
  EXPECT_EQ(GetMatchingWordsCount("Query", ""), 0);
  EXPECT_EQ(GetMatchingWordsCount("", ""), 0);
}

TEST(GetMatchingWordsCountTest, QueryOrCandidateHavePunctuations) {
  EXPECT_EQ(GetMatchingWordsCount("?! ,", "  ...  "), 0);
  EXPECT_EQ(GetMatchingWordsCount("Query", "  ...  "), 0);
  EXPECT_EQ(GetMatchingWordsCount("?! ,", "Title"), 0);
  EXPECT_EQ(GetMatchingWordsCount("test, query!", "This is a test query..."),
            2);
  EXPECT_EQ(GetMatchingWordsCount("test-query", "This is a test query"), 2);
}

TEST(GetMatchingWordsCountTest, QueryAndCandidateHaveNoMatch) {
  EXPECT_EQ(GetMatchingWordsCount("no match", "This is a test query"), 0);
}

TEST(GetMatchingWordsCountTest, QueryHasOnlyStopWords) {
  EXPECT_EQ(GetMatchingWordsCount("this is a", "This is a test query"), 0);
}

TEST(GetMatchingWordsCountTest, QueryAndCandidatesHaveMatches) {
  EXPECT_EQ(GetMatchingWordsCount("what's the weather", "Weather forecast"), 1);
  EXPECT_EQ(GetMatchingWordsCount("test query", "This is a test query"), 2);
  EXPECT_EQ(GetMatchingWordsCount("this is a test query", "Test query"), 2);
  EXPECT_EQ(GetMatchingWordsCount("test query example", "This is a test query"),
            2);
}

TEST(GetMatchingWordsCountTest, QueryHasDuplicateMatchingWords) {
  EXPECT_EQ(GetMatchingWordsCount("test test query", "This is a test query"),
            2);
}

TEST(GetMatchingWordsCountTest, NormalizedTextConsidered) {
  EXPECT_EQ(GetMatchingWordsCount("TEST QuErY", "tHis IS a TeSt qUeRy"), 2);
  EXPECT_EQ(
      GetMatchingWordsCount("  test   query  ", "  This is a test query  "), 2);
}

}  // namespace contextual_tasks