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 f237cdb8f4a06d32a6c8e8be7e676fa238bfadf6..1121b4b3acbdfe58632a03a096b8db408458c90e 100644 |
--- a/sync/notifier/non_blocking_invalidation_notifier_unittest.cc |
+++ b/sync/notifier/non_blocking_invalidation_notifier_unittest.cc |
@@ -12,10 +12,11 @@ |
#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" |
-#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/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" |
@@ -71,11 +72,11 @@ TEST_F(NonBlockingInvalidationNotifierTest, Basic) { |
InSequence dummy; |
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)); |
@@ -91,7 +92,7 @@ TEST_F(NonBlockingInvalidationNotifierTest, Basic) { |
invalidation_notifier_->OnNotificationsEnabled(); |
invalidation_notifier_->OnIncomingNotification( |
- ModelTypePayloadMapToObjectIdPayloadMap(type_payloads), |
+ ModelTypeStateMapToObjectIdStateMap(type_state_map), |
REMOTE_NOTIFICATION); |
invalidation_notifier_->OnNotificationsDisabled( |
TRANSIENT_NOTIFICATION_ERROR); |