PyAuto: Automation hooks to get/set theme

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 633 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 SetTheme(self, crx_file_path):
+    """Installs the given theme synchronously.
+
+    A theme file is file with .crx suffix, like an extension.
+    This method call waits until theme is installed and will trigger the
+    "theme installed" infobar.
+
+    Uses InstallExtension().
+
+    Returns:
+      True, on success.
+    """
+    return self.InstallExtension(crx_file_path, True)
+
+  def ClearTheme(self):
+    """Clear the theme.  Resets to default.
+
+    Has no effect when the theme is already the default one.
+    This is a blocking call.
+
+    Raises:
+      pyauto_errors.JSONInterfaceError if the automation call returns an error.
+    """
+    cmd_dict = {
+      'command': 'ClearTheme',
+    }
+    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 GetThemeInfo(self):
+    """Get info about theme.
+
+    This includes info about the theme name, its colors, images, etc.
+
+    Returns:
+      a dictionary containing info about the theme.
+      empty dictionary if no theme has been applied (default theme).
+    SAMPLE:
+    { u'colors': { u'frame': [71, 105, 91],
+                   u'ntp_link': [36, 70, 0],
+                   u'ntp_section': [207, 221, 192],
+                   u'ntp_text': [20, 40, 0],
+                   u'toolbar': [207, 221, 192]},
+      u'images': { u'theme_frame': u'images/theme_frame_camo.png',
+                   u'theme_ntp_background': u'images/theme_ntp_background.png',
+                   u'theme_toolbar': u'images/theme_toolbar_camo.png'},
+      u'name': u'camo theme',
+      u'tints': {u'buttons': [0.33000000000000002, 0.5, 0.46999999999999997]}}
+
+    Raises:
+      pyauto_errors.JSONInterfaceError if the automation call returns an error.
+    """
+    cmd_dict = {
+      'command': 'GetThemeInfo',
+    }
+    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
+
 
 class PyUITestSuite(pyautolib.PyUITestSuiteBase, unittest.TestSuite):
   """Base TestSuite for PyAuto UI tests."""
 
   def __init__(self, args):
     pyautolib.PyUITestSuiteBase.__init__(self, args)
 
     # Figure out path to chromium binaries
     browser_dir = os.path.normpath(os.path.dirname(pyautolib.__file__))
     logging.debug('Loading pyauto libs from %s', browser_dir)
@@ skipping 223 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()