Plumb invalidations from Tango to the extensions code for the Push Messaging API.

chrome/browser/extensions/api/push_messaging/push_messaging_apitest.cc

Index: chrome/browser/extensions/api/push_messaging/push_messaging_apitest.cc
diff --git a/chrome/browser/extensions/api/push_messaging/push_messaging_apitest.cc b/chrome/browser/extensions/api/push_messaging/push_messaging_apitest.cc
index c78f77a75f32ed901370b9a70b1916239277d7bb..77a58df6387c0e0094c12decab0f41ad6074a485 100644
--- a/chrome/browser/extensions/api/push_messaging/push_messaging_apitest.cc
+++ b/chrome/browser/extensions/api/push_messaging/push_messaging_apitest.cc
@@ -3,16 +3,35 @@
 // found in the LICENSE file.
 
 #include "chrome/browser/extensions/api/push_messaging/push_messaging_api.h"
+#include "chrome/browser/extensions/api/push_messaging/push_messaging_invalidation_mapper.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_service.h"
+#include "chrome/browser/extensions/extension_system.h"
 #include "chrome/browser/extensions/extension_test_message_listener.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/ui_test_utils.h"
+#include "testing/gmock/include/gmock/gmock.h"
+
+using ::testing::_;
+using ::testing::SaveArg;
+using ::testing::StrictMock;
 
 namespace extensions {
 
+class MockInvalidationMapper : public PushMessagingInvalidationMapper {
+ public:
+  MockInvalidationMapper();
+  ~MockInvalidationMapper();
+
+  MOCK_METHOD1(RegisterExtension, void(const std::string&));
+  MOCK_METHOD1(UnregisterExtension, void(const std::string&));
+};
+
+MockInvalidationMapper::MockInvalidationMapper() {}
+MockInvalidationMapper::~MockInvalidationMapper() {}
+
 class PushMessagingApiTest : public ExtensionApiTest {
  public:
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
@@ -34,11 +53,33 @@ IN_PROC_BROWSER_TEST_F(PushMessagingApiTest, EventDispatch) {
   EXPECT_TRUE(ready.WaitUntilSatisfied());
 
   // Trigger a callback.
-  browser()->profile()->GetExtensionService()->
+  ExtensionSystem::Get(browser()->profile())->extension_service()->
       push_messaging_event_router()->OnMessage(
           extension->id(), 1, "payload");
 
   EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
 }
 
+// Checks that an extension with the pushMessaging permission gets automatically
+// registered for invalidations when it is loaded.
+IN_PROC_BROWSER_TEST_F(PushMessagingApiTest, AutoRegistration) {
+  scoped_ptr<StrictMock<MockInvalidationMapper> > mapper(
+      new StrictMock<MockInvalidationMapper>);
+  StrictMock<MockInvalidationMapper>* unsafe_mapper = mapper.get();
+  // PushMessagingEventRouter owns the mapper now.
+  ExtensionSystem::Get(browser()->profile())->extension_service()->
+      push_messaging_event_router()->SetMapperForTest(
+          mapper.PassAs<PushMessagingInvalidationMapper>());
+
+  std::string extension_id;
+  EXPECT_CALL(*unsafe_mapper, RegisterExtension(_))
+      .WillOnce(SaveArg<0>(&extension_id));
+  const extensions::Extension* extension =
+      LoadExtension(test_data_dir_.AppendASCII("push_messaging"));
+  ASSERT_TRUE(extension);
+  EXPECT_EQ(extension->id(), extension_id);
+  EXPECT_CALL(*unsafe_mapper, UnregisterExtension(extension->id()));
+  UnloadExtension(extension->id());
+}
+
 }  // namespace extensions