| // Copyright 2015 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/sync/model/mutable_data_batch.h" |
| |
| #include "base/memory/ptr_util.h" |
| #include "components/sync/engine/entity_data.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace syncer { |
| |
| TEST(MutableDataBatchTest, PutAndNextWithReuse) { |
| EntityData* entity1 = new EntityData(); |
| EntityData* entity2 = new EntityData(); |
| |
| MutableDataBatch batch; |
| EXPECT_FALSE(batch.HasNext()); |
| |
| batch.Put("one", base::WrapUnique(entity1)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const auto& [key1, data1] = batch.Next(); |
| EXPECT_FALSE(batch.HasNext()); |
| EXPECT_EQ("one", key1); |
| EXPECT_EQ(entity1, data1.get()); |
| |
| batch.Put("two", base::WrapUnique(entity2)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const auto& [key2, data2] = batch.Next(); |
| EXPECT_FALSE(batch.HasNext()); |
| EXPECT_EQ("two", key2); |
| EXPECT_EQ(entity2, data2.get()); |
| } |
| |
| TEST(MutableDataBatchTest, PutAndNextInterleaved) { |
| EntityData* entity1 = new EntityData(); |
| EntityData* entity2 = new EntityData(); |
| EntityData* entity3 = new EntityData(); |
| |
| MutableDataBatch batch; |
| EXPECT_FALSE(batch.HasNext()); |
| |
| batch.Put("one", base::WrapUnique(entity1)); |
| EXPECT_TRUE(batch.HasNext()); |
| batch.Put("two", base::WrapUnique(entity2)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const auto& [key1, data1] = batch.Next(); |
| EXPECT_TRUE(batch.HasNext()); |
| EXPECT_EQ("one", key1); |
| EXPECT_EQ(entity1, data1.get()); |
| |
| batch.Put("three", base::WrapUnique(entity3)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const auto& [key2, data2] = batch.Next(); |
| EXPECT_TRUE(batch.HasNext()); |
| EXPECT_EQ("two", key2); |
| EXPECT_EQ(entity2, data2.get()); |
| |
| const auto& [key3, data3] = batch.Next(); |
| EXPECT_FALSE(batch.HasNext()); |
| EXPECT_EQ("three", key3); |
| EXPECT_EQ(entity3, data3.get()); |
| } |
| |
| TEST(MutableDataBatchTest, PutAndNextSharedKey) { |
| EntityData* entity1 = new EntityData(); |
| EntityData* entity2 = new EntityData(); |
| |
| MutableDataBatch batch; |
| EXPECT_FALSE(batch.HasNext()); |
| |
| batch.Put("same", base::WrapUnique(entity1)); |
| EXPECT_TRUE(batch.HasNext()); |
| batch.Put("same", base::WrapUnique(entity2)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const auto& [key1, data1] = batch.Next(); |
| EXPECT_TRUE(batch.HasNext()); |
| EXPECT_EQ("same", key1); |
| EXPECT_EQ(entity1, data1.get()); |
| |
| const auto& [key2, data2] = batch.Next(); |
| EXPECT_FALSE(batch.HasNext()); |
| EXPECT_EQ("same", key2); |
| EXPECT_EQ(entity2, data2.get()); |
| } |
| |
| } // namespace syncer |