Allow setting of user and device policies in functional tests

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 424 matching lines @@
     for item in os.listdir(chronos_dir):
       if item != 'user' and not item.startswith('.'):
         pyauto_utils.RemovePath(os.path.join(chronos_dir, item))
 
   @staticmethod
   def CleanupFlimflamDirsOnChromeOS():
     """Clean the contents of flimflam profiles and restart flimflam."""
     PyUITest.RunSuperuserActionOnChromeOS('CleanFlimflamDirs')
 
   @staticmethod
+  def RemoveAllCryptohomeVaultsOnChromeOS():
+    """Remove any existing cryptohome vaults."""
+    PyUITest.RunSuperuserActionOnChromeOS('RemoveAllCryptohomeVaults')
+
+  @staticmethod
   def _IsInodeNew(path, old_inode):
     """Determine whether an inode has changed. POSIX only.
 
     Args:
       path: The file path to check for changes.
       old_inode: The old inode number.
 
     Returns:
       True if the path exists and its inode number is different from old_inode.
       False otherwise.
@@ skipping 3212 matching lines @@
       Profiles will be listed in the same order as visible in preferences.
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation call returns an error.
     """
     cmd_dict = {  # Prepare command for the json interface
       'command': 'GetMultiProfileInfo'
     }
     return self._GetResultFromJSONRequest(cmd_dict, windex=None)
 
-  def SetPolicies(self, managed_platform=None, recommended_platform=None,
-                  managed_cloud=None, recommended_cloud=None):
-    """Sets the policies on the browser. Always fails on official builds.
-
-    Args:
-      managed_platform: a dictionary with the policy values for the managed
-                        platform provider.
-      recommended_platform: a dictionary with the policy values for the
-                            recommended platform provider.
-      managed_cloud: a dictionary with the policy values for the managed
-                     cloud provider.
-      recommended_cloud: a dictionary with the policy values for the recommended
-                         cloud provider.
-
-    Leaving an argument to None will restore the default behavior for that
-    provider.
-    """
-    assert not self.GetBrowserInfo()['properties']['is_official']
-    cmd_dict = {
-        'command': 'SetPolicies',
-        'managed_cloud': managed_cloud,
-        'managed_platform': managed_platform,
-        'recommended_cloud': recommended_cloud,
-        'recommended_platform': recommended_platform
-    }
-    return self._GetResultFromJSONRequest(cmd_dict)
-
   def GetPolicyDefinitionList(self):
     """Gets a dictionary of existing policies mapped to their definitions.
 
     SAMPLE OUTPUT:
     {
       'ShowHomeButton': ['bool', false],
       'DefaultSearchProviderSearchURL': ['str', false],
       ...
     }
 
@@ skipping 1539 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()