Allow chromedriver to install an extension and get all installed extension IDs.

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 1698 matching lines @@
     """
     cmd_dict = {  # Prepare command for the json interface
       'command': 'SelectTranslateOption',
       'tab_index': tab_index,
       'option': 'set_target_language',
       'target_language': new_language
     }
     return self._GetResultFromJSONRequest(
         cmd_dict, windex=window_index)['translation_success']
 
+  def InstallExtension(self, extension_path, with_ui=False):
+    """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.
+
+    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
+    }
+    return self._GetResultFromJSONRequest(cmd_dict, windex=-1)['id']
+
   def GetExtensionsInfo(self):
     """Returns information about all installed extensions.
 
     Returns:
       A list of dictionaries representing each of the installed extensions.
       Example:
       [ { u'api_permissions': [u'bookmarks', u'experimental', u'tabs'],
           u'background_url': u'',
           u'description': u'Bookmark Manager',
           u'effective_host_permissions': [u'chrome://favicon/*',
                                           u'chrome://resources/*'],
           u'host_permissions': [u'chrome://favicon/*', u'chrome://resources/*'],
           u'id': u'eemcgdkfndhakfknompkggombfjjjeno',
+          u'is_component': True,
+          u'is_internal': False,
           u'name': u'Bookmark Manager',
           u'options_url': u'',
           u'public_key': u'MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDQcByy+eN9jza\
                            zWF/DPn7NW47sW7lgmpk6eKc0BQM18q8hvEM3zNm2n7HkJv/R6f\
                            U+X5mtqkDuKvq5skF6qqUF4oEyaleWDFhd1xFwV7JV+/DU7bZ00\
                            w2+6gzqsabkerFpoP33ZRIw7OviJenP0c0uWqDWF8EGSyMhB3tx\
                            qhOtiQIDAQAB',
           u'version': u'0.1' },
         { u'api_permissions': [...],
           u'background_url': u'chrome-extension://\
                                lkdedmbpkaiahjjibfdmpoefffnbdkli/\
                                background.html',
           u'description': u'Extension which lets you read your Facebook news \
                             feed and wall. You can also post status updates.',
           u'effective_host_permissions': [...],
           u'host_permissions': [...],
           u'id': u'lkdedmbpkaiahjjibfdmpoefffnbdkli',
           u'name': u'Facebook for Google Chrome',
           u'options_url': u'',
           u'public_key': u'...',
           u'version': u'2.0.9'
           u'is_enabled': True,
           u'allowed_in_incognito': True} ]
     """
     cmd_dict = {  # Prepare command for the json interface
       'command': 'GetExtensionsInfo'
     }
-    return self._GetResultFromJSONRequest(cmd_dict)['extensions']
+    return self._GetResultFromJSONRequest(cmd_dict, windex=-1)['extensions']
 
   def UninstallExtensionById(self, id):
     """Uninstall the extension with the given id.
 
     Args:
       id: The string id of the extension.
 
     Returns:
       True, if the extension was successfully uninstalled, or
       False, otherwise.
@@ skipping 633 matching lines @@
       'tab_index': tab_index,
     }
     self._GetResultFromJSONRequest(cmd_dict, windex=windex)
 
   def SetTheme(self, crx_file_path):
     """Installs the given theme synchronously.
 
     A theme file is a file with a .crx suffix, like an extension.  The theme
     file must be specified with an absolute path.  This method call waits until
     the theme is installed and will trigger the "theme installed" infobar.
+    If the install is unsuccessful, will throw an exception.
 
     Uses InstallExtension().
 
     Returns:
-      The ID of the installed theme, on success.  The empty string, otherwise.
+      The ID of the installed theme.
+
+    Raises:
+      pyauto_errors.JSONInterfaceError if the automation call returns an error.
     """
     return self.InstallExtension(crx_file_path, True)
 
   def WaitUntilDownloadedThemeSet(self, theme_name):
     """Waits until the theme has been set.
 
     This should not be called after SetTheme(). It only needs to be called after
     downloading a theme file (which will automatically set the theme).
 
     Uses WaitUntil so timeout is capped by automation timeout.
@@ skipping 668 matching lines @@
     """
     cmd_dict = {
       'command': 'GetNTPInfo',
     }
     return self._GetResultFromJSONRequest(cmd_dict)
 
   def _CheckNTPThumbnailShown(self, thumbnail):
     if self.GetNTPThumbnailIndex(thumbnail) == -1:
       raise NTPThumbnailNotShownError()
 
-  def InstallApp(self, app_crx_file_path):
-    """Installs the specified app synchronously.
-
-    An app file is a file with a .crx suffix, like an extension or theme.  The
-    app file must be specified with an absolute path.  This method will not
-    return until the app is installed.
-
-    Returns:
-      The ID of the installed app, on success.  The empty string, otherwise.
-    """
-    return self.InstallExtension(app_crx_file_path, False)
-
-  def UninstallApp(self, app_id):
-    """Uninstalls the specified app synchronously.
-
-    Args:
-      app_id: The string ID of the app to uninstall.  It can be retrieved
-              through the call to GetNTPApps above.
-
-    Returns:
-      True, if the app was successfully uninstalled, or
-      False, otherwise.
-    """
-    return self.UninstallExtensionById(app_id)
-
   def LaunchApp(self, app_id, windex=0):
     """Opens the New Tab Page and launches the specified app from it.
 
     This method will not return until after the contents of a new tab for the
     launched app have stopped loading.
 
     Args:
       app_id: The string ID of the app to launch.
       windex: The index of the browser window to work on.  Defaults to 0 (the
               first window).
@@ skipping 1641 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()