You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

41 lines
1.7 KiB

# Copyright 2019 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.
"""Power utils for server tests."""
from autotest_lib.server import autotest
from autotest_lib.server.cros.power import servo_charger
def put_host_battery_in_range(host, min_level, max_level, timeout):
"""
Charges or drains the host's battery to the specified range within the
timeout. This uses a servo v4 and either the power_BatteryCharge or the
power_BatteryDrain client test.
@param host: DUT to use
@param min_level: battery percentage
@param max_level: battery percentage
@param timeout: in seconds
@throws: A TestFail error if getting the current battery level, setting the
servo's charge state, or running either of the client tests fails.
"""
current_level = host.get_battery_display_percentage()
if current_level >= min_level and current_level <= max_level:
return
autotest_client = autotest.Autotest(host)
charge_manager = servo_charger.ServoV4ChargeManager(host, host.servo)
if current_level < min_level:
charge_manager.start_charging()
autotest_client.run_test('power_BatteryCharge',
max_run_time=timeout,
percent_target_charge=min_level,
use_design_charge_capacity=False)
if current_level > max_level:
charge_manager.stop_charging()
autotest_client.run_test('power_BatteryDrain',
drain_to_percent=max_level,
drain_timeout=timeout)