• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Lint as: python2, python3
2# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
6from dbus.mainloop.glib import DBusGMainLoop
7# AU tests use ToT client code, but ToT -3 client version.
8try:
9    from gi.repository import GObject
10except ImportError:
11    import gobject as GObject
12
13from autotest_lib.client.bin import test
14from autotest_lib.client.common_lib import error
15from autotest_lib.client.common_lib.cros import policy, session_manager
16from autotest_lib.client.cros import cros_ui, cryptohome, ownership
17
18
19class login_MultipleSessions(test.test):
20    """Ensure that the session_manager can handle multiple calls to StartSession
21       correctly.
22    """
23    version = 1
24
25
26    def initialize(self):
27        super(login_MultipleSessions, self).initialize()
28        policy.install_protobufs(self.autodir, self.job)
29        # Ensure a clean beginning.
30        ownership.restart_ui_to_clear_ownership_files()
31
32        self._bus_loop = DBusGMainLoop(set_as_default=True)
33        self._session_manager = session_manager.connect(self._bus_loop)
34        self._listener = session_manager.OwnershipSignalListener(
35                GObject.MainLoop())
36        self._listener.listen_for_new_key_and_policy()
37
38
39    def run_once(self):
40        expected_owner = 'first_user@nowhere.com'
41        other_user = 'second_user@nowhere.com'
42        self.__start_session_for(expected_owner)
43        self.__start_session_for(other_user)
44        self._listener.wait_for_signals(desc='Initial policy push complete.')
45
46        # Ensure that the first user got to be the owner.
47        retrieved_policy = policy.get_policy(self._session_manager)
48        if retrieved_policy is None: raise error.TestFail('Policy not found')
49        policy.compare_policy_response(retrieved_policy, owner=expected_owner)
50        # bounce the session manager and wait for it to come back up before
51        # reconnecting.
52        cros_ui.restart()
53        self._session_manager = session_manager.connect(self._bus_loop)
54
55        # Destroy the owner's cryptohome and start sessions again in a
56        # different order
57        self.__start_session_for(other_user)
58        self.__start_session_for(expected_owner)
59
60        self._listener.wait_for_signals(desc='Re-taking of ownership complete.')
61
62        # Ensure that the first user still gets to be the owner.
63        retrieved_policy = policy.get_policy(self._session_manager)
64        if retrieved_policy is None: raise error.TestFail('Policy not found')
65        policy.compare_policy_response(retrieved_policy, owner=expected_owner)
66
67
68    def __start_session_for(self, user):
69        """Call StartSession() for user, ensure the user has clean on-device state
70
71        Make a fresh cryptohome for user, and then start a session for him
72        with the session manager.
73
74        @param user: the user to start a session for.
75
76        @raises error.TestFail: if the session cannot be started.
77        """
78        cryptohome.ensure_clean_cryptohome_for(user)
79        self._session_manager.StartSession(user, '')
80
81
82    def cleanup(self):
83        # Bounce UI, without waiting for the browser to come back. Best effort.
84        cros_ui.stop(allow_fail=True)
85        cros_ui.start(allow_fail=True, wait_for_login_prompt=False)
86