chromium/components/segmentation_platform/internal/data_collection/training_data_cache_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/data_collection/training_data_cache.h"

#include "base/functional/callback_helpers.h"
#include "base/test/task_environment.h"
#include "components/segmentation_platform/internal/database/segment_info_database.h"
#include "components/segmentation_platform/internal/database/test_segment_info_database.h"
#include "components/segmentation_platform/internal/proto/model_prediction.pb.h"
#include "components/segmentation_platform/public/proto/segmentation_platform.pb.h"
#include "components/segmentation_platform/public/trigger.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace segmentation_platform {

namespace {

// Test Ids.
const proto::SegmentId kSegmentId = …;

const TrainingRequestId kRequestId = …;

}  // namespace

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

TEST_F(TrainingDataCacheTest, GetTrainingDataFromEmptyCache) { … }

TEST_F(TrainingDataCacheTest, GetTrainingDataFromCache) { … }

TEST_F(TrainingDataCacheTest, GetTrainingDataFromDB) { … }

}  // namespace segmentation_platform