| # 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) |
