Add PyAuto tests for triggering browser/page action.

chrome/test/pyautolib/pyauto.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.
 
 """PyAuto: Python Interface to Chromium's Automation Proxy.
 
 PyAuto uses swig to expose Automation Proxy interfaces to Python.
 For complete documentation on the functionality available,
@@ skipping 1450 matching lines @@
           u'command_line_string': "COMMAND_LINE_STRING --WITH-FLAGS",
           u'branding': 'Chromium',
           u'is_official': False,}
         # The order of the windows and tabs listed here will be the same as
         # what shows up on screen.
         u'windows': [ { u'index': 0,
                         u'height': 1134,
                         u'incognito': False,
                         u'profile_path': u'Default',
                         u'fullscreen': False,
+                        u'visible_page_actions':
+                          [u'dgcoklnmbeljaehamekjpeidmbicddfj',
+                           u'osfcklnfasdofpcldmalwpicslasdfgd']
                         u'selected_tab': 0,
                         u'tabs': [ {
                           u'index': 0,
                           u'infobars': [],
                           u'pinned': True,
                           u'renderer_pid': 93747,
                           u'url': u'http://www.google.com/' }, {
                           u'index': 1,
                           u'infobars': [],
                           u'pinned': False,
@@ skipping 302 matching lines @@
       allow_in_incognito: A boolean, allow extension in incognito.
     """
     cmd_dict = {  # Prepare command for the json interface
       'command': 'SetExtensionStateById',
       'id': id,
       'enable': enable,
       'allow_in_incognito': allow_in_incognito,
     }
     self._GetResultFromJSONRequest(cmd_dict)
 
+  def TriggerPageActionById(self, id, windex=0):
+    """Trigger page action asynchronously in the active tab.
+
+    The page action icon must be displayed before invoking this function.
+
+    Args:
+      id: The string id of the extension.
+      windex: Integer index of the browser window to use; defaults to 0
+              (first window).
+    """
+    cmd_dict = {  # Prepare command for the json interface
+      'command': 'TriggerPageActionById',
+      'id': id,
+      'windex': windex,
+    }
+    self._GetResultFromJSONRequest(cmd_dict, windex=windex)
+
+  def TriggerBrowserActionById(self, id, windex=0):
+    """Trigger browser action asynchronously in the active tab.
+
+    Args:
+      id: The string id of the extension.
+      windex: Integer index of the browser window to use; defaults to 0
+              (first window).
+    """
+    cmd_dict = {  # Prepare command for the json interface
+      'command': 'TriggerBrowserActionById',
+      'id': id,
+      'windex': windex,
+    }
+    self._GetResultFromJSONRequest(cmd_dict, windex=windex)
+
   def UpdateExtensionsNow(self):
     """Auto-updates installed extensions.
 
     Waits until all extensions are updated, loaded, and ready for use.
     This is equivalent to clicking the "Update extensions now" button on the
     chrome://extensions page.
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation returns an error.
     """
@@ skipping 10 matching lines @@
     If there are more than one matching extension view, return one at random.
     Uses WaitUntil so timeout is capped by automation timeout.
     Refer to extension_view dictionary returned in GetBrowserInfo()
     for sample input/output values.
 
     Args:
       name: (optional) Name of the extension.
       extension_id: (optional) ID of the extension.
       url: (optional) URL of the extension view.
       view_type: (optional) Type of the extension view.
+        [EXTENSION_BACKGROUND_PAGE|EXTENSION_POPUP|EXTENSION_INFOBAR|
+         EXTENSION_DIALOG]

[dennis_jeffrey, 2011/11/23 21:54:54]

Should these be strings?

 
     Returns:
       The 'view' property of the extension view.
       None, if no view loaded.
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation returns an error.
     """
     def _GetExtensionViewLoaded():
       extension_views = self.GetBrowserInfo()['extension_views']
@@ skipping 2949 matching lines @@
     successful = result.wasSuccessful()
     if not successful:
       pyauto_tests_file = os.path.join(self.TestsDir(), self._tests_filename)
       print >>sys.stderr, 'Tests can be disabled by editing %s. ' \
                           'Ref: %s' % (pyauto_tests_file, _PYAUTO_DOC_URL)
     sys.exit(not successful)
 
 
 if __name__ == '__main__':
   Main()