Add PyAuto tests for triggering browser/page action.

chrome/test/functional/extensions.py

 #!/usr/bin/env python
 # Copyright (c) 2011 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.
 
 """
 This module is a simple qa tool that installs extensions and tests whether the
 browser crashes while visiting a list of urls.
 
 Usage: python extensions.py -v
@@ skipping 163 matching lines @@
     # Allow extension in incognito mode and verify.
     self.SetExtensionStateById(ext_id, enable=True, allow_in_incognito=True)
     extension = self._GetExtensionInfoById(self.GetExtensionsInfo(), ext_id)
     self.assertTrue(extension['allowed_in_incognito'])
 
     # Disallow extension in incognito mode and verify.
     self.SetExtensionStateById(ext_id, enable=True, allow_in_incognito=False)
     extension = self._GetExtensionInfoById(self.GetExtensionsInfo(), ext_id)
     self.assertFalse(extension['allowed_in_incognito'])
 
+  def testTriggerBrowserAction(self):
+    """Test triggering browser action."""
+    dir_path = os.path.abspath(
+        os.path.join(self.DataDir(), 'extensions', 'trigger_actions',
+                     'browser_action'))
+    ext_id = self.InstallExtension(dir_path)
+
+    self.NavigateToURL(self.GetFileURLForDataPath('simple.html'))
+
+    self.TriggerBrowserActionById(ext_id)
+
+    # Verify that the browser action turned the background red.
+    self.assertTrue(self.WaitUntil(
+        lambda: self.GetDOMValue('document.body.style.backgroundColor'),
+        expect_retval='red'),
+        msg='Browser action was not triggered.')
+
+  def testTriggerBrowserActionWithPopup(self):
+    """Test triggering browser action that shows a popup."""
+    dir_path = os.path.abspath(
+        os.path.join(self.DataDir(), 'extensions', 'trigger_actions',
+                     'browser_action_popup'))
+    ext_id = self.InstallExtension(dir_path)
+
+    self.TriggerBrowserActionById(ext_id)
+
+    # Verify that the extension popup is displayed.
+    popup = self.WaitUntilExtensionViewLoaded(
+        view_type='EXTENSION_POPUP')
+    self.assertTrue(popup,
+        msg='Browser action failed to display the popup (views=%s).' %
+        self.GetBrowserInfo()['extension_views'])
+
+  def testTriggerPageAction(self):
+    """Test triggering page action."""
+    dir_path = os.path.abspath(
+        os.path.join(self.DataDir(), 'extensions', 'trigger_actions',
+                     'page_action'))
+    ext_id = self.InstallExtension(dir_path)
+
+    # Page action icon is displayed when a tab is created.
+    self.NavigateToURL(self.GetFileURLForDataPath('simple.html'))
+    self.AppendTab(pyauto.GURL('chrome://newtab'))
+    self.ActivateTab(0)
+    self.assertTrue(self.WaitUntil(
+        lambda: ext_id in
+        self.GetBrowserInfo()['windows'][0]['visible_page_actions']),
+        msg='Page action icon is not visible.')
+
+    self.TriggerPageActionById(ext_id)
+
+    # Verify that page action turned the background red.
+    self.assertTrue(self.WaitUntil(
+        lambda: self.GetDOMValue('document.body.style.backgroundColor'),
+        expect_retval='red'),
+        msg='Page action was not triggered.')
+
+  def testTriggerPageActionWithPopup(self):
+    """Test triggering page action that shows a popup."""
+    dir_path = os.path.abspath(
+        os.path.join(self.DataDir(), 'extensions', 'trigger_actions',
+                     'page_action_popup'))
+    ext_id = self.InstallExtension(dir_path)
+
+    # Page action icon is displayed when a tab is created.
+    self.AppendTab(pyauto.GURL('chrome://newtab'))
+    self.ActivateTab(0)
+    self.assertTrue(self.WaitUntil(
+        lambda: ext_id in
+        self.GetBrowserInfo()['windows'][0]['visible_page_actions']),
+        msg='Page action icon is not visible.')
+
+    self.TriggerPageActionById(ext_id)
+
+    # Verify that the extension popup is displayed.
+    popup = self.WaitUntilExtensionViewLoaded(
+        view_type='EXTENSION_POPUP')
+    self.assertTrue(popup,
+        msg='Page action failed to display the popup (views=%s).' %
+        self.GetBrowserInfo()['extension_views'])
+
   def testAdblockExtensionCrash(self):
     """Test AdBlock extension successfully installed and enabled, and do not
     cause browser crash.
     """
     crx_file_path = os.path.abspath(
         os.path.join(self.DataDir(), 'extensions', 'adblock.crx'))
     ext_id = self.InstallExtension(crx_file_path)
 
     self.RestartBrowser(clear_profile=False)
     extension = self._GetExtensionInfoById(self.GetExtensionsInfo(), ext_id)
     self.assertTrue(extension['is_enabled'])
     self.assertFalse(extension['allowed_in_incognito'])
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()