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('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, False)
+    self.assertTrue(ext_id, msg='Failed to install extension: %s.' % 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, False)
+    self.assertTrue(ext_id, msg='Failed to install extension: %s.' % dir_path)
+
+    # Page action icon is displayed when a tab is created.

[kkania, 2011/11/21 18:53:58]

I think we need another function for GetVisiblePageActions that the user can
wait upon. The page action icon is not guaranteed to appear on tab creation or
by onload, since the hiding/showing is handled by the extension in whatever way
it wants.

[frankf, 2011/11/22 22:26:10]

I've added this to GetBrowserInfo, same thing should be done for browser actions
once we have a way to tie browser actions to extensions across platforms.

+    self.NavigateToURL(self.GetFileURLForDataPath('simple.html'))
+    self.AppendTab(pyauto.GURL('chrome://newtab'))
+    self.ActivateTab(0)
+
+    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, 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('chrome://newtab'))
+    self.ActivateTab(0)
+
+    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, 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()