chromium/components/segmentation_platform/internal/execution/optimization_guide/optimization_guide_segmentation_model_provider_unittest.cc

// Copyright 2022 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/segmentation_platform/internal/execution/optimization_guide/optimization_guide_segmentation_model_provider.h"
#include <memory>
#include <utility>

#include "base/run_loop.h"
#include "base/task/thread_pool.h"
#include "base/test/mock_callback.h"
#include "base/test/task_environment.h"
#include "components/optimization_guide/core/model_info.h"
#include "components/optimization_guide/core/optimization_guide_model_provider.h"
#include "components/optimization_guide/core/optimization_guide_util.h"
#include "components/optimization_guide/core/optimization_target_model_observer.h"
#include "components/optimization_guide/core/test_model_info_builder.h"
#include "components/optimization_guide/core/test_optimization_guide_model_provider.h"
#include "components/optimization_guide/proto/models.pb.h"
#include "components/segmentation_platform/public/proto/model_metadata.pb.h"
#include "testing/gtest/include/gtest/gtest.h"

_;
SaveArg;

namespace segmentation_platform {

namespace {

class ModelObserverTracker
    : public optimization_guide::TestOptimizationGuideModelProvider { … };

}  // namespace

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

TEST_F(OptimizationGuideSegmentationModelProviderTest, InitAndFetchModel) { … }

TEST_F(OptimizationGuideSegmentationModelProviderTest,
       ExecuteModelWithoutFetch) { … }

TEST_F(OptimizationGuideSegmentationModelProviderTest, ExecuteModelWithFetch) { … }

TEST_F(OptimizationGuideSegmentationModelProviderTest, NotifyOnDeletedModel) { … }

}  // namespace segmentation_platform