WebDriver extension support in TestingAutomationProvider.

chrome/test/pyautolib/pyauto.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.
 
 """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,
 run pydoc on this file.
@@ skipping 1804 matching lines @@
       id: The string id of the extension.
 
     Returns:
       True, if the extension was successfully uninstalled, or
       False, otherwise.
     """
     cmd_dict = {  # Prepare command for the json interface
       'command': 'UninstallExtensionById',
       'id': id,
     }
-    return self._GetResultFromJSONRequest(cmd_dict)['success']
+    return self._GetResultFromJSONRequest(cmd_dict, windex=-1)['success']
 
   def SetExtensionStateById(self, id, enable, allow_in_incognito):
     """Set extension state: enable/disable, allow/disallow in incognito mode.
 
     Args:
       id: The string id of the extension.
       enable: A boolean, enable extension.
       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)
+    self._GetResultFromJSONRequest(cmd_dict, windex=-1)
 
-  def TriggerPageActionById(self, id, windex=0):
+  def TriggerPageActionById(self, id, tab_index=0, 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.
+      tab_index: Integer index of the tab to use; defaults to 0 (first tab).
       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,
+      'tab_index': tab_index,
     }
-    self._GetResultFromJSONRequest(cmd_dict, windex=windex)
+    self._GetResultFromJSONRequest(cmd_dict, windex=-1)
 
-  def TriggerBrowserActionById(self, id, windex=0):
+  def TriggerBrowserActionById(self, id, tab_index=0, windex=0):
     """Trigger browser action asynchronously in the active tab.
 
     Args:
       id: The string id of the extension.
+      tab_index: Integer index of the tab to use; defaults to 0 (first tab).
       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,
+      'tab_index': tab_index,
     }
-    self._GetResultFromJSONRequest(cmd_dict, windex=windex)
+    self._GetResultFromJSONRequest(cmd_dict, windex=-1)
 
   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 2955 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()