server/site_tests/platform_ServoPyAuto/platform_ServoPyAuto.py

# Copyright (c) 2012 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

from autotest_lib.client.common_lib import error
from autotest_lib.server import test
from autotest_lib.server.cros import pyauto_proxy


class platform_ServoPyAuto(test.test):
    """
    A simple test demonstrating the synchronous use of Servo and PyAuto.

    The client is logged in using PyAuto, the device is put to sleep by closing
    the lid with Servo, then the device is woken up by opening the lid and the
    test asserts that the user is still logged in.
    """
    version = 1

    def initialize(self, host):
        self._pyauto = pyauto_proxy.create_pyauto_proxy(host)


    def cleanup(self, host):
        self._pyauto.cleanup()


    def run_once(self, host=None):
        self._pyauto.LoginToDefaultAccount()

        # Close and open lid.
        boot_id = host.get_boot_id()
        host.servo.lid_close()
        host.test_wait_for_sleep()
        host.servo.lid_open()
        host.test_wait_for_resume(boot_id)

        info = self._pyauto.GetLoginInfo()
        logging.info(info)
        if not info['is_logged_in']:
            raise error.TestFail(
                'User is no longer logged in after sleep/resume cycle.')