chromium/components/segmentation_platform/internal/database/cached_result_provider_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/database/cached_result_provider.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/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/public/config.h"
#include "components/segmentation_platform/public/proto/prediction_result.pb.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

_;
Invoke;
Return;
SaveArg;

namespace segmentation_platform {

namespace proto {
class SegmentInfo;
}  // namespace proto

namespace {

const char kClientKey[] = …;

// Labels for MultiClassClassifier.
constexpr std::array<const char*, 5> kMultiClassLabels{ … };

std::unique_ptr<Config> CreateTestConfig() { … }

proto::OutputConfig GetTestOutputConfigForMultiClassClassifier() { … }

}  // namespace

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

TEST_F(CachedResultProviderTest, CachedResultProviderWithEmptyPrefs) { … }

TEST_F(CachedResultProviderTest,
       GetPredictionResultForClient_WithNonEmptyResult) { … }

TEST_F(CachedResultProviderTest, GetPredictionResultForClient_WithNoResult) { … }

}  // namespace segmentation_platform