| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "components/invalidation/unacked_invalidation_set.h" | |
| 6 | |
| 7 #include "base/json/json_string_value_serializer.h" | |
| 8 #include "components/invalidation/object_id_invalidation_map.h" | |
| 9 #include "components/invalidation/single_object_invalidation_set.h" | |
| 10 #include "components/invalidation/unacked_invalidation_set_test_util.h" | |
| 11 #include "testing/gtest/include/gtest/gtest.h" | |
| 12 | |
| 13 namespace syncer { | |
| 14 | |
| 15 class UnackedInvalidationSetTest : public testing::Test { | |
| 16 public: | |
| 17 UnackedInvalidationSetTest() | |
| 18 : kObjectId_(10, "ASDF"), | |
| 19 unacked_invalidations_(kObjectId_) {} | |
| 20 | |
| 21 SingleObjectInvalidationSet GetStoredInvalidations() { | |
| 22 ObjectIdInvalidationMap map; | |
| 23 unacked_invalidations_.ExportInvalidations( | |
| 24 base::WeakPtr<AckHandler>(), | |
| 25 scoped_refptr<base::SingleThreadTaskRunner>(), | |
| 26 &map); | |
| 27 ObjectIdSet ids = map.GetObjectIds(); | |
| 28 if (ids.find(kObjectId_) != ids.end()) { | |
| 29 return map.ForObject(kObjectId_); | |
| 30 } else { | |
| 31 return SingleObjectInvalidationSet(); | |
| 32 } | |
| 33 } | |
| 34 | |
| 35 const invalidation::ObjectId kObjectId_; | |
| 36 UnackedInvalidationSet unacked_invalidations_; | |
| 37 }; | |
| 38 | |
| 39 namespace { | |
| 40 | |
| 41 // Test storage and retrieval of zero invalidations. | |
| 42 TEST_F(UnackedInvalidationSetTest, Empty) { | |
| 43 EXPECT_EQ(0U, GetStoredInvalidations().GetSize()); | |
| 44 } | |
| 45 | |
| 46 // Test storage and retrieval of a single invalidation. | |
| 47 TEST_F(UnackedInvalidationSetTest, OneInvalidation) { | |
| 48 Invalidation inv1 = Invalidation::Init(kObjectId_, 10, "payload"); | |
| 49 unacked_invalidations_.Add(inv1); | |
| 50 | |
| 51 SingleObjectInvalidationSet set = GetStoredInvalidations(); | |
| 52 ASSERT_EQ(1U, set.GetSize()); | |
| 53 EXPECT_FALSE(set.StartsWithUnknownVersion()); | |
| 54 } | |
| 55 | |
| 56 // Test that calling Clear() returns us to the empty state. | |
| 57 TEST_F(UnackedInvalidationSetTest, Clear) { | |
| 58 Invalidation inv1 = Invalidation::Init(kObjectId_, 10, "payload"); | |
| 59 unacked_invalidations_.Add(inv1); | |
| 60 unacked_invalidations_.Clear(); | |
| 61 | |
| 62 EXPECT_EQ(0U, GetStoredInvalidations().GetSize()); | |
| 63 } | |
| 64 | |
| 65 // Test that repeated unknown version invalidations are squashed together. | |
| 66 TEST_F(UnackedInvalidationSetTest, UnknownVersions) { | |
| 67 Invalidation inv1 = Invalidation::Init(kObjectId_, 10, "payload"); | |
| 68 Invalidation inv2 = Invalidation::InitUnknownVersion(kObjectId_); | |
| 69 Invalidation inv3 = Invalidation::InitUnknownVersion(kObjectId_); | |
| 70 unacked_invalidations_.Add(inv1); | |
| 71 unacked_invalidations_.Add(inv2); | |
| 72 unacked_invalidations_.Add(inv3); | |
| 73 | |
| 74 SingleObjectInvalidationSet set = GetStoredInvalidations(); | |
| 75 ASSERT_EQ(2U, set.GetSize()); | |
| 76 EXPECT_TRUE(set.StartsWithUnknownVersion()); | |
| 77 } | |
| 78 | |
| 79 // Tests that no truncation occurs while we're under the limit. | |
| 80 TEST_F(UnackedInvalidationSetTest, NoTruncation) { | |
| 81 size_t kMax = UnackedInvalidationSet::kMaxBufferedInvalidations; | |
| 82 | |
| 83 for (size_t i = 0; i < kMax; ++i) { | |
| 84 Invalidation inv = Invalidation::Init(kObjectId_, i, "payload"); | |
| 85 unacked_invalidations_.Add(inv); | |
| 86 } | |
| 87 | |
| 88 SingleObjectInvalidationSet set = GetStoredInvalidations(); | |
| 89 ASSERT_EQ(kMax, set.GetSize()); | |
| 90 EXPECT_FALSE(set.StartsWithUnknownVersion()); | |
| 91 EXPECT_EQ(0, set.begin()->version()); | |
| 92 EXPECT_EQ(kMax-1, static_cast<size_t>(set.rbegin()->version())); | |
| 93 } | |
| 94 | |
| 95 // Test that truncation happens as we reach the limit. | |
| 96 TEST_F(UnackedInvalidationSetTest, Truncation) { | |
| 97 size_t kMax = UnackedInvalidationSet::kMaxBufferedInvalidations; | |
| 98 | |
| 99 for (size_t i = 0; i < kMax + 1; ++i) { | |
| 100 Invalidation inv = Invalidation::Init(kObjectId_, i, "payload"); | |
| 101 unacked_invalidations_.Add(inv); | |
| 102 } | |
| 103 | |
| 104 SingleObjectInvalidationSet set = GetStoredInvalidations(); | |
| 105 ASSERT_EQ(kMax, set.GetSize()); | |
| 106 EXPECT_TRUE(set.StartsWithUnknownVersion()); | |
| 107 EXPECT_TRUE(set.begin()->is_unknown_version()); | |
| 108 EXPECT_EQ(kMax, static_cast<size_t>(set.rbegin()->version())); | |
| 109 } | |
| 110 | |
| 111 // Test that we don't truncate while a handler is registered. | |
| 112 TEST_F(UnackedInvalidationSetTest, RegistrationAndTruncation) { | |
| 113 unacked_invalidations_.SetHandlerIsRegistered(); | |
| 114 | |
| 115 size_t kMax = UnackedInvalidationSet::kMaxBufferedInvalidations; | |
| 116 | |
| 117 for (size_t i = 0; i < kMax + 1; ++i) { | |
| 118 Invalidation inv = Invalidation::Init(kObjectId_, i, "payload"); | |
| 119 unacked_invalidations_.Add(inv); | |
| 120 } | |
| 121 | |
| 122 SingleObjectInvalidationSet set = GetStoredInvalidations(); | |
| 123 ASSERT_EQ(kMax+1, set.GetSize()); | |
| 124 EXPECT_FALSE(set.StartsWithUnknownVersion()); | |
| 125 EXPECT_EQ(0, set.begin()->version()); | |
| 126 EXPECT_EQ(kMax, static_cast<size_t>(set.rbegin()->version())); | |
| 127 | |
| 128 // Unregistering should re-enable truncation. | |
| 129 unacked_invalidations_.SetHandlerIsUnregistered(); | |
| 130 SingleObjectInvalidationSet set2 = GetStoredInvalidations(); | |
| 131 ASSERT_EQ(kMax, set2.GetSize()); | |
| 132 EXPECT_TRUE(set2.StartsWithUnknownVersion()); | |
| 133 EXPECT_TRUE(set2.begin()->is_unknown_version()); | |
| 134 EXPECT_EQ(kMax, static_cast<size_t>(set2.rbegin()->version())); | |
| 135 } | |
| 136 | |
| 137 // Test acknowledgement. | |
| 138 TEST_F(UnackedInvalidationSetTest, Acknowledge) { | |
| 139 // inv2 is included in this test just to make sure invalidations that | |
| 140 // are supposed to be unaffected by this operation will be unaffected. | |
| 141 | |
| 142 // We don't expect to be receiving acks or drops unless this flag is set. | |
| 143 // Not that it makes much of a difference in behavior. | |
| 144 unacked_invalidations_.SetHandlerIsRegistered(); | |
| 145 | |
| 146 Invalidation inv1 = Invalidation::Init(kObjectId_, 10, "payload"); | |
| 147 Invalidation inv2 = Invalidation::InitUnknownVersion(kObjectId_); | |
| 148 AckHandle inv1_handle = inv1.ack_handle(); | |
| 149 | |
| 150 unacked_invalidations_.Add(inv1); | |
| 151 unacked_invalidations_.Add(inv2); | |
| 152 | |
| 153 unacked_invalidations_.Acknowledge(inv1_handle); | |
| 154 | |
| 155 SingleObjectInvalidationSet set = GetStoredInvalidations(); | |
| 156 EXPECT_EQ(1U, set.GetSize()); | |
| 157 EXPECT_TRUE(set.StartsWithUnknownVersion()); | |
| 158 } | |
| 159 | |
| 160 // Test drops. | |
| 161 TEST_F(UnackedInvalidationSetTest, Drop) { | |
| 162 // inv2 is included in this test just to make sure invalidations that | |
| 163 // are supposed to be unaffected by this operation will be unaffected. | |
| 164 | |
| 165 // We don't expect to be receiving acks or drops unless this flag is set. | |
| 166 // Not that it makes much of a difference in behavior. | |
| 167 unacked_invalidations_.SetHandlerIsRegistered(); | |
| 168 | |
| 169 Invalidation inv1 = Invalidation::Init(kObjectId_, 10, "payload"); | |
| 170 Invalidation inv2 = Invalidation::Init(kObjectId_, 15, "payload"); | |
| 171 AckHandle inv1_handle = inv1.ack_handle(); | |
| 172 | |
| 173 unacked_invalidations_.Add(inv1); | |
| 174 unacked_invalidations_.Add(inv2); | |
| 175 | |
| 176 unacked_invalidations_.Drop(inv1_handle); | |
| 177 | |
| 178 SingleObjectInvalidationSet set = GetStoredInvalidations(); | |
| 179 ASSERT_EQ(2U, set.GetSize()); | |
| 180 EXPECT_TRUE(set.StartsWithUnknownVersion()); | |
| 181 EXPECT_EQ(15, set.rbegin()->version()); | |
| 182 } | |
| 183 | |
| 184 class UnackedInvalidationSetSerializationTest | |
| 185 : public UnackedInvalidationSetTest { | |
| 186 public: | |
| 187 UnackedInvalidationSet SerializeDeserialize() { | |
| 188 scoped_ptr<base::DictionaryValue> value = unacked_invalidations_.ToValue(); | |
| 189 UnackedInvalidationSet deserialized(kObjectId_); | |
| 190 deserialized.ResetFromValue(*value.get()); | |
| 191 return deserialized; | |
| 192 } | |
| 193 }; | |
| 194 | |
| 195 TEST_F(UnackedInvalidationSetSerializationTest, Empty) { | |
| 196 UnackedInvalidationSet deserialized = SerializeDeserialize(); | |
| 197 EXPECT_THAT(unacked_invalidations_, test_util::Eq(deserialized)); | |
| 198 } | |
| 199 | |
| 200 TEST_F(UnackedInvalidationSetSerializationTest, OneInvalidation) { | |
| 201 Invalidation inv = Invalidation::Init(kObjectId_, 10, "payload"); | |
| 202 unacked_invalidations_.Add(inv); | |
| 203 | |
| 204 UnackedInvalidationSet deserialized = SerializeDeserialize(); | |
| 205 EXPECT_THAT(unacked_invalidations_, test_util::Eq(deserialized)); | |
| 206 } | |
| 207 | |
| 208 TEST_F(UnackedInvalidationSetSerializationTest, WithUnknownVersion) { | |
| 209 Invalidation inv1 = Invalidation::Init(kObjectId_, 10, "payload"); | |
| 210 Invalidation inv2 = Invalidation::InitUnknownVersion(kObjectId_); | |
| 211 Invalidation inv3 = Invalidation::InitUnknownVersion(kObjectId_); | |
| 212 unacked_invalidations_.Add(inv1); | |
| 213 unacked_invalidations_.Add(inv2); | |
| 214 unacked_invalidations_.Add(inv3); | |
| 215 | |
| 216 UnackedInvalidationSet deserialized = SerializeDeserialize(); | |
| 217 EXPECT_THAT(unacked_invalidations_, test_util::Eq(deserialized)); | |
| 218 } | |
| 219 | |
| 220 } // namespace | |
| 221 | |
| 222 } // namespace syncer | |
| OLD | NEW |