Add a test for the "ephemeral_users_enabled" device policy

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 78 matching lines @@
 from pyauto_errors import NTPThumbnailNotShownError
 import pyauto_utils
 import simplejson as json  # found in third_party
 
 _CHROME_DRIVER_FACTORY = None
 _HTTP_SERVER = None
 _REMOTE_PROXY = None
 _OPTIONS = None
 _BROWSER_PID = None
 
+# TODO(bartfab): Remove when crosbug.com/20709 is fixed.
+AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE = '/root/.forget_usernames'
+
 
 class PyUITest(pyautolib.PyUITestBase, unittest.TestCase):
   """Base class for UI Test Cases in Python.
 
   A browser is created before executing each test, and is destroyed after
   each test irrespective of whether the test passed or failed.
 
   You should derive from this class and create methods with 'test' prefix,
   and use methods inherited from PyUITestBase (the C++ side).
 
@@ skipping 332 matching lines @@
   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 TryToDisableLocalStateAutoClearingOnChromeOS():
+    """Disable clearing of the local state on session manager startup.
+
+    TODO(bartfab): Remove this method when crosbug.com/20709 is fixed.
+    """
+    PyUITest.RunSuperuserActionOnChromeOS('TryToDisableLocalStateAutoClearing')
+
+  @staticmethod
+  def TryToEnableLocalStateAutoClearingOnChromeOS():
+    """Enable clearing of the local state on session manager startup.
+
+    TODO(bartfab): Remove this method when crosbug.com/20709 is fixed.
+    """
+    PyUITest.RunSuperuserActionOnChromeOS('TryToEnableLocalStateAutoClearing')
+
+  @staticmethod
+  def IsLocalStateAutoClearingEnabledOnChromeOS():
+    """Check if the session manager is set to clear the local state on startup.
+
+    TODO(bartfab): Remove this method when crosbug.com/20709 is fixed.
+    """
+    return os.path.exists(AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE)
+
+  @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 824 matching lines @@
     """Return info about preferences.
 
     This represents a snapshot of the local state preferences. If you expect
     local state preferences to have changed, you need to call this method again
     to get a fresh snapshot.
 
     Returns:
       an instance of prefs_info.PrefsInfo
     """
     return prefs_info.PrefsInfo(
-        self._SendJSONRequest(0,
+        self._SendJSONRequest(-1,
                               json.dumps({'command': 'GetLocalStatePrefsInfo'}),
                               self.action_max_timeout_ms()))
 
   def SetLocalStatePrefs(self, path, value):
     """Set local state preference for the given path.
 
     Preferences are stored by Chromium as a hierarchical dictionary.
     dot-separated paths can be used to refer to a particular preference.
     example: "session.restore_on_startup"
 
@@ skipping 3982 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()