[Invalidations] Add GetInvalidatorState() to Invalidator{,Frontend}

sync/notifier/p2p_invalidator_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "sync/notifier/p2p_invalidator.h"
 
 #include <cstddef>
 
 #include "jingle/notifier/listener/fake_push_client.h"
 #include "sync/internal_api/public/base/model_type.h"
@@ skipping 38 matching lines @@
 
   void DestroyInvalidator() {
     invalidator_.reset();
     fake_push_client_ = NULL;
   }
 
   void WaitForInvalidator() {
     // Do Nothing.
   }
 
-  void TriggerOnNotificationsEnabled() {
-    fake_push_client_->EnableNotifications();
+  void TriggerOnInvalidatorStateChange(InvalidatorState state) {
+    if (state == INVALIDATIONS_ENABLED) {
+      fake_push_client_->EnableNotifications();
+    } else {
+      fake_push_client_->DisableNotifications(ToNotifierReasonForTest(state));
+    }
   }
 
-  void TriggerOnIncomingNotification(const ObjectIdStateMap& id_state_map,
-                                     IncomingNotificationSource source) {
+  void TriggerOnIncomingInvalidation(const ObjectIdStateMap& id_state_map,
+                                     IncomingInvalidationSource source) {
     const P2PNotificationData notification_data(
         "", NOTIFY_ALL, id_state_map, source);
     notifier::Notification notification;
     notification.channel = kSyncP2PNotificationChannel;
     notification.data = notification_data.ToString();
     fake_push_client_->SimulateIncomingNotification(notification);
   }
 
-  void TriggerOnNotificationsDisabled(NotificationsDisabledReason reason) {
-    fake_push_client_->DisableNotifications(ToNotifierReasonForTest(reason));
-  }
-
   static bool InvalidatorHandlesDeprecatedState() {
     return false;
   }
 
  private:
   // Owned by |invalidator_|.
   notifier::FakePushClient* fake_push_client_;
   scoped_ptr<P2PInvalidator> invalidator_;
 };
 
@@ skipping 42 matching lines @@
     EXPECT_FALSE(target_str.empty());
     EXPECT_EQ(target, P2PNotificationTargetFromString(target_str));
   }
   EXPECT_EQ(NOTIFY_SELF, P2PNotificationTargetFromString("unknown"));
 }
 
 // Make sure notification targeting works correctly.
 TEST_F(P2PInvalidatorTest, P2PNotificationDataIsTargeted) {
   {
     const P2PNotificationData notification_data(
-        "sender", NOTIFY_SELF, ObjectIdStateMap(), REMOTE_NOTIFICATION);
+        "sender", NOTIFY_SELF, ObjectIdStateMap(), REMOTE_INVALIDATION);
     EXPECT_TRUE(notification_data.IsTargeted("sender"));
     EXPECT_FALSE(notification_data.IsTargeted("other1"));
     EXPECT_FALSE(notification_data.IsTargeted("other2"));
   }
   {
     const P2PNotificationData notification_data(
-        "sender", NOTIFY_OTHERS, ObjectIdStateMap(), REMOTE_NOTIFICATION);
+        "sender", NOTIFY_OTHERS, ObjectIdStateMap(), REMOTE_INVALIDATION);
     EXPECT_FALSE(notification_data.IsTargeted("sender"));
     EXPECT_TRUE(notification_data.IsTargeted("other1"));
     EXPECT_TRUE(notification_data.IsTargeted("other2"));
   }
   {
     const P2PNotificationData notification_data(
-        "sender", NOTIFY_ALL, ObjectIdStateMap(), REMOTE_NOTIFICATION);
+        "sender", NOTIFY_ALL, ObjectIdStateMap(), REMOTE_INVALIDATION);
     EXPECT_TRUE(notification_data.IsTargeted("sender"));
     EXPECT_TRUE(notification_data.IsTargeted("other1"));
     EXPECT_TRUE(notification_data.IsTargeted("other2"));
   }
 }
 
 // Make sure the P2PNotificationData <-> string conversions work for a
 // default-constructed P2PNotificationData.
 TEST_F(P2PInvalidatorTest, P2PNotificationDataDefault) {
   const P2PNotificationData notification_data;
@@ skipping 11 matching lines @@
   EXPECT_TRUE(notification_data.Equals(notification_data_parsed));
 }
 
 // Make sure the P2PNotificationData <-> string conversions work for a
 // non-default-constructed P2PNotificationData.
 TEST_F(P2PInvalidatorTest, P2PNotificationDataNonDefault) {
   const ObjectIdStateMap& id_state_map =
       ObjectIdSetToStateMap(
           ModelTypeSetToObjectIdSet(ModelTypeSet(BOOKMARKS, THEMES)), "");
   const P2PNotificationData notification_data(
-      "sender", NOTIFY_ALL, id_state_map, LOCAL_NOTIFICATION);
+      "sender", NOTIFY_ALL, id_state_map, LOCAL_INVALIDATION);
   EXPECT_TRUE(notification_data.IsTargeted("sender"));
   EXPECT_TRUE(notification_data.IsTargeted("other1"));
   EXPECT_TRUE(notification_data.IsTargeted("other2"));
   EXPECT_THAT(id_state_map, Eq(notification_data.GetIdStateMap()));
   const std::string& notification_data_str = notification_data.ToString();
   EXPECT_EQ(
       "{\"idStateMap\":["
       "{\"objectId\":{\"name\":\"BOOKMARK\",\"source\":1004},"
       "\"state\":{\"ackHandle\":{},\"payload\":\"\"}},"
       "{\"objectId\":{\"name\":\"THEME\",\"source\":1004},"
@@ skipping 30 matching lines @@
     expected_subscription.from = kEmail;
     EXPECT_TRUE(notifier::SubscriptionListsEqual(
         push_client->subscriptions(),
         notifier::SubscriptionList(1, expected_subscription)));
   }
   EXPECT_EQ(kEmail, push_client->email());
   EXPECT_EQ(kToken, push_client->token());
 
   ReflectSentNotifications();
   push_client->EnableNotifications();
