1# Copyright 2020 The Chromium OS Authors. All rights reserved. 2# Use of this source code is governed by a BSD-style license that can be 3# found in the LICENSE file. 4 5import logging 6 7from autotest_lib.client.common_lib import error 8from autotest_lib.client.cros.power import power_status 9from autotest_lib.client.cros.power import power_test 10from autotest_lib.client.cros.power import power_utils 11 12 13class power_WaitForCoolDown(power_test.power_Test): 14 """class for power_WaitForCoolDown test.""" 15 version = 1 16 17 def initialize(self, pdash_note='', seconds_period=20., 18 force_discharge=False): 19 power_utils.set_display_power(power_utils.DISPLAY_POWER_ALL_OFF) 20 super(power_WaitForCoolDown, self).initialize( 21 seconds_period=seconds_period, 22 pdash_note=pdash_note, 23 force_discharge=force_discharge) 24 for log in self._meas_logs: 25 if type(log) == power_status.TempLogger: 26 self._tlog = log 27 break 28 else: 29 raise error.TestNAError('DUT does not have temp sensor.') 30 31 def run_once(self, target_temp=48, max_runtime=600): 32 """"Look at temperature every |seconds_period| seconds for at most 33 |max_runtime| seconds until reported temps do not exceed |target_temp|. 34 35 @param target_temp: Target temperature in celsius. 36 @param max_runtime: Maximum runtime in seconds. 37 """ 38 loop_secs = max(1, int(self._seconds_period)) 39 num_loop = int(max_runtime / loop_secs) 40 self.start_measurements() 41 42 for i in range(num_loop): 43 max_temp = max(self._tlog.refresh()) 44 if max_temp <= target_temp: 45 logging.info('Cooldown at %d seconds, temp: %.1fC', 46 i * self._seconds_period, max_temp) 47 return 48 self.loop_sleep(i, loop_secs) 49 50 max_temp = max(self._tlog.refresh()) 51 logging.warn( 52 'Fail to cool down after %d seconds, temp: %.1fC, target: %dC', 53 num_loop * loop_secs, max_temp, target_temp) 54 55 def cleanup(self): 56 power_utils.set_display_power(power_utils.DISPLAY_POWER_ALL_ON) 57 super(power_WaitForCoolDown, self).cleanup() 58