chromium/components/segmentation_platform/internal/selection/result_refresh_manager_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 "components/segmentation_platform/internal/selection/result_refresh_manager.h"

#include "base/memory/raw_ptr.h"
#include "base/metrics/user_metrics.h"
#include "base/run_loop.h"
#include "base/test/gmock_callback_support.h"
#include "base/test/simple_test_clock.h"
#include "base/test/task_environment.h"
#include "components/prefs/pref_registry_simple.h"
#include "components/prefs/testing_pref_service.h"
#include "components/segmentation_platform/internal/constants.h"
#include "components/segmentation_platform/internal/data_collection/training_data_collector.h"
#include "components/segmentation_platform/internal/database/client_result_prefs.h"
#include "components/segmentation_platform/internal/metadata/metadata_utils.h"
#include "components/segmentation_platform/internal/metadata/metadata_writer.h"
#include "components/segmentation_platform/internal/post_processor/post_processing_test_utils.h"
#include "components/segmentation_platform/internal/selection/segment_result_provider.h"
#include "components/segmentation_platform/public/config.h"
#include "components/segmentation_platform/public/prediction_options.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

_;
Invoke;

namespace segmentation_platform {
namespace {

// Test clients.
const char kTestClient1[] = …;
const char kTestClient2[] = …;

// Test Ids.
const proto::SegmentId kSegmentId1 = …;
const proto::SegmentId kSegmentId2 = …;

class MockResultProvider : public SegmentResultProvider { … };

class MockTrainingDataCollector : public TrainingDataCollector { … };

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

TEST_F(ResultRefreshManagerTest, TestRefreshModelResultsSuccess) { … }

TEST_F(ResultRefreshManagerTest, TestRefreshModelResultWithNoResult) { … }

TEST_F(ResultRefreshManagerTest, TestOnModelUpdated) { … }

TEST_F(ResultRefreshManagerTest, TestOnModelUpdatedWithDelay) { … }

}  // namespace
}  // namespace segmentation_platform