[autotest] Add test for the re-taking of ownership after the owner key is lost

client/cros/login.py

 # Copyright (c) 2011 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 errno, logging, os, re, shutil, signal, subprocess, time
 import common
 import constants, cros_logging, cros_ui, cryptohome
 from autotest_lib.client.bin import utils
 from autotest_lib.client.common_lib import error
 
 
 _DEFAULT_TIMEOUT = 30
-_DEFAULT_OWNERSHIP_TIMEOUT = 300  # Ownership is an inherently random process.
 
 # Log messages used to signal when we're in a logout situation. Used to detect
 # crashes by cros_ui_test.UITest.
 LOGOUT_ATTEMPT_MSG = 'cros/login.py: Attempting logout...'
 LOGOUT_COMPLETE_MSG = 'cros/login.py: Logout complete.'
 
 
 class TimeoutError(error.TestError):
     """Error raised when we time out while waiting on a condition."""
     pass
@@ skipping 285 matching lines @@
     wait_for_condition(
         lambda: os.access(
             constants.CHROME_WINDOW_MAPPED_MAGIC_FILE, os.F_OK),
         'Timed out waiting for initial Chrome window',
         timeout=timeout,
         process='chrome',
         log_reader=log_reader,
         crash_msg='Chrome crashed before first tab rendered.')
 
 
-def wait_for_ownership(timeout=_DEFAULT_OWNERSHIP_TIMEOUT):
+def wait_for_ownership(timeout=constants.DEFAULT_OWNERSHIP_TIMEOUT):
     log_reader = cros_logging.LogReader()
     log_reader.set_start_by_current()
     wait_for_condition(
         condition=lambda: os.access(constants.OWNER_KEY_FILE, os.F_OK),
         timeout_msg='Timed out waiting for ownership',
         timeout=timeout,
         process=constants.BROWSER,
         log_reader=log_reader,
         crash_msg='Chrome crashed before ownership could be taken.')
 
@@ skipping 60 matching lines @@
     except (IOError, OSError) as err:
         logging.error(err)
 
     # Restart the UI.
     nuke_login_manager()
     utils.poll_for_condition(
         lambda: __session_manager_restarted(oldpid),
         TimeoutError('Timed out waiting for logout'),
         timeout)
     wait_for_login_prompt()