chromium/components/history_embeddings/ml_answerer_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/ml_answerer.h"

#include "base/memory/raw_ptr.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "components/optimization_guide/core/mock_optimization_guide_model_executor.h"
#include "components/optimization_guide/proto/features/history_answer.pb.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace history_embeddings {

TestFuture;
OptimizationGuideModelExecutionError;
OptimizationGuideModelStreamingExecutionResult;
HistoryAnswerResponse;
_;

namespace {

constexpr char kAnswerResponseTypeURL[] = …;

}  // namespace

class MockModelExecutor
    : public optimization_guide::MockOptimizationGuideModelExecutor { … };

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

TEST_F(MlAnswererTest, ComputeAnswerNoSession) { … }

#if !BUILDFLAG(IS_FUCHSIA)
TEST_F(MlAnswererTest, ComputeAnswerExecutionFailure) { … }
#endif

TEST_F(MlAnswererTest, ComputeAnswerSingleUrl) { … }

TEST_F(MlAnswererTest, ComputeAnswerMultipleUrls) { … }

TEST_F(MlAnswererTest, ComputeAnswerUnanswerable) { … }

}  // namespace history_embeddings