-  EXPECT_EQ(NO_NOTIFICATION_ERROR,
-            fake_handler_.GetNotificationsDisabledReason());
+  EXPECT_EQ(INVALIDATIONS_ENABLED, fake_handler_.GetInvalidatorState());
 
   ReflectSentNotifications();
-  EXPECT_EQ(1, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(1, fake_handler_.GetInvalidationCount());
   EXPECT_THAT(
       ModelTypeStateMapToObjectIdStateMap(MakeStateMap(enabled_types)),
-      Eq(fake_handler_.GetLastNotificationIdStateMap()));
-  EXPECT_EQ(REMOTE_NOTIFICATION, fake_handler_.GetLastNotificationSource());
+      Eq(fake_handler_.GetLastInvalidationIdStateMap()));
+  EXPECT_EQ(REMOTE_INVALIDATION, fake_handler_.GetLastInvalidationSource());
 
   // Sent with target NOTIFY_OTHERS so should not be propagated to
   // |fake_handler_|.
   {
     const ObjectIdStateMap& id_state_map =
         ObjectIdSetToStateMap(
             ModelTypeSetToObjectIdSet(ModelTypeSet(THEMES, APPS)), "");
-    invalidator->SendNotification(id_state_map);
+    invalidator->SendInvalidation(id_state_map);
   }
 
   ReflectSentNotifications();
