Add a test for the "ephemeral_users_enabled" device policy

chrome/test/pyautolib/chromeos/suid_actions.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.
 
 """Helper script to perform actions as a super-user on ChromeOS.
 
 Needs to be run with superuser privileges, typically using the
 suid_python binary.
 
 Usage:
   sudo python suid_actions.py --action=CleanFlimflamDirs
 """
 
+import logging
 import optparse
 import os
 import shutil
 import sys
 
-
 sys.path.append('/usr/local')  # to import autotest libs.
 from autotest.cros import cryptohome
 
+# TODO(bartfab): Remove when crosbug.com/20709 is fixed.
+sys.path.append(os.path.join(os.path.dirname(__file__), os.pardir))
+from pyauto import AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE
+
 
 class SuidAction(object):
   """Helper to perform some super-user actions on ChromeOS."""
 
   def _ParseArgs(self):
     parser = optparse.OptionParser()
     parser.add_option(
         '-a', '--action', help='Action to perform.')
     self._options = parser.parse_args()[0]
     if not self._options.action:
@@ skipping 24 matching lines @@
             shutil.rmtree(path)
           else:
             os.remove(path)
     finally:
       os.system('start flimflam')
 
   def RemoveAllCryptohomeVaults(self):
     """Remove any existing cryptohome vaults."""
     cryptohome.remove_all_vaults()
 
+  def TryToDisableLocalStateAutoClearing(self):
+    """Try to disable clearing of the local state on session manager startup.
+
+    This will fail if rootfs verification is on.
+    TODO(bartfab): Remove this method when crosbug.com/20709 is fixed.
+    """
+    os.system('mount -o remount,rw /')
+    os.remove(AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE)
+    os.system('mount -o remount,ro /')
+    if os.path.exists(AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE):
+      logging.debug('Failed to remove %s. Session manager will clear local '
+                    'state on startup.' % AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE)
+    else:
+      logging.debug('Removed %s. Session manager will not clear local state on '
+                    'startup.' % AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE)
+
+  def TryToEnableLocalStateAutoClearing(self):
+    """Try to enable clearing of the local state on session manager startup.
+
+    This will fail if rootfs verification is on.
+    TODO(bartfab): Remove this method when crosbug.com/20709 is fixed.
+    """
+    os.system('mount -o remount,rw /')
+    open(AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE, 'w').close()
+    os.system('mount -o remount,ro /')
+    if os.path.exists(AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE):
+      logging.debug('Created %s. Session manager will clear local state on '
+                    'startup.' % AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE)
+    else:
+      logging.debug('Failed to create %s. Session manager will not clear local '
+                    'state on startup.' % AUTO_CLEAR_LOCAL_STATE_MAGIC_FILE)
+
 
 if __name__ == '__main__':
   sys.exit(SuidAction().Run())