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 7a30f585fe48e5e4391fcfbf894ce3bd37b2295d..36f4532bb5d38b79ed2e6a29fa301855af0943e2 100644 |
--- a/sync/notifier/non_blocking_invalidation_notifier_unittest.cc |
+++ b/sync/notifier/non_blocking_invalidation_notifier_unittest.cc |
@@ -8,6 +8,7 @@ |
#include "base/memory/scoped_ptr.h" |
#include "base/message_loop.h" |
#include "base/threading/thread.h" |
+#include "google/cacheinvalidation/v2/types.pb.h" |
#include "jingle/notifier/base/fake_base_task.h" |
#include "net/url_request/url_request_test_util.h" |
#include "sync/internal_api/public/base/model_type.h" |
@@ -45,11 +46,10 @@ class NonBlockingInvalidationNotifierTest : public testing::Test { |
std::string(), // initial_invalidation_state |
MakeWeakHandle(base::WeakPtr<InvalidationStateTracker>()), |
"fake_client_info")); |
- invalidation_notifier_->AddObserver(&mock_observer_); |
} |
virtual void TearDown() { |
- invalidation_notifier_->RemoveObserver(&mock_observer_); |
+ invalidation_notifier_->UpdateRegisteredIds(&mock_observer_, ObjectIdSet()); |
invalidation_notifier_.reset(); |
request_context_getter_ = NULL; |
io_thread_.Stop(); |
@@ -67,15 +67,16 @@ class NonBlockingInvalidationNotifierTest : public testing::Test { |
TEST_F(NonBlockingInvalidationNotifierTest, Basic) { |
InSequence dummy; |
- ModelTypePayloadMap type_payloads; |
- type_payloads[PREFERENCES] = "payload"; |
- type_payloads[BOOKMARKS] = ""; |
- type_payloads[AUTOFILL] = ""; |
+ ModelTypeSet models(PREFERENCES, BOOKMARKS, AUTOFILL); |
+ invalidation_notifier_->UpdateRegisteredIds( |
+ &mock_observer_, ModelTypeSetToObjectIdSet(models)); |
+ const 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_, |
@@ -86,8 +87,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( |