Add PyAuto tests for triggering browser/page action.

chrome/test/functional/extensions.py

 #!/usr/bin/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 164 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, False);
+    self.assertTrue(ext_id, msg='Failed to install extension: %s.' % dir_path)
+
+    self.NavigateToURL(self.GetFileURLForDataPath(
+        'frame_dom_access', 'frame_dom_access.html'))
+
+    self.TriggerExtensionActionById(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 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, False);
+    self.assertTrue(ext_id, msg='Failed to install extension: %s.' % dir_path)
+
+    # Page action icon is displayed when a tab is created.
+    self.AppendTab(pyauto.GURL(self.GetFileURLForDataPath(
+        'frame_dom_access', 'frame_dom_access.html')))
+
+    self.TriggerExtensionActionById(ext_id)
+
+    # Verify that page action turned the background red.
+    self.assertTrue(self.WaitUntil(
+        lambda: self.GetDOMValue('document.body.style.backgroundColor',
+                                 tab_index=1),
+        expect_retval='red'),
+        msg='Browser action was not triggered.')

[dennis_jeffrey, 2011/11/17 00:53:17]

'Browser' --> 'Page'

+
   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, False)
     self.assertTrue(ext_id, msg='Failed to install extension.')
 
     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()