autotest: Test RestartEntd dbus call

client/site_tests/login_DBusCalls/login_DBusCalls.py

Index: client/site_tests/login_DBusCalls/login_DBusCalls.py
diff --git a/client/site_tests/login_DBusCalls/login_DBusCalls.py b/client/site_tests/login_DBusCalls/login_DBusCalls.py
new file mode 100644
index 0000000000000000000000000000000000000000..c3d11cc40f5eefa71751a65051a29d3fbc693441
--- /dev/null
+++ b/client/site_tests/login_DBusCalls/login_DBusCalls.py
@@ -0,0 +1,66 @@
+# Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import logging, os
+from autotest_lib.client.bin import chromeos_constants, site_log_reader
+from autotest_lib.client.bin import site_ui_test, site_utils
+from autotest_lib.client.common_lib import error, utils
+
+_SESSION_MANAGER_DEST='org.chromium.SessionManager'
+_SESSION_MANAGER_OBJECT='org.chromium.SessionManagerInterface'
+_SESSION_MANAGER_PATH='/org/chromium/SessionManager'
+
+class login_DBusCalls(site_ui_test.UITest):
+    version = 1
+
+    def _call_session_manager_method(self, method, user='chronos'):
+        """Call session manager dbus method as given user.
+
+        We assume the method exists and succeeds.
+
+        Args:
+            method: name of method
+            user: system user to run as
+        Returns:
+            None
+        """
+        utils.system('su %s -c \'dbus-send --system --type=method_call '
+                     '--print-reply --dest=%s %s %s.%s\'' %
+                     (user, _SESSION_MANAGER_DEST, _SESSION_MANAGER_PATH,
+                      _SESSION_MANAGER_OBJECT, method))
+
+
+    def _test_restart_entd(self):
+        """Test the RestartEntd method."""
+        message_log = site_log_reader.LogReader()
+        message_log.set_start_by_current()
+        ui_log = site_log_reader.LogReader(chromeos_constants.UI_LOG)
+        ui_log.set_start_by_current()
+        # Make sure we can call RestartEntd from user chronos.
+        self._call_session_manager_method('RestartEntd')
+        try:
+            site_utils.poll_for_condition(
+                lambda: ui_log.can_find('Restart was successful'),
+                timeout=30,
+                exception=error.TestFail(
+                    'Found no ui log message about attempting to restart entd'))
+        finally:
+            logging.debug('UI log from RestartEntd: ' +
+                          ui_log.get_logs())
+
+        grep_bait = 'entdwife.sh: Username: ' + self.username
+        try:
+            site_utils.poll_for_condition(
+                lambda: message_log.can_find(grep_bait),
+                timeout=30,
+                exception=error.TestFail('Did not find %s in message log' %
+                                         grep_bait))
+        finally:
+            logging.debug('Message log from RestartEntd: ' +
+                          message_log.get_logs())
+
+
+    def run_once(self):
+        self.login()
+        self._test_restart_entd()