1# Lint as: python2, python3
2# Copyright 2014 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
6# Most of this code is based on login_GuestAndActualSession, which performs
7# similar ownership clearing/checking tasks.
8
9from dbus.mainloop.glib import DBusGMainLoop
10# AU tests use ToT client code, but ToT -3 client version.
11try:
12    from gi.repository import GObject
13except ImportError:
14    import gobject as GObject
15
16from autotest_lib.client.bin import test
17from autotest_lib.client.common_lib import error
18from autotest_lib.client.common_lib.cros import session_manager
19from autotest_lib.client.cros import cros_ui, cryptohome, ownership
20
21class login_CryptohomeOwnerQuery(test.test):
22    """Verify that the cryptohome owner user query works properly."""
23    version = 1
24
25
26    def initialize(self):
27        super(login_CryptohomeOwnerQuery, self).initialize()
28        # Ensure a clean beginning.
29        ownership.restart_ui_to_clear_ownership_files()
30
31        bus_loop = DBusGMainLoop(set_as_default=True)
32        self._session_manager = session_manager.connect(bus_loop)
33        self._listener = session_manager.OwnershipSignalListener(
34                GObject.MainLoop())
35        self._listener.listen_for_new_key_and_policy()
36
37
38    def run_once(self):
39        owner = '[email protected]'
40
41        if cryptohome.get_login_status()['owner_user_exists']:
42            raise error.TestFail('Owner existed before login')
43
44        cryptohome.ensure_clean_cryptohome_for(owner)
45        self._session_manager.StartSession(owner, '')
46
47        self._listener.wait_for_signals(desc='Device ownership complete.')
48
49        if not cryptohome.get_login_status()['owner_user_exists']:
50            raise error.TestFail('Owner does not exist after login')
51
52
53    def cleanup(self):
54        self._session_manager.StopSession('')
55        cros_ui.start(allow_fail=True, wait_for_login_prompt=False)
56