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

#include "base/memory/raw_ptr.h"
#include "components/optimization_guide/core/optimization_guide_model_executor.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace optimization_guide {

MockOptimizationGuideModelExecutor::MockOptimizationGuideModelExecutor() =
    default;
MockOptimizationGuideModelExecutor::~MockOptimizationGuideModelExecutor() =
    default;

MockSession::MockSession() = default;
MockSession::~MockSession() = default;

MockSessionWrapper::MockSessionWrapper(MockSession* session)
    : … { … }
MockSessionWrapper::~MockSessionWrapper() = default;

const optimization_guide::TokenLimits& MockSessionWrapper::GetTokenLimits()
    const { … }
void MockSessionWrapper::AddContext(
    const google::protobuf::MessageLite& request_metadata) { … }
void MockSessionWrapper::Score(const std::string& text,
                               OptimizationGuideModelScoreCallback callback) { … }
void MockSessionWrapper::ExecuteModel(
    const google::protobuf::MessageLite& request_metadata,
    OptimizationGuideModelExecutionResultStreamingCallback callback) { … }
void MockSessionWrapper::GetSizeInTokens(
    const std::string& text,
    OptimizationGuideModelSizeInTokenCallback callback) { … }
const SamplingParams MockSessionWrapper::GetSamplingParams() const { … }

}  // namespace optimization_guide