chromium/components/optimization_guide/core/model_execution/on_device_model_validator_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/optimization_guide/core/model_execution/on_device_model_validator.h"

#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "components/optimization_guide/core/model_execution/test/fake_on_device_model_service_controller.h"
#include "components/optimization_guide/core/model_execution/test/feature_config_builder.h"
#include "services/on_device_model/public/cpp/test_support/fake_service.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace optimization_guide {
namespace {

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

TEST_F(OnDeviceModelValidatorTest, Succeeds) { … }

TEST_F(OnDeviceModelValidatorTest, SucceedsWithEmptyConfig) { … }

TEST_F(OnDeviceModelValidatorTest, FailsWhenSessionKilled) { … }

TEST_F(OnDeviceModelValidatorTest, Fails) { … }

TEST_F(OnDeviceModelValidatorTest, SucceedsWithChunkedResponse) { … }

TEST_F(OnDeviceModelValidatorTest, SucceedsWithMultiplePrompts) { … }

TEST_F(OnDeviceModelValidatorTest, FailsWithSecondPromptNotMatching) { … }

}  // namespace
}  // namespace optimization_guide