chromium/components/segmentation_platform/internal/ukm_data_manager_impl_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/ukm_data_manager_impl.h"

#include "base/files/scoped_temp_dir.h"
#include "base/metrics/metrics_hashes.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "components/history/core/browser/history_service.h"
#include "components/history/core/test/history_service_test_util.h"
#include "components/prefs/testing_pref_service.h"
#include "components/segmentation_platform/internal/database/mock_ukm_database.h"
#include "components/segmentation_platform/internal/database/ukm_types.h"
#include "components/segmentation_platform/internal/execution/model_manager_impl.h"
#include "components/segmentation_platform/internal/proto/model_prediction.pb.h"
#include "components/segmentation_platform/internal/segmentation_platform_service_impl.h"
#include "components/segmentation_platform/internal/segmentation_platform_service_test_base.h"
#include "components/segmentation_platform/internal/signals/ukm_observer.h"
#include "components/segmentation_platform/public/features.h"
#include "components/segmentation_platform/public/local_state_helper.h"
#include "components/ukm/test_ukm_recorder.h"
#include "services/metrics/public/cpp/ukm_builders.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace segmentation_platform {

namespace {

_;
PageLoad;
PaintPreviewCapture;

constexpr ukm::SourceId kSourceId = …;
constexpr ukm::SourceId kSourceId2 = …;

ukm::mojom::UkmEntryPtr GetSamplePageLoadEntry(
    ukm::SourceId source_id = kSourceId) { … }

ukm::mojom::UkmEntryPtr GetSamplePaintPreviewEntry(
    ukm::SourceId source_id = kSourceId) { … }

proto::SegmentationModelMetadata PageLoadModelMetadata() { … }

proto::SegmentationModelMetadata PaintPreviewModelMetadata() { … }

}  // namespace

class TestServicesForPlatform : public SegmentationPlatformServiceTestBase { … };

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

MATCHER_P(HasEventHash, event_hash, "") { … }

TEST_F(UkmDataManagerImplTest, HistoryNotification) { … }

TEST_F(UkmDataManagerImplTest, UkmSourceObservation) { … }

TEST_F(UkmDataManagerImplTest, UkmEntryObservation) { … }

TEST_F(UkmDataManagerImplTest, UkmServiceCreatedBeforePlatform) { … }

TEST_F(UkmDataManagerImplTest, UrlValidationWithHistory) { … }

TEST_F(UkmDataManagerImplTest, MultiplePlatforms) { … }

}  // namespace segmentation_platform