OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 // | 4 // |
5 | 5 |
6 #include "chrome/browser/sync/invalidations/invalidator_storage.h" | 6 #include "chrome/browser/sync/invalidations/invalidator_storage.h" |
7 | 7 |
8 #include "base/bind.h" | 8 #include "base/bind.h" |
9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
10 #include "base/message_loop_proxy.h" | 10 #include "base/message_loop_proxy.h" |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
54 const invalidation::ObjectId kPreferencesId_; | 54 const invalidation::ObjectId kPreferencesId_; |
55 const invalidation::ObjectId kAppNotificationsId_; | 55 const invalidation::ObjectId kAppNotificationsId_; |
56 const invalidation::ObjectId kAutofillId_; | 56 const invalidation::ObjectId kAutofillId_; |
57 | 57 |
58 MessageLoop loop_; | 58 MessageLoop loop_; |
59 }; | 59 }; |
60 | 60 |
61 // Set invalidation states for various keys and verify that they are written and | 61 // Set invalidation states for various keys and verify that they are written and |
62 // read back correctly. | 62 // read back correctly. |
63 TEST_F(InvalidatorStorageTest, SetMaxVersionAndPayload) { | 63 TEST_F(InvalidatorStorageTest, SetMaxVersionAndPayload) { |
64 InvalidatorStorage storage(&pref_service_); | 64 InvalidatorStorage storage(&pref_service_, pref_service_.registry()); |
65 | 65 |
66 InvalidationStateMap expected_states; | 66 InvalidationStateMap expected_states; |
67 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); | 67 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); |
68 | 68 |
69 expected_states[kBookmarksId_].version = 2; | 69 expected_states[kBookmarksId_].version = 2; |
70 expected_states[kBookmarksId_].payload = "hello"; | 70 expected_states[kBookmarksId_].payload = "hello"; |
71 storage.SetMaxVersionAndPayload(kBookmarksId_, 2, "hello"); | 71 storage.SetMaxVersionAndPayload(kBookmarksId_, 2, "hello"); |
72 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); | 72 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); |
73 | 73 |
74 expected_states[kPreferencesId_].version = 5; | 74 expected_states[kPreferencesId_].version = 5; |
75 storage.SetMaxVersionAndPayload(kPreferencesId_, 5, std::string()); | 75 storage.SetMaxVersionAndPayload(kPreferencesId_, 5, std::string()); |
76 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); | 76 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); |
77 | 77 |
78 expected_states[kAppNotificationsId_].version = 3; | 78 expected_states[kAppNotificationsId_].version = 3; |
79 expected_states[kAppNotificationsId_].payload = "world"; | 79 expected_states[kAppNotificationsId_].payload = "world"; |
80 storage.SetMaxVersionAndPayload(kAppNotificationsId_, 3, "world"); | 80 storage.SetMaxVersionAndPayload(kAppNotificationsId_, 3, "world"); |
81 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); | 81 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); |
82 | 82 |
83 expected_states[kAppNotificationsId_].version = 4; | 83 expected_states[kAppNotificationsId_].version = 4; |
84 expected_states[kAppNotificationsId_].payload = "again"; | 84 expected_states[kAppNotificationsId_].payload = "again"; |
85 storage.SetMaxVersionAndPayload(kAppNotificationsId_, 4, "again"); | 85 storage.SetMaxVersionAndPayload(kAppNotificationsId_, 4, "again"); |
86 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); | 86 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); |
87 } | 87 } |
88 | 88 |
89 // Forgetting an entry should cause that entry to be deleted. | 89 // Forgetting an entry should cause that entry to be deleted. |
90 TEST_F(InvalidatorStorageTest, Forget) { | 90 TEST_F(InvalidatorStorageTest, Forget) { |
91 InvalidatorStorage storage(&pref_service_); | 91 InvalidatorStorage storage(&pref_service_, pref_service_.registry()); |
92 EXPECT_TRUE(storage.GetAllInvalidationStates().empty()); | 92 EXPECT_TRUE(storage.GetAllInvalidationStates().empty()); |
93 | 93 |
94 InvalidationStateMap expected_states; | 94 InvalidationStateMap expected_states; |
95 expected_states[kBookmarksId_].version = 2; | 95 expected_states[kBookmarksId_].version = 2; |
96 expected_states[kBookmarksId_].payload = "a"; | 96 expected_states[kBookmarksId_].payload = "a"; |
97 expected_states[kPreferencesId_].version = 5; | 97 expected_states[kPreferencesId_].version = 5; |
98 expected_states[kPreferencesId_].payload = "b"; | 98 expected_states[kPreferencesId_].payload = "b"; |
99 storage.SetMaxVersionAndPayload(kBookmarksId_, 2, "a"); | 99 storage.SetMaxVersionAndPayload(kBookmarksId_, 2, "a"); |
100 storage.SetMaxVersionAndPayload(kPreferencesId_, 5, "b"); | 100 storage.SetMaxVersionAndPayload(kPreferencesId_, 5, "b"); |
101 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); | 101 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); |
102 | 102 |
103 expected_states.erase(kPreferencesId_); | 103 expected_states.erase(kPreferencesId_); |
104 syncer::ObjectIdSet to_forget; | 104 syncer::ObjectIdSet to_forget; |
105 to_forget.insert(kPreferencesId_); | 105 to_forget.insert(kPreferencesId_); |
106 storage.Forget(to_forget); | 106 storage.Forget(to_forget); |
107 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); | 107 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); |
108 } | 108 } |
109 | 109 |
110 // Clearing the storage should erase all version map entries and the bootstrap | 110 // Clearing the storage should erase all version map entries and the bootstrap |
111 // data. | 111 // data. |
112 TEST_F(InvalidatorStorageTest, Clear) { | 112 TEST_F(InvalidatorStorageTest, Clear) { |
113 InvalidatorStorage storage(&pref_service_); | 113 InvalidatorStorage storage(&pref_service_, pref_service_.registry()); |
114 EXPECT_TRUE(storage.GetAllInvalidationStates().empty()); | 114 EXPECT_TRUE(storage.GetAllInvalidationStates().empty()); |
115 EXPECT_TRUE(storage.GetBootstrapData().empty()); | 115 EXPECT_TRUE(storage.GetBootstrapData().empty()); |
116 | 116 |
117 storage.SetBootstrapData("test"); | 117 storage.SetBootstrapData("test"); |
118 EXPECT_EQ("test", storage.GetBootstrapData()); | 118 EXPECT_EQ("test", storage.GetBootstrapData()); |
119 { | 119 { |
120 InvalidationStateMap expected_states; | 120 InvalidationStateMap expected_states; |
121 expected_states[kAppNotificationsId_].version = 3; | 121 expected_states[kAppNotificationsId_].version = 3; |
122 storage.SetMaxVersionAndPayload(kAppNotificationsId_, 3, std::string()); | 122 storage.SetMaxVersionAndPayload(kAppNotificationsId_, 3, std::string()); |
123 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); | 123 EXPECT_EQ(expected_states, storage.GetAllInvalidationStates()); |
(...skipping 251 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
375 } | 375 } |
376 | 376 |
377 // Test that the migration code for the legacy preference works as expected. | 377 // Test that the migration code for the legacy preference works as expected. |
378 // Migration should happen on construction of InvalidatorStorage. | 378 // Migration should happen on construction of InvalidatorStorage. |
379 TEST_F(InvalidatorStorageTest, MigrateLegacyPreferences) { | 379 TEST_F(InvalidatorStorageTest, MigrateLegacyPreferences) { |
380 base::DictionaryValue* legacy_dict = new DictionaryValue; | 380 base::DictionaryValue* legacy_dict = new DictionaryValue; |
381 legacy_dict->SetString(base::IntToString(syncer::AUTOFILL), "10"); | 381 legacy_dict->SetString(base::IntToString(syncer::AUTOFILL), "10"); |
382 legacy_dict->SetString(base::IntToString(syncer::BOOKMARKS), "32"); | 382 legacy_dict->SetString(base::IntToString(syncer::BOOKMARKS), "32"); |
383 legacy_dict->SetString(base::IntToString(syncer::PREFERENCES), "54"); | 383 legacy_dict->SetString(base::IntToString(syncer::PREFERENCES), "54"); |
384 pref_service_.SetUserPref(prefs::kSyncMaxInvalidationVersions, legacy_dict); | 384 pref_service_.SetUserPref(prefs::kSyncMaxInvalidationVersions, legacy_dict); |
385 InvalidatorStorage storage(&pref_service_); | 385 InvalidatorStorage storage(&pref_service_, pref_service_.registry()); |
386 | 386 |
387 // Legacy pref should be cleared. | 387 // Legacy pref should be cleared. |
388 const base::DictionaryValue* dict = | 388 const base::DictionaryValue* dict = |
389 pref_service_.GetDictionary(prefs::kSyncMaxInvalidationVersions); | 389 pref_service_.GetDictionary(prefs::kSyncMaxInvalidationVersions); |
390 EXPECT_TRUE(dict->empty()); | 390 EXPECT_TRUE(dict->empty()); |
391 | 391 |
392 // Validate the new pref is set correctly. | 392 // Validate the new pref is set correctly. |
393 InvalidationStateMap map; | 393 InvalidationStateMap map; |
394 const base::ListValue* list = | 394 const base::ListValue* list = |
395 pref_service_.GetList(prefs::kInvalidatorMaxInvalidationVersions); | 395 pref_service_.GetList(prefs::kInvalidatorMaxInvalidationVersions); |
396 InvalidatorStorage::DeserializeFromList(*list, &map); | 396 InvalidatorStorage::DeserializeFromList(*list, &map); |
397 | 397 |
398 EXPECT_EQ(3U, map.size()); | 398 EXPECT_EQ(3U, map.size()); |
399 EXPECT_EQ(10, map[kAutofillId_].version); | 399 EXPECT_EQ(10, map[kAutofillId_].version); |
400 EXPECT_EQ(32, map[kBookmarksId_].version); | 400 EXPECT_EQ(32, map[kBookmarksId_].version); |
401 EXPECT_EQ(54, map[kPreferencesId_].version); | 401 EXPECT_EQ(54, map[kPreferencesId_].version); |
402 } | 402 } |
403 | 403 |
404 TEST_F(InvalidatorStorageTest, SetGetBootstrapData) { | 404 TEST_F(InvalidatorStorageTest, SetGetBootstrapData) { |
405 InvalidatorStorage storage(&pref_service_); | 405 InvalidatorStorage storage(&pref_service_, pref_service_.registry()); |
406 const std::string mess("n\0tK\0\0l\344", 8); | 406 const std::string mess("n\0tK\0\0l\344", 8); |
407 ASSERT_FALSE(IsStringUTF8(mess)); | 407 ASSERT_FALSE(IsStringUTF8(mess)); |
408 | 408 |
409 storage.SetBootstrapData(mess); | 409 storage.SetBootstrapData(mess); |
410 EXPECT_EQ(mess, storage.GetBootstrapData()); | 410 EXPECT_EQ(mess, storage.GetBootstrapData()); |
411 } | 411 } |
412 | 412 |
413 // Test that we correctly generate ack handles, acknowledge them, and persist | 413 // Test that we correctly generate ack handles, acknowledge them, and persist |
414 // them. | 414 // them. |
415 TEST_F(InvalidatorStorageTest, GenerateAckHandlesAndAcknowledge) { | 415 TEST_F(InvalidatorStorageTest, GenerateAckHandlesAndAcknowledge) { |
416 InvalidatorStorage storage(&pref_service_); | 416 InvalidatorStorage storage(&pref_service_, pref_service_.registry()); |
417 syncer::ObjectIdSet ids; | 417 syncer::ObjectIdSet ids; |
418 InvalidationStateMap state_map; | 418 InvalidationStateMap state_map; |
419 syncer::AckHandleMap ack_handle_map; | 419 syncer::AckHandleMap ack_handle_map; |
420 syncer::AckHandleMap::const_iterator it; | 420 syncer::AckHandleMap::const_iterator it; |
421 | 421 |
422 // Test that it works as expected if the key doesn't already exist in the map, | 422 // Test that it works as expected if the key doesn't already exist in the map, |
423 // e.g. the first invalidation received for the object ID was not for a | 423 // e.g. the first invalidation received for the object ID was not for a |
424 // specific version. | 424 // specific version. |
425 ids.insert(kAutofillId_); | 425 ids.insert(kAutofillId_); |
426 storage.GenerateAckHandles( | 426 storage.GenerateAckHandles( |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
480 EXPECT_TRUE(it->second.IsValid()); | 480 EXPECT_TRUE(it->second.IsValid()); |
481 state_map[kBookmarksId_].expected = it->second; | 481 state_map[kBookmarksId_].expected = it->second; |
482 EXPECT_EQ(state_map, storage.GetAllInvalidationStates()); | 482 EXPECT_EQ(state_map, storage.GetAllInvalidationStates()); |
483 | 483 |
484 storage.Acknowledge(kBookmarksId_, it->second); | 484 storage.Acknowledge(kBookmarksId_, it->second); |
485 state_map[kBookmarksId_].current = it->second; | 485 state_map[kBookmarksId_].current = it->second; |
486 EXPECT_EQ(state_map, storage.GetAllInvalidationStates()); | 486 EXPECT_EQ(state_map, storage.GetAllInvalidationStates()); |
487 } | 487 } |
488 | 488 |
489 } // namespace browser_sync | 489 } // namespace browser_sync |
OLD | NEW |