-  EXPECT_EQ(1, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(1, fake_handler_.GetInvalidationCount());
 }
 
 // Set up the P2PInvalidator and send out notifications with various
 // target settings.  The notifications received by the observer should
 // be consistent with the target settings.
 TEST_F(P2PInvalidatorTest, SendNotificationData) {
   const ModelTypeSet enabled_types(BOOKMARKS, PREFERENCES, THEMES);
   const ModelTypeSet changed_types(THEMES, APPS);
   const ModelTypeSet expected_types(THEMES);
 
   const ObjectIdStateMap& id_state_map =
       ObjectIdSetToStateMap(
           ModelTypeSetToObjectIdSet(changed_types), "");
 
   P2PInvalidator* const invalidator = delegate_.GetInvalidator();
   notifier::FakePushClient* const push_client = delegate_.GetPushClient();
 
   invalidator->UpdateRegisteredIds(&fake_handler_,
                                    ModelTypeSetToObjectIdSet(enabled_types));
 
   invalidator->SetUniqueId("sender");
   invalidator->UpdateCredentials("foo@bar.com", "fake_token");
 
   ReflectSentNotifications();
   push_client->EnableNotifications();
-  EXPECT_EQ(NO_NOTIFICATION_ERROR,
-            fake_handler_.GetNotificationsDisabledReason());
+  EXPECT_EQ(INVALIDATIONS_ENABLED, fake_handler_.GetInvalidatorState());
 
   ReflectSentNotifications();
-  EXPECT_EQ(1, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(1, fake_handler_.GetInvalidationCount());
   EXPECT_THAT(
       ModelTypeStateMapToObjectIdStateMap(MakeStateMap(enabled_types)),
-      Eq(fake_handler_.GetLastNotificationIdStateMap()));
-  EXPECT_EQ(REMOTE_NOTIFICATION, fake_handler_.GetLastNotificationSource());
+      Eq(fake_handler_.GetLastInvalidationIdStateMap()));
+  EXPECT_EQ(REMOTE_INVALIDATION, fake_handler_.GetLastInvalidationSource());
 
   // Should be dropped.
   invalidator->SendNotificationDataForTest(P2PNotificationData());
   ReflectSentNotifications();
-  EXPECT_EQ(1, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(1, fake_handler_.GetInvalidationCount());
 
   const ObjectIdStateMap& expected_ids =
       ModelTypeStateMapToObjectIdStateMap(MakeStateMap(expected_types));
 
   // Should be propagated.
   invalidator->SendNotificationDataForTest(
       P2PNotificationData("sender", NOTIFY_SELF,
-                          id_state_map, REMOTE_NOTIFICATION));
+                          id_state_map, REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(2, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(2, fake_handler_.GetInvalidationCount());
   EXPECT_THAT(
       expected_ids,
-      Eq(fake_handler_.GetLastNotificationIdStateMap()));
+      Eq(fake_handler_.GetLastInvalidationIdStateMap()));
 
   // Should be dropped.
   invalidator->SendNotificationDataForTest(
       P2PNotificationData("sender2", NOTIFY_SELF,
-                          id_state_map, REMOTE_NOTIFICATION));
+                          id_state_map, REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(2, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(2, fake_handler_.GetInvalidationCount());
 
   // Should be dropped.
   invalidator->SendNotificationDataForTest(
       P2PNotificationData("sender", NOTIFY_SELF,
-                          ObjectIdStateMap(), REMOTE_NOTIFICATION));
+                          ObjectIdStateMap(), REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(2, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(2, fake_handler_.GetInvalidationCount());
 
   // Should be dropped.
   invalidator->SendNotificationDataForTest(
       P2PNotificationData("sender", NOTIFY_OTHERS,
-                          id_state_map, REMOTE_NOTIFICATION));
+                          id_state_map, REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(2, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(2, fake_handler_.GetInvalidationCount());
 
   // Should be propagated.
   invalidator->SendNotificationDataForTest(
       P2PNotificationData("sender2", NOTIFY_OTHERS,
-                          id_state_map, REMOTE_NOTIFICATION));
+                          id_state_map, REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(3, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(3, fake_handler_.GetInvalidationCount());
   EXPECT_THAT(
       expected_ids,
-      Eq(fake_handler_.GetLastNotificationIdStateMap()));
+      Eq(fake_handler_.GetLastInvalidationIdStateMap()));
 
   // Should be dropped.
   invalidator->SendNotificationDataForTest(
       P2PNotificationData("sender2", NOTIFY_OTHERS,
-                          ObjectIdStateMap(), REMOTE_NOTIFICATION));
+                          ObjectIdStateMap(), REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(3, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(3, fake_handler_.GetInvalidationCount());
 
   // Should be propagated.
   invalidator->SendNotificationDataForTest(
       P2PNotificationData("sender", NOTIFY_ALL,
-                          id_state_map, REMOTE_NOTIFICATION));
+                          id_state_map, REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(4, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(4, fake_handler_.GetInvalidationCount());
   EXPECT_THAT(
       expected_ids,
-      Eq(fake_handler_.GetLastNotificationIdStateMap()));
+      Eq(fake_handler_.GetLastInvalidationIdStateMap()));
 
   // Should be propagated.
   invalidator->SendNotificationDataForTest(
       P2PNotificationData("sender2", NOTIFY_ALL,
-                          id_state_map, REMOTE_NOTIFICATION));
+                          id_state_map, REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(5, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(5, fake_handler_.GetInvalidationCount());
   EXPECT_THAT(
       expected_ids,
-      Eq(fake_handler_.GetLastNotificationIdStateMap()));
+      Eq(fake_handler_.GetLastInvalidationIdStateMap()));
 
   // Should be dropped.
   invalidator->SendNotificationDataForTest(
   P2PNotificationData("sender2", NOTIFY_ALL,
-                      ObjectIdStateMap(), REMOTE_NOTIFICATION));
+                      ObjectIdStateMap(), REMOTE_INVALIDATION));
   ReflectSentNotifications();
-  EXPECT_EQ(5, fake_handler_.GetNotificationCount());
+  EXPECT_EQ(5, fake_handler_.GetInvalidationCount());
 }
 
 INSTANTIATE_TYPED_TEST_CASE_P(
     P2PInvalidatorTest, InvalidatorTest,
     P2PInvalidatorTestDelegate);
 
 }  // namespace
 
 }  // namespace syncer