Chromium Code Reviews| 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..8b7384a3423f6c07632ded6b8a4cbc1ac45bcd1c 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,18 @@ 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, |
|
akalin
2012/07/21 01:09:47
one line?
dcheng
2012/07/21 14:06:53
Done.
|
| + AUTOFILL); |
| + invalidation_notifier_->UpdateRegisteredIds( |
| + &mock_observer_, ModelTypeSetToObjectIdSet(models)); |
| + ModelTypePayloadMap type_payloads = |
|
akalin
2012/07/21 01:09:47
const ref
dcheng
2012/07/21 14:06:53
Done.
|
| + 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 +89,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( |