Browser test for the push messaging feature

chrome/browser/extensions/api/push_messaging/push_messaging_apitest.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 "chrome/browser/extensions/api/push_messaging/push_messaging_api.h"
 #include "chrome/browser/extensions/api/push_messaging/push_messaging_invalidation_handler.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/sync/profile_sync_service.h"
+#include "chrome/browser/sync/profile_sync_service_factory.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;
 
+// TODO(dcheng): This is hardcoded for now since the svn export is not done yet.
+// Once it's done, use ipc::invalidation::ObjectSource::CHROME_PUSH_MESSAGING.
+const int kSourceId = 1030;
+
 namespace extensions {
 
 class MockInvalidationMapper : public PushMessagingInvalidationMapper {
  public:
   MockInvalidationMapper();
   ~MockInvalidationMapper();
 
   MOCK_METHOD1(RegisterExtension, void(const std::string&));
   MOCK_METHOD1(UnregisterExtension, void(const std::string&));
 };
@@ skipping 24 matching lines @@
   ASSERT_TRUE(extension);
   GURL page_url = extension->GetResourceURL("event_dispatch.html");
   ui_test_utils::NavigateToURL(browser(), page_url);
   EXPECT_TRUE(ready.WaitUntilSatisfied());
 
   GetEventRouter()->TriggerMessageForTest(extension->id(), 1, "payload");
 
   EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
 }
 
+// Test that a push introduced into the sync code makes it to the extension
+// that we install.
+IN_PROC_BROWSER_TEST_F(PushMessagingApiTest, ReceivesPush) {
+  ResultCatcher catcher;
+  catcher.RestrictToProfile(browser()->profile());
+  ExtensionTestMessageListener ready("ready", true);
+
+  const extensions::Extension* extension =
+    LoadExtension(test_data_dir_.AppendASCII("push_messaging"));
+  ASSERT_TRUE(extension);
+  GURL page_url = extension->GetResourceURL("event_dispatch.html");
+  ui_test_utils::NavigateToURL(browser(), page_url);
+  EXPECT_TRUE(ready.WaitUntilSatisfied());
+
+  ProfileSyncService* pss = ProfileSyncServiceFactory::GetForProfile(
+      browser()->profile());
+  ASSERT_TRUE(pss);
+
+  // Construct a sync id for the object "U/<extension-id>/1".
+  std::string id("U/");
+  id += extension->id();
+  id += "/1";
+
+  invalidation::ObjectId object_id(kSourceId, id);
+
+  pss->SendInvalidationForTest(object_id, "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.
   GetEventRouter()->SetMapperForTest(
           mapper.PassAs<PushMessagingInvalidationMapper>());
 
@@ skipping 20 matching lines @@
 }
 
 IN_PROC_BROWSER_TEST_F(PushMessagingApiTest, Restart) {
   PushMessagingInvalidationHandler* handler =
       static_cast<PushMessagingInvalidationHandler*>(
           GetEventRouter()->GetMapperForTest());
   EXPECT_EQ(1U, handler->GetRegisteredExtensionsForTest().size());
 }
 
 }  // namespace extensions