# Copyright (c) 2013 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 json, re, time

from autotest_lib.client.common_lib import error
from autotest_lib.server import hosts
from autotest_lib.server import test


_TIME_TO_SUSPEND = 10
_EXTRA_DELAY = 10


class power_BatteryStateOnResume(test.test):
    """
    Test to verify the battery charge state of the DUT on resume after the AC
    charger gets unplugged and plugged in its suspend state.

    If DUT is connected to RPM(default) - No need to pass any command line args.
    If DUT is connected to USB powerstrip(via servo), Need to pass cmdlineargs
    as --args=power_control="servoj10".
    If DUT is not connected to servo and RPM. i.e to handle manually, Need to
    pass cmdlineargs as --args=power_control="manual".
    """
    version = 1

    def initialize(self, host, client_ip, cmdline_args):
        self._client = hosts.create_host(client_ip)
        self.ensure_battery_present()

        args = {}
        for arg in cmdline_args:
            match = re.search("^(\w+)=(.+)", arg)
            if match:
                args[match.group(1)] = match.group(2)
        self.power_control = args.get('power_control', host.POWER_CONTROL_RPM)
        if self.power_control not in host.POWER_CONTROL_VALID_ARGS:
            raise error.TestError('Valid values for --args=power_control '
                                  'are %s. But you entered wrong argument '
                                  'as "%s".'
                                   % (host.POWER_CONTROL_VALID_ARGS,
                                   self.power_control))


    def run_once(self, host):
        if not self.loggedin(host):
            raise error.TestError("Not logged in!")
        self.unplug_ac_on_suspend(host)
        self.test_discharge_state(host)
        self.plug_ac_on_suspend(host)
        self.test_charge_state(host)


    def unplug_ac_on_suspend(self, host):
        """Unplugs AC when device in Suspend state."""
        host.servo.lid_close()
        time.sleep(_TIME_TO_SUSPEND + _EXTRA_DELAY)
        host.power_off(self.power_control)
        host.servo.lid_open()
        time.sleep(_EXTRA_DELAY)


    def plug_ac_on_suspend(self, host):
        """Plugs AC when device in Suspend state."""
        host.servo.lid_close()
        time.sleep(_TIME_TO_SUSPEND + _EXTRA_DELAY)
        host.power_on(self.power_control)
        host.servo.lid_open()
        time.sleep(_EXTRA_DELAY)


    def ensure_battery_present(self):
        """Ensure we have battery exists in DUT."""
        result = self._client.run('power_supply_info | egrep present')
        if 'yes' not in result.stdout:
            raise error.TestError('Find no batteries')


    def loggedin(self, host):
        """
        Checks if the host has a logged in user.

        @return True if a user is logged in on the device.
        """
        try:
            cmd_out = host.run('cryptohome --action=status').stdout.strip()
        except:
            return False
        status = json.loads(cmd_out)
        return any((mount['mounted'] for mount in status['mounts']))


    def test_charge_state(self, host):
        """Tests whether battery is in 'Charging/Charged' state."""
        bat_state = self.get_bat_status(host)
        if not (bat_state == 'Charging' or bat_state == 'Fully charged'):
            raise error.TestFail('Not Charging.  BAT state is %s.' % bat_state)


    def test_discharge_state(self, host):
        """Tests whether battery is in 'Discharging' state."""
        bat_state = self.get_bat_status(host)
        if not bat_state == 'Discharging':
            raise error.TestFail(
                'Not Discharging.  BAT state is %s.' % bat_state)


    def get_bat_status(self, host):
        """Returns the battery state per the 'power_supply_info' tool.

        @return battery power 'state' value. (i.e, Charging/Discharging ..)
        """
        bat_state_info = host.run(
            'power_supply_info | egrep state').stdout.strip()
        split_list = bat_state_info.split(":")
        return split_list[-1].strip()