[pyauto] Add ability to force install an extension with experimental perms

chrome/test/pyautolib/pyauto.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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 2028 matching lines @@
       'command': 'WaitUntilTranslateComplete',
       'tab_index': tab_index
     }
     # TODO(phajdan.jr): We need a solid automation infrastructure to handle
     # these cases. See crbug.com/53647.
     return self.WaitUntil(
         lambda tab_index, window_index: self.GetTranslateInfo(
             tab_index=tab_index, window_index=window_index)['page_translated'],
         args=[tab_index, window_index])
 
-  def InstallExtension(self, extension_path, with_ui=False, windex=0):
+  def InstallExtension(self, extension_path, with_ui=False, from_webstore=None,
+                       windex=0):
     """Installs an extension from the given path.
 
     The path must be absolute and may be a crx file or an unpacked extension
     directory. Returns the extension ID if successfully installed and loaded.
     Otherwise, throws an exception. The extension must not already be installed.
 
     Args:
       extension_path: The absolute path to the extension to install. If the
                       extension is packed, it must have a .crx extension.
       with_ui: Whether the extension install confirmation UI should be shown.
+      from_webstore: If True, forces a .crx extension to be recognized as one
+          from the webstore. Can be used to force install an extension with
+          'experimental' permissions.
       windex: Integer index of the browser window to use; defaults to 0
               (first window).
 
     Returns:
       The ID of the installed extension.
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation call returns an error.
     """
     cmd_dict = {
         'command': 'InstallExtension',
         'path': extension_path,
         'with_ui': with_ui,
         'windex': windex,
     }
+    if from_webstore:
+      cmd_dict['from_webstore'] = True
     return self._GetResultFromJSONRequest(cmd_dict, windex=None)['id']
 
   def GetExtensionsInfo(self, windex=0):
     """Returns information about all installed extensions.
 
     Args:
       windex: Integer index of the browser window to use; defaults to 0
               (first window).
 
     Returns:
@@ skipping 3875 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()