Index: sync/notifier/non_blocking_invalidation_notifier_unittest.cc |
diff --git a/sync/notifier/non_blocking_invalidation_notifier_unittest.cc b/sync/notifier/non_blocking_invalidation_notifier_unittest.cc |
index bd8ccebbb52c713578295355bc61952fb9c0a4a4..4d472009859ff02bf916d53eb067a7a01a801a32 100644 |
--- a/sync/notifier/non_blocking_invalidation_notifier_unittest.cc |
+++ b/sync/notifier/non_blocking_invalidation_notifier_unittest.cc |
@@ -23,6 +23,7 @@ namespace syncer { |
namespace { |
using ::testing::InSequence; |
+using ::testing::Return; |
using ::testing::StrictMock; |
class NonBlockingInvalidationNotifierTest : public testing::Test { |
@@ -46,11 +47,11 @@ class NonBlockingInvalidationNotifierTest : public testing::Test { |
syncer::MakeWeakHandle( |
base::WeakPtr<syncer::InvalidationStateTracker>()), |
"fake_client_info")); |
- invalidation_notifier_->AddObserver(&mock_observer_); |
+ invalidation_notifier_->AddHandler(&mock_observer_); |
} |
virtual void TearDown() { |
- invalidation_notifier_->RemoveObserver(&mock_observer_); |
+ invalidation_notifier_->RemoveHandler(&mock_observer_); |
invalidation_notifier_.reset(); |
request_context_getter_ = NULL; |
io_thread_.Stop(); |
@@ -68,15 +69,20 @@ class NonBlockingInvalidationNotifierTest : public testing::Test { |
TEST_F(NonBlockingInvalidationNotifierTest, Basic) { |
InSequence dummy; |
- syncer::ModelTypePayloadMap type_payloads; |
- type_payloads[syncer::PREFERENCES] = "payload"; |
- type_payloads[syncer::BOOKMARKS] = ""; |
- type_payloads[syncer::AUTOFILL] = ""; |
+ ModelTypeSet models(syncer::PREFERENCES, |
+ syncer::BOOKMARKS, |
+ syncer::AUTOFILL); |
+ ModelTypePayloadMap type_payloads = |
+ ModelTypePayloadMapFromEnumSet(models, "payload"); |
+ |
+ EXPECT_CALL(mock_observer_, GetHandledIds()) |
+ .WillOnce(Return(ModelTypeSetToObjectIdSet(models))); |
+ invalidation_notifier_->ReloadHandlers(); |
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_, |
@@ -87,8 +93,9 @@ TEST_F(NonBlockingInvalidationNotifierTest, Basic) { |
invalidation_notifier_->UpdateCredentials("foo@bar.com", "fake_token"); |
invalidation_notifier_->OnNotificationsEnabled(); |
- invalidation_notifier_->OnIncomingNotification(type_payloads, |
- REMOTE_NOTIFICATION); |
+ invalidation_notifier_->OnIncomingNotification( |
+ ModelTypePayloadMapToObjectIdPayloadMap(type_payloads), |
+ REMOTE_NOTIFICATION); |
invalidation_notifier_->OnNotificationsDisabled( |
TRANSIENT_NOTIFICATION_ERROR); |
invalidation_notifier_->OnNotificationsDisabled( |