Add KeepaliveImpulse to extension process manager.

chrome/browser/extensions/lazy_background_page_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 "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/bookmarks/bookmark_model.h"
 #include "chrome/browser/bookmarks/bookmark_model_factory.h"
 #include "chrome/browser/bookmarks/bookmark_test_helpers.h"
@@ skipping 63 matching lines @@
 };
 
 }  // namespace
 
 class LazyBackgroundPageApiTest : public ExtensionApiTest {
  public:
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     ExtensionApiTest::SetUpCommandLine(command_line);
     // Set shorter delays to prevent test timeouts.
     command_line->AppendSwitchASCII(
-        extensions::switches::kEventPageIdleTime, "1");
+        extensions::switches::kEventPageIdleTime, "1000");
     command_line->AppendSwitchASCII(
-        extensions::switches::kEventPageSuspendingTime, "1");
+        extensions::switches::kEventPageSuspendingTime, "1000");
   }
 
   // Loads the extension, which temporarily starts the lazy background page
   // to dispatch the onInstalled event. We wait until it shuts down again.
   const Extension* LoadExtensionAndWait(const std::string& test_name) {
     LazyBackgroundObserver page_complete;
     base::FilePath extdir = test_data_dir_.AppendASCII("lazy_background_page").
         AppendASCII(test_name);
     const Extension* extension = LoadExtension(extdir);
     if (extension)
@@ skipping 307 matching lines @@
   EXPECT_TRUE(pm->GetBackgroundHostForExtension(last_loaded_extension_id()));
 
   // Navigate away, closing the message channel and therefore the background
   // page.
   ui_test_utils::NavigateToURL(browser(), GURL("about:blank"));
   lazybg.WaitUntilClosed();
 
   EXPECT_FALSE(pm->GetBackgroundHostForExtension(last_loaded_extension_id()));
 }
 
+// Tests that a KeepaliveImpulse increments the keep alive count, but eventually
+// times out and background page will still close.
+IN_PROC_BROWSER_TEST_F(LazyBackgroundPageApiTest, ImpulseAddsCount) {
+  ASSERT_TRUE(StartEmbeddedTestServer());
+  const Extension* extension = LoadExtensionAndWait("messaging");
+  ASSERT_TRUE(extension);
+
+  // Lazy Background Page doesn't exist yet.
+  extensions::ProcessManager* pm =
+      extensions::ExtensionSystem::Get(browser()->profile())->process_manager();
+  EXPECT_FALSE(pm->GetBackgroundHostForExtension(last_loaded_extension_id()));
+  EXPECT_EQ(1, browser()->tab_strip_model()->count());
+
+  // Navigate to a page that opens a message channel to the background page.
+  ResultCatcher catcher;
+  LazyBackgroundObserver lazybg;
+  ui_test_utils::NavigateToURL(
+      browser(), embedded_test_server()->GetURL("/extensions/test_file.html"));
+  lazybg.WaitUntilLoaded();
+
+  // Add an impulse and the keep alive count increases.
+  int previous_keep_alive_count = pm->GetLazyKeepaliveCount(extension);
+  pm->KeepaliveImpulse(extension);
+  EXPECT_EQ(previous_keep_alive_count + 1,
+            pm->GetLazyKeepaliveCount(extension));
+
+  // Navigate away, closing the message channel and therefore the background
+  // page after the impulse times out.
+  ui_test_utils::NavigateToURL(browser(), GURL("about:blank"));
+  lazybg.WaitUntilClosed();
+
+  EXPECT_FALSE(pm->GetBackgroundHostForExtension(last_loaded_extension_id()));
+}
+
 // Tests that the lazy background page receives the unload event when we
 // close it, and that it can execute simple API calls that don't require an
 // asynchronous response.
 IN_PROC_BROWSER_TEST_F(LazyBackgroundPageApiTest, OnUnload) {
   ASSERT_TRUE(LoadExtensionAndWait("on_unload"));
 
   // Lazy Background Page has been shut down.
   extensions::ProcessManager* pm =
       extensions::ExtensionSystem::Get(browser()->profile())->process_manager();
   EXPECT_FALSE(pm->GetBackgroundHostForExtension(last_loaded_extension_id()));
@@ skipping 85 matching lines @@
 
 // Tests that the lazy background page will be unloaded if the onSuspend event
 // handler calls an API function such as chrome.storage.local.set().
 // See: http://crbug.com/296834
 IN_PROC_BROWSER_TEST_F(LazyBackgroundPageApiTest, OnSuspendUseStorageApi) {
   EXPECT_TRUE(LoadExtensionAndWait("on_suspend"));
 }
 
 // TODO: background page with timer.
 // TODO: background page that interacts with popup.