Index: sync/notifier/invalidation_notifier_unittest.cc |
diff --git a/sync/notifier/invalidation_notifier_unittest.cc b/sync/notifier/invalidation_notifier_unittest.cc |
index 0329ad95861ac397726a527a72f5174cc7f2ebb9..d74c22d2fb508e5c20dee62b5f47732e52f31ee6 100644 |
--- a/sync/notifier/invalidation_notifier_unittest.cc |
+++ b/sync/notifier/invalidation_notifier_unittest.cc |
@@ -50,11 +50,10 @@ class InvalidationNotifierTest : public testing::Test { |
initial_invalidation_state, |
syncer::MakeWeakHandle(mock_tracker_.AsWeakPtr()), |
"fake_client_info")); |
- invalidation_notifier_->AddObserver(&mock_observer_); |
} |
void ResetNotifier() { |
- invalidation_notifier_->RemoveObserver(&mock_observer_); |
+ invalidation_notifier_->UpdateRegisteredIds(&mock_observer_, ObjectIdSet()); |
// Stopping the invalidation notifier stops its scheduler, which deletes any |
// pending tasks without running them. Some tasks "run and delete" another |
// task, so they must be run in order to avoid leaking the inner task. |
@@ -75,15 +74,18 @@ TEST_F(InvalidationNotifierTest, Basic) { |
CreateAndObserveNotifier("fake_state"); |
InSequence dummy; |
- syncer::ModelTypePayloadMap type_payloads; |
- type_payloads[syncer::PREFERENCES] = "payload"; |
- type_payloads[syncer::BOOKMARKS] = "payload"; |
- type_payloads[syncer::AUTOFILL] = "payload"; |
+ ModelTypeSet models(syncer::PREFERENCES, |
+ syncer::BOOKMARKS, |
+ syncer::AUTOFILL); |
+ invalidation_notifier_->UpdateRegisteredIds( |
+ &mock_observer_, ModelTypeSetToObjectIdSet(models)); |
+ ModelTypePayloadMap type_payloads = |
+ ModelTypePayloadMapFromEnumSet(models, "payload"); |
EXPECT_CALL(mock_observer_, OnNotificationsEnabled()); |
- EXPECT_CALL(mock_observer_, |
- OnIncomingNotification(type_payloads, |
- REMOTE_NOTIFICATION)); |
+ EXPECT_CALL(mock_observer_, OnIncomingNotification( |
+ ModelTypePayloadMapToObjectIdPayloadMap(type_payloads), |
+ REMOTE_NOTIFICATION)); |
EXPECT_CALL(mock_observer_, |
OnNotificationsDisabled(TRANSIENT_NOTIFICATION_ERROR)); |
EXPECT_CALL(mock_observer_, |
@@ -99,14 +101,8 @@ TEST_F(InvalidationNotifierTest, Basic) { |
invalidation_notifier_->OnNotificationsEnabled(); |
- ObjectIdPayloadMap id_payloads; |
- for (syncer::ModelTypePayloadMap::const_iterator it = type_payloads.begin(); |
- it != type_payloads.end(); ++it) { |
- invalidation::ObjectId id; |
- ASSERT_TRUE(RealModelTypeToObjectId(it->first, &id)); |
- id_payloads[id] = "payload"; |
- } |
- invalidation_notifier_->OnInvalidate(id_payloads); |
+ invalidation_notifier_->OnInvalidate( |
+ ModelTypePayloadMapToObjectIdPayloadMap(type_payloads)); |
invalidation_notifier_->OnNotificationsDisabled( |
TRANSIENT_NOTIFICATION_ERROR); |