123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354 |
- // Copyright 2021 The Chromium Authors. All rights reserved.
- // 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/dummy_segmentation_platform_service.h"
- #include <string>
- #include "base/bind.h"
- #include "base/run_loop.h"
- #include "base/test/task_environment.h"
- #include "components/segmentation_platform/internal/constants.h"
- #include "components/segmentation_platform/public/segment_selection_result.h"
- #include "testing/gtest/include/gtest/gtest.h"
- namespace segmentation_platform {
- class DummySegmentationPlatformServiceTest : public testing::Test {
- public:
- DummySegmentationPlatformServiceTest() = default;
- ~DummySegmentationPlatformServiceTest() override = default;
- void SetUp() override {
- segmentation_platform_service_ =
- std::make_unique<DummySegmentationPlatformService>();
- }
- void OnGetSelectedSegment(base::RepeatingClosure closure,
- const SegmentSelectionResult& expected,
- const SegmentSelectionResult& actual) {
- ASSERT_EQ(expected, actual);
- std::move(closure).Run();
- }
- protected:
- base::test::TaskEnvironment task_environment_;
- std::unique_ptr<DummySegmentationPlatformService>
- segmentation_platform_service_;
- };
- TEST_F(DummySegmentationPlatformServiceTest, GetSelectedSegment) {
- SegmentSelectionResult expected;
- base::RunLoop loop;
- segmentation_platform_service_->GetSelectedSegment(
- "test_key",
- base::BindOnce(
- &DummySegmentationPlatformServiceTest::OnGetSelectedSegment,
- base::Unretained(this), loop.QuitClosure(), expected));
- loop.Run();
- ASSERT_EQ(expected,
- segmentation_platform_service_->GetCachedSegmentResult("test_key"));
- }
- } // namespace segmentation_platform
|