chromium/services/on_device_model/on_device_model_service_unittest.cc

// Copyright 2023 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/on_device_model/on_device_model_service.h"

#include "base/files/scoped_temp_file.h"
#include "base/test/bind.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "services/on_device_model/fake/on_device_model_fake.h"
#include "services/on_device_model/public/cpp/model_assets.h"
#include "services/on_device_model/public/cpp/test_support/test_response_holder.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace on_device_model {
namespace {

ElementsAre;

class ContextClientWaiter : public mojom::ContextClient { … };

class FakeFile { … };

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

TEST_F(OnDeviceModelServiceTest, Responds) { … }

TEST_F(OnDeviceModelServiceTest, AddContext) { … }

TEST_F(OnDeviceModelServiceTest, CloneContext) { … }

TEST_F(OnDeviceModelServiceTest, MultipleSessionsCloneContextAndContinue) { … }

TEST_F(OnDeviceModelServiceTest, MultipleSessionsAddContext) { … }

TEST_F(OnDeviceModelServiceTest, IgnoresContext) { … }

TEST_F(OnDeviceModelServiceTest, MultipleSessionsIgnoreContext) { … }

TEST_F(OnDeviceModelServiceTest, AddContextWithTokenLimits) { … }

TEST_F(OnDeviceModelServiceTest, CancelsPreviousSession) { … }

TEST_F(OnDeviceModelServiceTest, MultipleSessionsWaitPreviousSession) { … }

TEST_F(OnDeviceModelServiceTest, LoadsAdaptation) { … }

TEST_F(OnDeviceModelServiceTest, LoadsAdaptationWithPath) { … }

TEST_F(OnDeviceModelServiceTest, LoadingAdaptationCancelsSession) { … }

TEST_F(OnDeviceModelServiceTest,
       MultipleSessionsLoadingAdaptationNotCancelsSession) { … }

TEST_F(OnDeviceModelServiceTest, DeletesModel) { … }

TEST_F(OnDeviceModelServiceTest, Score) { … }

}  // namespace
}  // namespace on_device_model