Adding a new PyAuto hook for importing settings.

chrome/test/pyautolib/pyauto.py

 #!/usr/bin/python
 
 # Copyright (c) 2010 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 739 matching lines @@
     if ret_dict.has_key('error'):
       raise JSONInterfaceError(ret_dict['error'])
     try:
       f = open(filename)
       all_data = f.read()
       f.close()
       return all_data
     finally:
       shutil.rmtree(tempdir)
 
+  def ImportSettings(self, import_from, first_run, import_items):
+    """Import the specified import items from the specified browser.
+
+    Implements the features available in the "Import Settings" part of the
+    first-run UI dialog.
+
+    Args:
+      import_from: A string indicating which browser to import from. Possible
+                   strings (depending on which browsers are installed on the
+                   machine) are: 'Mozilla Firefox', 'Google Toolbar',
+                   'Microsoft Internet Explorer', 'Safari'
+      first_run: A boolean indicating whether this is the first run of
+                 the browser.
+                 If it is not the first run then:
+                 1) Bookmarks are only imported to the bookmarks bar if there
+                    aren't already bookmarks.
+                 2) The bookmark bar is shown.
+      import_items: A list of strings indicating which items to import.
+                    Strings that can be in the list are:
+                    HISTORY, FAVORITES, PASSWORDS, SEARCH_ENGINES, HOME_PAGE,
+                    ALL (note: COOKIES is not supported by the browser yet)
+    Raises:
+      pyauto_errors.JSONInterfaceError if the automation call returns an error.
+    """
+    cmd_dict = {  # Prepare command for the json interface
+      'command': 'ImportSettings',
+      'import_from': import_from,
+      'first_run': first_run,
+      'import_items': import_items
+    }
+    ret_dict = json.loads(self._SendJSONRequest(0, json.dumps(cmd_dict)))
+    if ret_dict.has_key('error'):
+      raise JSONInterfaceError(ret_dict['error'])
+    return ret_dict
+
   def ClearBrowsingData(self, to_remove, time_period):
     """Clear the specified browsing data. Implements the features available in
        the "ClearBrowsingData" UI.
 
     Args:
       to_remove: a list of strings indicating which types of browsing data
                  should be removed. Strings that can be in the list are:
                  HISTORY, DOWNLOADS, COOKIES, PASSWORDS, FORM_DATA, CACHE
       time_period: a string indicating the time period for the removal.
                    Possible strings are:
@@ skipping 310 matching lines @@
     if self._options.verbose:
       verbosity = 2
     result = unittest.TextTestRunner(verbosity=verbosity).run(pyauto_suite)
     del loaded_tests  # Need to destroy test cases before the suite
     del pyauto_suite
     sys.exit(not result.wasSuccessful())
 
 
 if __name__ == '__main__':
   Main()