| Index: sync/notifier/p2p_invalidator_unittest.cc
|
| diff --git a/sync/notifier/p2p_invalidator_unittest.cc b/sync/notifier/p2p_invalidator_unittest.cc
|
| index be9ee3f970977a3beb929d56349c4720bdaa225f..24cfe027f6cb83df84f6ba26c8e76fcce64d1476 100644
|
| --- a/sync/notifier/p2p_invalidator_unittest.cc
|
| +++ b/sync/notifier/p2p_invalidator_unittest.cc
|
| @@ -8,7 +8,6 @@
|
|
|
| #include "jingle/notifier/listener/fake_push_client.h"
|
| #include "sync/internal_api/public/base/model_type.h"
|
| -#include "sync/internal_api/public/base/model_type_invalidation_map.h"
|
| #include "sync/notifier/fake_invalidation_handler.h"
|
| #include "sync/notifier/invalidator_test_template.h"
|
| #include "sync/notifier/object_id_invalidation_map_test_util.h"
|
| @@ -96,8 +95,12 @@ class P2PInvalidatorTest : public testing::Test {
|
| delegate_.GetInvalidator()->UnregisterHandler(&fake_handler_);
|
| }
|
|
|
| - ModelTypeInvalidationMap MakeInvalidationMap(ModelTypeSet types) {
|
| - return ModelTypeSetToInvalidationMap(types, std::string());
|
| + ObjectIdInvalidationMap MakeInvalidationMap(ModelTypeSet types) {
|
| + ObjectIdInvalidationMap invalidations;
|
| + ObjectIdSet ids = ModelTypeSetToObjectIdSet(types);
|
| + return ObjectIdSetToInvalidationMap(ids,
|
| + Invalidation::kUnknownVersion,
|
| + std::string());
|
| }
|
|
|
| // Simulate receiving all the notifications we sent out since last
|
| @@ -240,8 +243,7 @@ TEST_F(P2PInvalidatorTest, NotificationsBasic) {
|
| ReflectSentNotifications();
|
| EXPECT_EQ(1, fake_handler_.GetInvalidationCount());
|
| EXPECT_THAT(
|
| - ModelTypeInvalidationMapToObjectIdInvalidationMap(
|
| - MakeInvalidationMap(enabled_types)),
|
| + MakeInvalidationMap(enabled_types),
|
| Eq(fake_handler_.GetLastInvalidationMap()));
|
|
|
| // Sent with target NOTIFY_OTHERS so should not be propagated to
|
| @@ -286,10 +288,8 @@ TEST_F(P2PInvalidatorTest, SendNotificationData) {
|
|
|
| ReflectSentNotifications();
|
| EXPECT_EQ(1, fake_handler_.GetInvalidationCount());
|
| - EXPECT_THAT(
|
| - ModelTypeInvalidationMapToObjectIdInvalidationMap(
|
| - MakeInvalidationMap(enabled_types)),
|
| - Eq(fake_handler_.GetLastInvalidationMap()));
|
| + EXPECT_THAT(MakeInvalidationMap(enabled_types),
|
| + Eq(fake_handler_.GetLastInvalidationMap()));
|
|
|
| // Should be dropped.
|
| invalidator->SendNotificationDataForTest(P2PNotificationData());
|
| @@ -297,8 +297,7 @@ TEST_F(P2PInvalidatorTest, SendNotificationData) {
|
| EXPECT_EQ(1, fake_handler_.GetInvalidationCount());
|
|
|
| const ObjectIdInvalidationMap& expected_ids =
|
| - ModelTypeInvalidationMapToObjectIdInvalidationMap(
|
| - MakeInvalidationMap(expected_types));
|
| + MakeInvalidationMap(expected_types);
|
|
|
| // Should be propagated.
|
| invalidator->SendNotificationDataForTest(
|
|
|