chromium/components/history_embeddings/history_embeddings_service_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/history_embeddings/history_embeddings_service.h"

#include <memory>
#include <optional>

#include "base/files/file_path.h"
#include "base/files/scoped_temp_dir.h"
#include "base/run_loop.h"
#include "base/task/cancelable_task_tracker.h"
#include "base/test/bind.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "base/time/time.h"
#include "components/history/core/browser/history_service.h"
#include "components/history/core/browser/history_types.h"
#include "components/history/core/test/history_service_test_util.h"
#include "components/history_embeddings/answerer.h"
#include "components/history_embeddings/history_embeddings_features.h"
#include "components/history_embeddings/vector_database.h"
#include "components/optimization_guide/core/test_model_info_builder.h"
#include "components/optimization_guide/core/test_optimization_guide_decider.h"
#include "components/optimization_guide/core/test_optimization_guide_model_provider.h"
#include "components/os_crypt/async/browser/test_utils.h"
#include "components/page_content_annotations/core/test_page_content_annotations_service.h"
#include "components/page_content_annotations/core/test_page_content_annotator.h"
#include "testing/gtest/include/gtest/gtest.h"

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

namespace history_embeddings {

OptimizationGuideModelExecutor;

class HistoryEmbeddingsServicePublic : public HistoryEmbeddingsService { … };

class HistoryEmbeddingsServiceTest : public testing::Test { … };

TEST_F(HistoryEmbeddingsServiceTest, ConstructsAndInvalidatesWeakPtr) { … }

TEST_F(HistoryEmbeddingsServiceTest, OnHistoryDeletions) { … }

TEST_F(HistoryEmbeddingsServiceTest, SearchSetsValidSessionId) { … }

TEST_F(HistoryEmbeddingsServiceTest, SearchCallsCallbackWithAnswer) { … }

TEST_F(HistoryEmbeddingsServiceTest, SearchReportsHistograms) { … }

TEST_F(HistoryEmbeddingsServiceTest, SearchUsesCorrectThresholds) { … }

TEST_F(HistoryEmbeddingsServiceTest, SearchFiltersLowScoringResults) { … }

TEST_F(HistoryEmbeddingsServiceTest, CountWords) { … }

TEST_F(HistoryEmbeddingsServiceTest, StaticHashVerificationTest) { … }

TEST_F(HistoryEmbeddingsServiceTest, FilterHashes) { … }

TEST_F(HistoryEmbeddingsServiceTest, AnswerMocked) { … }

}  // namespace history_embeddings