chromium/components/omnibox/browser/history_embeddings_provider_unittest.cc

// 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/omnibox/browser/history_embeddings_provider.h"

#include <memory>
#include <optional>
#include <string>
#include <utility>
#include <vector>

#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/memory/scoped_refptr.h"
#include "base/test/bind.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "base/time/time.h"
#include "components/history/core/browser/history_service.h"
#include "components/history/core/browser/url_row.h"
#include "components/history/core/test/history_service_test_util.h"
#include "components/history_embeddings/history_embeddings_features.h"
#include "components/history_embeddings/history_embeddings_service.h"
#include "components/history_embeddings/mock_history_embeddings_service.h"
#include "components/omnibox/browser/autocomplete_input.h"
#include "components/omnibox/browser/autocomplete_match.h"
#include "components/omnibox/browser/autocomplete_provider_listener.h"
#include "components/omnibox/browser/fake_autocomplete_provider_client.h"
#include "components/omnibox/browser/test_scheme_classifier.h"
#include "components/search_engines/template_url.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/metrics_proto/omnibox_event.pb.h"
#include "ui/base/page_transition_types.h"
#include "url/gurl.h"

#if BUILDFLAG(IS_CHROMEOS)
#include "chromeos/constants/chromeos_features.h"
#endif  // BUILDFLAG(IS_CHROMEOS)

namespace {
AutocompleteInput CreateAutocompleteInput(const std::u16string input) { … }

history_embeddings::ScoredUrlRow CreateScoredUrlRow(
    float score,
    const std::string& url,
    const std::u16string& title) { … }

history_embeddings::SearchResult CreateSearchResult(
    const std::u16string& title) { … }
}  // namespace

class FakeHistoryEmbeddingsProvider : public HistoryEmbeddingsProvider { … };

class HistoryEmbeddingsProviderTest : public testing::Test,
                                      public AutocompleteProviderListener { … };

TEST_F(HistoryEmbeddingsProviderTest, Start) { … }

TEST_F(HistoryEmbeddingsProviderTest, Start_MultipleSequentialSearches) { … }

TEST_F(HistoryEmbeddingsProviderTest, Start_MultipleParallelSearches) { … }

TEST_F(HistoryEmbeddingsProviderTest,
       Start_MultipleParallelSearchesWithSameQuery) { … }

TEST_F(HistoryEmbeddingsProviderTest,
       Start_MultipleParallelSearchesWithIneligibleQuery) { … }

TEST_F(HistoryEmbeddingsProviderTest, Start_Stop_SearchCompletesAfterStop) { … }

TEST_F(HistoryEmbeddingsProviderTest, Stop) { … }

TEST_F(HistoryEmbeddingsProviderTest, DeleteMatch) { … }

TEST_F(HistoryEmbeddingsProviderTest,
       OnReceivedSearchResult_CreatesAutocompleteMatches) { … }

TEST_F(HistoryEmbeddingsProviderTest,
       OnReceivedSearchResult_CreatesScopedAutocompleteMatches) { … }