# Copyright (c) 2014 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 time
from autotest_lib.server import test

LINE_STATUS_WAIT_TIME = 5


class power_RPMTest(test.test):
    """Test RPM functionality."""
    version = 1


    def initialize(self, host, verify=True):
        """
        @param host: The host to run the test on
        @param verify: True to test both on and off for the AC power and to
            check with the host whether it sees the same state
        """
        self._host = host
        self._host_at = autotest.Autotest(host)
        self._verify = verify


    def _set_power(self, power_on):
        if power_on:
            host.power_on()
        else:
            host.power_off()

        if self._verify:
            time.sleep(LINE_STATUS_WAIT_TIME)
            self._host_at.run_test('power_CheckAC', power_on=power_on)


    def run_once(self, power_on=True):
        """Run the test.

        @param power_on: True to turn the power on and False for off
        """
        if self._verify:
            self._set_power(not power_on)

        self._set_power(power_on)