| // 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 "sync/internal_api/public/data_batch_impl.h" |
| |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace syncer_v2 { |
| |
| TEST(DataBatchImplTest, PutAndNextWithReuse) { |
| EntityData* entity1 = new EntityData(); |
| EntityData* entity2 = new EntityData(); |
| |
| DataBatchImpl batch; |
| EXPECT_FALSE(batch.HasNext()); |
| |
| batch.Put("one", make_scoped_ptr(entity1)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const KeyAndData& pair1 = batch.Next(); |
| EXPECT_FALSE(batch.HasNext()); |
| EXPECT_EQ("one", pair1.first); |
| EXPECT_EQ(entity1, pair1.second.get()); |
| |
| batch.Put("two", make_scoped_ptr(entity2)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const KeyAndData& pair2 = batch.Next(); |
| EXPECT_FALSE(batch.HasNext()); |
| EXPECT_EQ("two", pair2.first); |
| EXPECT_EQ(entity2, pair2.second.get()); |
| } |
| |
| TEST(DataBatchImplTest, PutAndNextInterleaved) { |
| EntityData* entity1 = new EntityData(); |
| EntityData* entity2 = new EntityData(); |
| EntityData* entity3 = new EntityData(); |
| |
| DataBatchImpl batch; |
| EXPECT_FALSE(batch.HasNext()); |
| |
| batch.Put("one", make_scoped_ptr(entity1)); |
| EXPECT_TRUE(batch.HasNext()); |
| batch.Put("two", make_scoped_ptr(entity2)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const KeyAndData& pair1 = batch.Next(); |
| EXPECT_TRUE(batch.HasNext()); |
| EXPECT_EQ("one", pair1.first); |
| EXPECT_EQ(entity1, pair1.second.get()); |
| |
| batch.Put("three", make_scoped_ptr(entity3)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const KeyAndData& pair2 = batch.Next(); |
| EXPECT_TRUE(batch.HasNext()); |
| EXPECT_EQ("two", pair2.first); |
| EXPECT_EQ(entity2, pair2.second.get()); |
| |
| const KeyAndData& pair3 = batch.Next(); |
| EXPECT_FALSE(batch.HasNext()); |
| EXPECT_EQ("three", pair3.first); |
| EXPECT_EQ(entity3, pair3.second.get()); |
| } |
| |
| TEST(DataBatchImplTest, PutAndNextSharedKey) { |
| EntityData* entity1 = new EntityData(); |
| EntityData* entity2 = new EntityData(); |
| |
| DataBatchImpl batch; |
| EXPECT_FALSE(batch.HasNext()); |
| |
| batch.Put("same", make_scoped_ptr(entity1)); |
| EXPECT_TRUE(batch.HasNext()); |
| batch.Put("same", make_scoped_ptr(entity2)); |
| EXPECT_TRUE(batch.HasNext()); |
| |
| const KeyAndData& pair1 = batch.Next(); |
| EXPECT_TRUE(batch.HasNext()); |
| EXPECT_EQ("same", pair1.first); |
| EXPECT_EQ(entity1, pair1.second.get()); |
| |
| const KeyAndData& pair2 = batch.Next(); |
| EXPECT_FALSE(batch.HasNext()); |
| EXPECT_EQ("same", pair2.first); |
| EXPECT_EQ(entity2, pair2.second.get()); |
| } |
| |
| } // namespace syncer_v2 |