| // 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_date_search.h" |
| |
| #include <vector> |
| |
| #include "base/check.h" |
| #include "base/strings/strcat.h" |
| #include "base/time/time.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace ash { |
| namespace { |
| |
| using ::testing::AllOf; |
| using ::testing::Combine; |
| using ::testing::Each; |
| using ::testing::Field; |
| using ::testing::IsEmpty; |
| using ::testing::Not; |
| using ::testing::Optional; |
| using ::testing::Pointwise; |
| using ::testing::Property; |
| using ::testing::Values; |
| using ::testing::VariantWith; |
| |
| base::Time TimeFromDateString(const std::string& time_string) { |
| base::Time date; |
| bool result = base::Time::FromString(time_string.c_str(), &date); |
| CHECK(result); |
| return date; |
| } |
| |
| MATCHER(ResultMatchesDate, "") { |
| const auto& [actual_result, expected_result] = arg; |
| return ExplainMatchResult( |
| AllOf(VariantWith<QuickInsertTextResult>( |
| Field("text", &QuickInsertTextResult::primary_text, |
| expected_result.primary_text)), |
| VariantWith<QuickInsertTextResult>( |
| Field("text", &QuickInsertTextResult::secondary_text, |
| expected_result.secondary_text)), |
| VariantWith<QuickInsertTextResult>( |
| Field("source", &QuickInsertTextResult::source, |
| expected_result.source))), |
| actual_result, result_listener); |
| } |
| |
| struct TestCase { |
| std::string_view date; |
| std::u16string_view query; |
| std::vector<QuickInsertTextResult> expected_results; |
| }; |
| |
| QuickInsertTextResult MakeResult(std::u16string primary_text, |
| std::u16string secondary_text = u"") { |
| return QuickInsertTextResult(primary_text, secondary_text, ui::ImageModel(), |
| QuickInsertTextResult::Source::kDate); |
| } |
| |
| class QuickInsertDateSearchTest |
| : public ::testing::TestWithParam<std::tuple<std::string_view, TestCase>> { |
| }; |
| |
| INSTANTIATE_TEST_SUITE_P( |
| , |
| QuickInsertDateSearchTest, |
| Combine(Values("00:00", "12:00", "23:59"), |
| Values( |
| // No result |
| TestCase{ |
| .date = "23 Jan 2000", |
| .query = u"abc", |
| .expected_results = {}, |
| }, |
| // Today |
| TestCase{ |
| .date = "23 Jan 2000", |
| .query = u"today", |
| .expected_results = {MakeResult(u"Jan 23")}, |
| }, |
| // Yesterday |
| TestCase{ |
| .date = "23 Jan 2000", |
| .query = u"yesterday", |
| .expected_results = {MakeResult(u"Jan 22")}, |
| }, |
| // Tomorrow |
| TestCase{ |
| .date = "23 Jan 2000", |
| .query = u"tomorrow", |
| .expected_results = {MakeResult(u"Jan 24")}, |
| }, |
| // X days from now |
| TestCase{ |
| .date = "23 Jan 2000", |
| .query = u"10 days from now", |
| .expected_results = {MakeResult(u"Feb 2")}, |
| }, |
| // X days ago |
| TestCase{ |
| .date = "23 Jan 2000", |
| .query = u"five days ago", |
| .expected_results = {MakeResult(u"Jan 18")}, |
| }, |
| // X weeks from now |
| TestCase{ |
| .date = "23 Jan 2000", |
| .query = u"three weeks from now", |
| .expected_results = {MakeResult(u"Feb 13")}, |
| }, |
| // X weeks ago |
| TestCase{ |
| .date = "23 Jan 2000", |
| .query = u"2 weeks ago", |
| .expected_results = {MakeResult(u"Jan 9")}, |
| }, |
| // search for Friday on Tuesday |
| TestCase{ |
| .date = "19 Mar 2024", |
| .query = u"Friday", |
| .expected_results = {MakeResult(u"Mar 22")}, |
| }, |
| // search for this Friday on Tuesday |
| TestCase{ |
| .date = "19 Mar 2024", |
| .query = u"this Friday", |
| .expected_results = {MakeResult(u"Mar 22")}, |
| }, |
| // search for next Friday on Tuesday |
| TestCase{ |
| .date = "19 Mar 2024", |
| .query = u"next Friday", |
| .expected_results = |
| {MakeResult(u"Mar 29", u"Friday next week"), |
| MakeResult(u"Mar 22", u"This coming Friday")}, |
| }, |
| // search for last Friday on Tuesday |
| TestCase{ |
| .date = "19 Mar 2024", |
| .query = u"last Friday", |
| .expected_results = {MakeResult(u"Mar 15")}, |
| }, |
| // search for Tuesday on Friday |
| TestCase{ |
| .date = "22 Mar 2024", |
| .query = u"Tuesday", |
| .expected_results = |
| {MakeResult(u"Mar 26", u"This coming Tuesday"), |
| MakeResult(u"Mar 19", u"This past Tuesday")}, |
| }, |
| // search for this Tuesday on Friday |
| TestCase{ |
| .date = "22 Mar 2024", |
| .query = u"this Tuesday", |
| .expected_results = |
| {MakeResult(u"Mar 26", u"This coming Tuesday"), |
| MakeResult(u"Mar 19", u"This past Tuesday")}, |
| }, |
| // search for next Tuesday on Friday |
| TestCase{ |
| .date = "22 Mar 2024", |
| .query = u"next Tuesday", |
| .expected_results = {MakeResult(u"Mar 26")}, |
| }, |
| // search for last Tuesday on Friday |
| TestCase{ |
| .date = "22 Mar 2024", |
| .query = u"last Tuesday", |
| .expected_results = |
| {MakeResult(u"Mar 12", u"Tuesday last week"), |
| MakeResult(u"Mar 19", u"This past Tuesday")}, |
| }, |
| // search for Monday on Monday |
| TestCase{ |
| .date = "18 Mar 2024", |
| .query = u"Monday", |
| .expected_results = {MakeResult(u"Mar 18")}, |
| }, |
| // search for this Monday on Monday |
| TestCase{ |
| .date = "18 Mar 2024", |
| .query = u"this Monday", |
| .expected_results = {MakeResult(u"Mar 18")}, |
| }, |
| // search for next Monday on Monday |
| TestCase{ |
| .date = "18 Mar 2024", |
| .query = u"next Monday", |
| .expected_results = {MakeResult(u"Mar 25")}, |
| }, |
| // search for last Monday on Monday |
| TestCase{ |
| .date = "18 Mar 2024", |
| .query = u"last Monday", |
| .expected_results = {MakeResult(u"Mar 11")}, |
| }))); |
| |
| TEST_P(QuickInsertDateSearchTest, ReturnsExpectedDates) { |
| std::string_view time = std::get<0>(GetParam()); |
| const auto& [date, query, expected_results] = std::get<1>(GetParam()); |
| EXPECT_THAT(QuickInsertDateSearch( |
| TimeFromDateString(base::StrCat({date, " ", time})), query), |
| Pointwise(ResultMatchesDate(), expected_results)); |
| } |
| |
| TEST(QuickInsertSuggestedDateResults, ReturnsSuggestedResults) { |
| std::vector<QuickInsertSearchResult> results = |
| QuickInsertSuggestedDateResults(); |
| 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 |
