chromium/components/segmentation_platform/internal/selection/request_handler_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/selection/request_handler.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/signal_database.h"
#include "components/segmentation_platform/internal/database/signal_storage_config.h"
#include "components/segmentation_platform/internal/database/storage_service.h"
#include "components/segmentation_platform/internal/metadata/metadata_writer.h"
#include "components/segmentation_platform/internal/mock_ukm_data_manager.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/model_provider.h"
#include "components/segmentation_platform/public/prediction_options.h"
#include "components/segmentation_platform/public/proto/prediction_result.pb.h"
#include "components/segmentation_platform/public/result.h"
#include "components/segmentation_platform/public/trigger.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

_;
ElementsAre;
FloatNear;
Invoke;
Return;

namespace segmentation_platform {
namespace {

// Test Ids.
const proto::SegmentId kSegmentId = …;
const std::string& kTestClientKey = …;

class MockResultProvider : public SegmentResultProvider { … };

class MockTrainingDataCollector : public TrainingDataCollector { … };

proto::PredictionResult CreatePredictionResultWithBinaryClassifier() { … }

proto::PredictionResult CreatePredictionResultWithGenericPredictor() { … }

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

TEST_F(RequestHandlerTest, GetPredictionResult) { … }

TEST_F(RequestHandlerTest, ExecuteOndemandAsFallbackCase) { … }

TEST_F(RequestHandlerTest, GetGenericPredictionResult) { … }

}  // namespace
}  // namespace segmentation_platform