Index: sync/notifier/invalidation_notifier_unittest.cc |
diff --git a/sync/notifier/invalidation_notifier_unittest.cc b/sync/notifier/invalidation_notifier_unittest.cc |
index 973c26f95a394d5a21b1819808c0ea481e6845a0..bdf88ebc93e7c26a953f2c460af29a5b76364cb5 100644 |
--- a/sync/notifier/invalidation_notifier_unittest.cc |
+++ b/sync/notifier/invalidation_notifier_unittest.cc |
@@ -11,11 +11,12 @@ |
#include "jingle/notifier/listener/fake_push_client.h" |
#include "net/url_request/url_request_test_util.h" |
#include "sync/internal_api/public/base/model_type.h" |
-#include "sync/internal_api/public/base/model_type_payload_map.h" |
+#include "sync/internal_api/public/base/model_type_state_map.h" |
#include "sync/internal_api/public/util/weak_handle.h" |
#include "sync/notifier/fake_invalidation_state_tracker.h" |
#include "sync/notifier/invalidation_state_tracker.h" |
#include "sync/notifier/mock_sync_notifier_observer.h" |
+#include "sync/notifier/object_id_state_map_test_util.h" |
#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
@@ -84,11 +85,11 @@ TEST_F(InvalidationNotifierTest, Basic) { |
CreateNotifier("fake_state"); |
const ModelTypeSet models(PREFERENCES, BOOKMARKS, AUTOFILL); |
- const ModelTypePayloadMap& type_payloads = |
- ModelTypePayloadMapFromEnumSet(models, "payload"); |
+ const ModelTypeStateMap& type_state_map = |
+ ModelTypeSetToStateMap(models, "payload"); |
EXPECT_CALL(mock_observer_, OnNotificationsEnabled()); |
EXPECT_CALL(mock_observer_, OnIncomingNotification( |
- ModelTypePayloadMapToObjectIdPayloadMap(type_payloads), |
+ ModelTypeStateMapToObjectIdStateMap(type_state_map), |
REMOTE_NOTIFICATION)); |
EXPECT_CALL(mock_observer_, |
OnNotificationsDisabled(TRANSIENT_NOTIFICATION_ERROR)); |
@@ -110,7 +111,7 @@ TEST_F(InvalidationNotifierTest, Basic) { |
invalidation_notifier_->OnNotificationsEnabled(); |
invalidation_notifier_->OnInvalidate( |
- ModelTypePayloadMapToObjectIdPayloadMap(type_payloads)); |
+ ModelTypeStateMapToObjectIdStateMap(type_state_map)); |
invalidation_notifier_->OnNotificationsDisabled( |
TRANSIENT_NOTIFICATION_ERROR); |