chromium/services/passage_embeddings/passage_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 "services/passage_embeddings/passage_embeddings_service.h"

#include "base/path_service.h"
#include "base/test/bind.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "services/passage_embeddings/passage_embedder.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace passage_embeddings {
namespace {

constexpr uint32_t kInputWindowSize = …;
constexpr size_t kEmbeddingsOutputSize = …;

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

TEST_F(PassageEmbeddingsServiceTest, LoadValidModels) { … }

TEST_F(PassageEmbeddingsServiceTest, LoadModelsWithInvalidEmbeddingsModel) { … }

TEST_F(PassageEmbeddingsServiceTest, LoadModelsWithInvalidSpModel) { … }

TEST_F(PassageEmbeddingsServiceTest, LoadModelsWithInvalidInputWindowSize) { … }

TEST_F(PassageEmbeddingsServiceTest, RespondsWithEmbeddings) { … }

TEST_F(PassageEmbeddingsServiceTest, CacheHits) { … }

TEST_F(PassageEmbeddingsServiceTest, RecordsDurationHistogramsWithPriority) { … }

}  // namespace
}  // namespace passage_embeddings