diff options
Diffstat (limited to 'cros_utils/device_setup_utils_unittest.py')
-rwxr-xr-x | cros_utils/device_setup_utils_unittest.py | 1321 |
1 files changed, 712 insertions, 609 deletions
diff --git a/cros_utils/device_setup_utils_unittest.py b/cros_utils/device_setup_utils_unittest.py index 12a70811..d7339e25 100755 --- a/cros_utils/device_setup_utils_unittest.py +++ b/cros_utils/device_setup_utils_unittest.py @@ -1,16 +1,14 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- # -# Copyright 2019 The Chromium OS Authors. All rights reserved. +# Copyright 2019 The ChromiumOS Authors # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Unittest for device_setup_utils.""" -from __future__ import print_function import time - import unittest from unittest import mock @@ -18,6 +16,7 @@ from cros_utils import command_executer from cros_utils import logger from cros_utils.device_setup_utils import DutWrapper + BIG_LITTLE_CPUINFO = """processor : 0 model name : ARMv8 Processor rev 4 (v8l) BogoMIPS : 48.00 @@ -94,609 +93,713 @@ Serial : 0000000000000000 class DutWrapperTest(unittest.TestCase): - """Class of DutWrapper test.""" - real_logger = logger.GetLogger() - - mock_cmd_exec = mock.Mock(spec=command_executer.CommandExecuter) - mock_logger = mock.Mock(spec=logger.Logger) - - def __init__(self, *args, **kwargs): - super(DutWrapperTest, self).__init__(*args, **kwargs) - - def setUp(self): - self.dw = DutWrapper( - '/tmp/chromeos', - 'lumpy.cros2', - log_level='verbose', - logger=self.mock_logger, - ce=self.mock_cmd_exec, - dut_config={}) - - @mock.patch.object(command_executer.CommandExecuter, 'CrosRunCommandWOutput') - def test_run_command_on_dut(self, mock_cros_runcmd): - self.mock_cmd_exec.CrosRunCommandWOutput = mock_cros_runcmd - mock_cros_runcmd.return_value = (0, '', '') - mock_cros_runcmd.assert_not_called() - self.dw.RunCommandOnDut('run command;') - mock_cros_runcmd.assert_called_once_with( - 'run command;', chromeos_root='/tmp/chromeos', machine='lumpy.cros2') - - @mock.patch.object(command_executer.CommandExecuter, 'CrosRunCommandWOutput') - def test_dut_wrapper_fatal_error(self, mock_cros_runcmd): - self.mock_cmd_exec.CrosRunCommandWOutput = mock_cros_runcmd - # Command returns error 1. - mock_cros_runcmd.return_value = (1, '', 'Error!') - mock_cros_runcmd.assert_not_called() - self.dw.RunCommandOnDut('run command;') - mock_cros_runcmd.assert_called_once_with( - 'run command;', chromeos_root='/tmp/chromeos', machine='lumpy.cros2') - # Error status causes log fatal. - self.assertEqual( - self.mock_logger.method_calls[-1], - mock.call.LogFatal('Command execution on DUT lumpy.cros2 failed.\n' - 'Failing command: run command;\nreturned 1\n' - 'Error message: Error!')) - - @mock.patch.object(command_executer.CommandExecuter, 'CrosRunCommandWOutput') - def test_dut_wrapper_ignore_error(self, mock_cros_runcmd): - self.mock_cmd_exec.CrosRunCommandWOutput = mock_cros_runcmd - # Command returns error 1. - mock_cros_runcmd.return_value = (1, '', 'Error!') - self.dw.RunCommandOnDut('run command;', ignore_status=True) - mock_cros_runcmd.assert_called_once_with( - 'run command;', chromeos_root='/tmp/chromeos', machine='lumpy.cros2') - # Error status is not fatal. LogError records the error message. - self.assertEqual( - self.mock_logger.method_calls[-1], - mock.call.LogError('Command execution on DUT lumpy.cros2 failed.\n' - 'Failing command: run command;\nreturned 1\n' - 'Error message: Error!\n' - '(Failure is considered non-fatal. Continue.)')) - - def test_disable_aslr(self): - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, '', '')) - self.dw.DisableASLR() - # pyformat: disable - set_cpu_cmd = ('set -e; ' - 'if [[ -e /proc/sys/kernel/randomize_va_space ]]; then ' - ' echo 0 > /proc/sys/kernel/randomize_va_space; ' - 'fi') - self.dw.RunCommandOnDut.assert_called_once_with(set_cpu_cmd) - - def test_set_cpu_governor(self): - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, '', '')) - self.dw.SetCpuGovernor('new_governor', ignore_status=False) - set_cpu_cmd = ( - 'for f in `ls -d /sys/devices/system/cpu/cpu*/cpufreq 2>/dev/null`; do ' - # Skip writing scaling_governor if cpu is offline. - ' [[ -e ${f/cpufreq/online} ]] && grep -q 0 ${f/cpufreq/online} ' - ' && continue; ' - ' cd $f; ' - ' if [[ -e scaling_governor ]]; then ' - ' echo %s > scaling_governor; fi; ' - 'done; ') - self.dw.RunCommandOnDut.assert_called_once_with( - set_cpu_cmd % 'new_governor', ignore_status=False) - - def test_set_cpu_governor_propagate_error(self): - self.dw.RunCommandOnDut = mock.Mock(return_value=(1, '', 'Error.')) - self.dw.SetCpuGovernor('non-exist_governor') - set_cpu_cmd = ( - 'for f in `ls -d /sys/devices/system/cpu/cpu*/cpufreq 2>/dev/null`; do ' - # Skip writing scaling_governor if cpu is not online. - ' [[ -e ${f/cpufreq/online} ]] && grep -q 0 ${f/cpufreq/online} ' - ' && continue; ' - ' cd $f; ' - ' if [[ -e scaling_governor ]]; then ' - ' echo %s > scaling_governor; fi; ' - 'done; ') - # By default error status is fatal. - self.dw.RunCommandOnDut.assert_called_once_with( - set_cpu_cmd % 'non-exist_governor', ignore_status=False) - - def test_set_cpu_governor_ignore_status(self): - self.dw.RunCommandOnDut = mock.Mock(return_value=(1, '', 'Error.')) - ret_code = self.dw.SetCpuGovernor('non-exist_governor', ignore_status=True) - set_cpu_cmd = ( - 'for f in `ls -d /sys/devices/system/cpu/cpu*/cpufreq 2>/dev/null`; do ' - # Skip writing scaling_governor if cpu is not online. - ' [[ -e ${f/cpufreq/online} ]] && grep -q 0 ${f/cpufreq/online} ' - ' && continue; ' - ' cd $f; ' - ' if [[ -e scaling_governor ]]; then ' - ' echo %s > scaling_governor; fi; ' - 'done; ') - self.dw.RunCommandOnDut.assert_called_once_with( - set_cpu_cmd % 'non-exist_governor', ignore_status=True) - self.assertEqual(ret_code, 1) - - def test_disable_turbo(self): - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, '', '')) - self.dw.DisableTurbo() - set_cpu_cmd = ( - # Disable Turbo in Intel pstate driver - 'if [[ -e /sys/devices/system/cpu/intel_pstate/no_turbo ]]; then ' - ' if grep -q 0 /sys/devices/system/cpu/intel_pstate/no_turbo; then ' - ' echo -n 1 > /sys/devices/system/cpu/intel_pstate/no_turbo; ' - ' fi; ' - 'fi; ') - self.dw.RunCommandOnDut.assert_called_once_with(set_cpu_cmd) - - def test_get_cpu_online_two(self): - """Test one digit CPU #.""" - self.dw.RunCommandOnDut = mock.Mock( - return_value=(0, '/sys/devices/system/cpu/cpu0/online 0\n' - '/sys/devices/system/cpu/cpu1/online 1\n', '')) - cpu_online = self.dw.GetCpuOnline() - self.assertEqual(cpu_online, {0: 0, 1: 1}) - - def test_get_cpu_online_twelve(self): - """Test two digit CPU #.""" - self.dw.RunCommandOnDut = mock.Mock( - return_value=(0, '/sys/devices/system/cpu/cpu0/online 1\n' - '/sys/devices/system/cpu/cpu1/online 0\n' - '/sys/devices/system/cpu/cpu10/online 1\n' - '/sys/devices/system/cpu/cpu11/online 1\n' - '/sys/devices/system/cpu/cpu2/online 1\n' - '/sys/devices/system/cpu/cpu3/online 0\n' - '/sys/devices/system/cpu/cpu4/online 1\n' - '/sys/devices/system/cpu/cpu5/online 0\n' - '/sys/devices/system/cpu/cpu6/online 1\n' - '/sys/devices/system/cpu/cpu7/online 0\n' - '/sys/devices/system/cpu/cpu8/online 1\n' - '/sys/devices/system/cpu/cpu9/online 0\n', '')) - cpu_online = self.dw.GetCpuOnline() - self.assertEqual(cpu_online, { - 0: 1, - 1: 0, - 2: 1, - 3: 0, - 4: 1, - 5: 0, - 6: 1, - 7: 0, - 8: 1, - 9: 0, - 10: 1, - 11: 1 - }) - - def test_get_cpu_online_no_output(self): - """Test error case, no output.""" - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, '', '')) - with self.assertRaises(AssertionError): - self.dw.GetCpuOnline() - - def test_get_cpu_online_command_error(self): - """Test error case, command error.""" - self.dw.RunCommandOnDut = mock.Mock(side_effect=AssertionError) - with self.assertRaises(AssertionError): - self.dw.GetCpuOnline() - - @mock.patch.object(DutWrapper, 'SetupArmCores') - def test_setup_cpu_usage_little_on_arm(self, mock_setup_arm): - self.dw.SetupArmCores = mock_setup_arm - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, 'armv7l', '')) - self.dw.dut_config['cpu_usage'] = 'little_only' - self.dw.SetupCpuUsage() - self.dw.SetupArmCores.assert_called_once_with() - - @mock.patch.object(DutWrapper, 'SetupArmCores') - def test_setup_cpu_usage_big_on_aarch64(self, mock_setup_arm): - self.dw.SetupArmCores = mock_setup_arm - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, 'aarch64', '')) - self.dw.dut_config['cpu_usage'] = 'big_only' - self.dw.SetupCpuUsage() - self.dw.SetupArmCores.assert_called_once_with() - - @mock.patch.object(DutWrapper, 'SetupArmCores') - def test_setup_cpu_usage_big_on_intel(self, mock_setup_arm): - self.dw.SetupArmCores = mock_setup_arm - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, 'x86_64', '')) - self.dw.dut_config['cpu_usage'] = 'big_only' - self.dw.SetupCpuUsage() - # Check that SetupArmCores not called with invalid setup. - self.dw.SetupArmCores.assert_not_called() - - @mock.patch.object(DutWrapper, 'SetupArmCores') - def test_setup_cpu_usage_all_on_intel(self, mock_setup_arm): - self.dw.SetupArmCores = mock_setup_arm - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, 'x86_64', '')) - self.dw.dut_config['cpu_usage'] = 'all' - self.dw.SetupCpuUsage() - # Check that SetupArmCores not called in general case. - self.dw.SetupArmCores.assert_not_called() - - def test_setup_arm_cores_big_on_big_little(self): - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, BIG_LITTLE_CPUINFO, ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_usage'] = 'big_only' - self.dw.SetupArmCores() - self.dw.RunCommandOnDut.assert_called_with( - 'echo 1 | tee /sys/devices/system/cpu/cpu{2}/online; ' - 'echo 0 | tee /sys/devices/system/cpu/cpu{0,1}/online') - - def test_setup_arm_cores_little_on_big_little(self): - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, BIG_LITTLE_CPUINFO, ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_usage'] = 'little_only' - self.dw.SetupArmCores() - self.dw.RunCommandOnDut.assert_called_with( - 'echo 1 | tee /sys/devices/system/cpu/cpu{0,1}/online; ' - 'echo 0 | tee /sys/devices/system/cpu/cpu{2}/online') - - def test_setup_arm_cores_invalid_config(self): - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, LITTLE_ONLY_CPUINFO, ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_usage'] = 'big_only' - self.dw.SetupArmCores() - # Check that setup command is not sent when trying - # to use 'big_only' on a platform with all little cores. - self.dw.RunCommandOnDut.assert_called_once_with('cat /proc/cpuinfo') - - def test_setup_arm_cores_not_big_little(self): - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, NOT_BIG_LITTLE_CPUINFO, ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_usage'] = 'big_only' - self.dw.SetupArmCores() - # Check that setup command is not sent when trying - # to use 'big_only' on a platform w/o support of big/little. - self.dw.RunCommandOnDut.assert_called_once_with('cat /proc/cpuinfo') - - def test_setup_arm_cores_unsupported_cpu_usage(self): - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, BIG_LITTLE_CPUINFO, ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_usage'] = 'exclusive_cores' - self.dw.SetupArmCores() - # Check that setup command is not sent when trying to use - # 'exclusive_cores' on ARM CPU setup. - self.dw.RunCommandOnDut.assert_called_once_with('cat /proc/cpuinfo') - - def test_setup_cpu_freq_single_full(self): - online = [0] - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n', - ''), - (0, '1 2 3 4 5 6 7 8 9 10', ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_freq_pct'] = 100 - self.dw.SetupCpuFreq(online) - self.assertGreaterEqual(self.dw.RunCommandOnDut.call_count, 3) - self.assertEqual( - self.dw.RunCommandOnDut.call_args, - mock.call('echo 10 | tee ' - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq ' - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_min_freq')) - - def test_setup_cpu_freq_middle(self): - online = [0] - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n', - ''), - (0, '1 2 3 4 5 6 7 8 9 10', ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_freq_pct'] = 60 - self.dw.SetupCpuFreq(online) - self.assertGreaterEqual(self.dw.RunCommandOnDut.call_count, 2) - self.assertEqual( - self.dw.RunCommandOnDut.call_args, - mock.call('echo 6 | tee ' - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq ' - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_min_freq')) - - def test_setup_cpu_freq_lowest(self): - online = [0] - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n', - ''), - (0, '1 2 3 4 5 6 7 8 9 10', ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_freq_pct'] = 0 - self.dw.SetupCpuFreq(online) - self.assertGreaterEqual(self.dw.RunCommandOnDut.call_count, 2) - self.assertEqual( - self.dw.RunCommandOnDut.call_args, - mock.call('echo 1 | tee ' - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq ' - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_min_freq')) - - def test_setup_cpu_freq_multiple_middle(self): - online = [0, 1] - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n' - '/sys/devices/system/cpu/cpu1/cpufreq/scaling_available_frequencies\n', - ''), - (0, '1 2 3 4 5 6 7 8 9 10', ''), - (0, '', ''), - (0, '1 4 6 8 10 12 14 16 18 20', ''), - (0, '', ''), - ]) - self.dw.dut_config['cpu_freq_pct'] = 70 - self.dw.SetupCpuFreq(online) - self.assertEqual(self.dw.RunCommandOnDut.call_count, 5) - self.assertEqual( - self.dw.RunCommandOnDut.call_args_list[2], - mock.call('echo 7 | tee ' - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq ' - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_min_freq')) - self.assertEqual( - self.dw.RunCommandOnDut.call_args_list[4], - mock.call('echo 14 | tee ' - '/sys/devices/system/cpu/cpu1/cpufreq/scaling_max_freq ' - '/sys/devices/system/cpu/cpu1/cpufreq/scaling_min_freq')) - - def test_setup_cpu_freq_no_scaling_available(self): - online = [0, 1] - self.dw.RunCommandOnDut = mock.Mock( - return_value=(2, '', 'No such file or directory')) - self.dw.dut_config['cpu_freq_pct'] = 50 - self.dw.SetupCpuFreq(online) - self.dw.RunCommandOnDut.assert_called_once() - self.assertNotRegex(self.dw.RunCommandOnDut.call_args_list[0][0][0], - '^echo.*scaling_max_freq$') - - def test_setup_cpu_freq_multiple_no_access(self): - online = [0, 1] - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, - '/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n' - '/sys/devices/system/cpu/cpu1/cpufreq/scaling_available_frequencies\n', - ''), - (0, '1 4 6 8 10 12 14 16 18 20', ''), - AssertionError(), - ]) - self.dw.dut_config['cpu_freq_pct'] = 30 - # Error status causes log fatal. - with self.assertRaises(AssertionError): - self.dw.SetupCpuFreq(online) - - @mock.patch.object(time, 'sleep') - def test_wait_cooldown_nowait(self, mock_sleep): - mock_sleep.return_value = 0 - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, '39000', '')) - self.dw.dut_config['cooldown_time'] = 10 - self.dw.dut_config['cooldown_temp'] = 40 - wait_time = self.dw.WaitCooldown() - # Send command to DUT only once to check temperature - # and make sure it does not exceed the threshold. - self.dw.RunCommandOnDut.assert_called_once() - mock_sleep.assert_not_called() - self.assertEqual(wait_time, 0) - - @mock.patch.object(time, 'sleep') - def test_wait_cooldown_needwait_once(self, mock_sleep): - """Wait one iteration for cooldown. - - Set large enough timeout and changing temperature - output. Make sure it exits when expected value - received. - Expect that WaitCooldown check temp twice. - """ - mock_sleep.return_value = 0 - self.dw.RunCommandOnDut = mock.Mock(side_effect=[(0, '41000', - ''), (0, '39999', '')]) - self.dw.dut_config['cooldown_time'] = 100 - self.dw.dut_config['cooldown_temp'] = 40 - wait_time = self.dw.WaitCooldown() - self.dw.RunCommandOnDut.assert_called() - self.assertEqual(self.dw.RunCommandOnDut.call_count, 2) - mock_sleep.assert_called() - self.assertGreater(wait_time, 0) - - @mock.patch.object(time, 'sleep') - def test_wait_cooldown_needwait(self, mock_sleep): - """Test exit by timeout. - - Send command to DUT checking the temperature and - check repeatedly until timeout goes off. - Output from temperature sensor never changes. - """ - mock_sleep.return_value = 0 - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, '41000', '')) - self.dw.dut_config['cooldown_time'] = 60 - self.dw.dut_config['cooldown_temp'] = 40 - wait_time = self.dw.WaitCooldown() - self.dw.RunCommandOnDut.assert_called() - self.assertGreater(self.dw.RunCommandOnDut.call_count, 2) - mock_sleep.assert_called() - self.assertGreater(wait_time, 0) - - @mock.patch.object(time, 'sleep') - def test_wait_cooldown_needwait_multtemp(self, mock_sleep): - """Wait until all temps go down. - - Set large enough timeout and changing temperature - output. Make sure it exits when expected value - for all temperatures received. - Expect 3 checks. - """ - mock_sleep.return_value = 0 - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (0, '41000\n20000\n30000\n45000', ''), - (0, '39000\n20000\n30000\n41000', ''), - (0, '39000\n20000\n30000\n31000', ''), - ]) - self.dw.dut_config['cooldown_time'] = 100 - self.dw.dut_config['cooldown_temp'] = 40 - wait_time = self.dw.WaitCooldown() - self.dw.RunCommandOnDut.assert_called() - self.assertEqual(self.dw.RunCommandOnDut.call_count, 3) - mock_sleep.assert_called() - self.assertGreater(wait_time, 0) - - @mock.patch.object(time, 'sleep') - def test_wait_cooldown_thermal_error(self, mock_sleep): - """Handle error status. - - Any error should be considered non-fatal. - """ - mock_sleep.return_value = 0 - self.dw.RunCommandOnDut = mock.Mock(side_effect=[ - (1, '39000\n20000\n30000\n41000', 'Thermal error'), - (1, '39000\n20000\n30000\n31000', 'Thermal error'), - ]) - self.dw.dut_config['cooldown_time'] = 10 - self.dw.dut_config['cooldown_temp'] = 40 - wait_time = self.dw.WaitCooldown() - # Check that errors are ignored. - self.dw.RunCommandOnDut.assert_called_with( - 'cat /sys/class/thermal/thermal_zone*/temp', ignore_status=True) - self.assertEqual(self.dw.RunCommandOnDut.call_count, 2) - # Check that we are waiting even when an error is returned - # as soon as data is coming. - mock_sleep.assert_called() - self.assertGreater(wait_time, 0) - - @mock.patch.object(time, 'sleep') - def test_wait_cooldown_thermal_no_output(self, mock_sleep): - """Handle no output. - - Check handling of empty stdout. - """ - mock_sleep.return_value = 0 - self.dw.RunCommandOnDut = mock.Mock(side_effect=[(1, '', 'Thermal error')]) - self.dw.dut_config['cooldown_time'] = 10 - self.dw.dut_config['cooldown_temp'] = 40 - wait_time = self.dw.WaitCooldown() - # Check that errors are ignored. - self.dw.RunCommandOnDut.assert_called_once_with( - 'cat /sys/class/thermal/thermal_zone*/temp', ignore_status=True) - # No wait. - mock_sleep.assert_not_called() - self.assertEqual(wait_time, 0) - - @mock.patch.object(time, 'sleep') - def test_wait_cooldown_thermal_ws_output(self, mock_sleep): - """Handle whitespace output. - - Check handling of whitespace only. - """ - mock_sleep.return_value = 0 - self.dw.RunCommandOnDut = mock.Mock(side_effect=[(1, '\n', - 'Thermal error')]) - self.dw.dut_config['cooldown_time'] = 10 - self.dw.dut_config['cooldown_temp'] = 40 - wait_time = self.dw.WaitCooldown() - # Check that errors are ignored. - self.dw.RunCommandOnDut.assert_called_once_with( - 'cat /sys/class/thermal/thermal_zone*/temp', ignore_status=True) - # No wait. - mock_sleep.assert_not_called() - self.assertEqual(wait_time, 0) - - def test_stop_ui(self): - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, '', '')) - self.dw.StopUI() - self.dw.RunCommandOnDut.assert_called_once_with( - 'stop ui', ignore_status=True) - - def test_start_ui(self): - self.dw.RunCommandOnDut = mock.Mock(return_value=(0, '', '')) - self.dw.StartUI() - self.dw.RunCommandOnDut.assert_called_once_with( - 'start ui', ignore_status=True) - - def test_setup_device(self): - - def FakeRunner(command, ignore_status=False): - # pylint fix for unused variable. - del command, ignore_status - return 0, '', '' - - def SetupMockFunctions(): - self.dw.RunCommandOnDut = mock.Mock(return_value=FakeRunner) - self.dw.KerncmdUpdateNeeded = mock.Mock(return_value=True) - self.dw.UpdateKerncmdIntelPstate = mock.Mock(return_value=0) - self.dw.DisableASLR = mock.Mock(return_value=0) - self.dw.SetupCpuUsage = mock.Mock(return_value=0) - self.dw.SetupCpuFreq = mock.Mock(return_value=0) - self.dw.GetCpuOnline = mock.Mock(return_value={0: 1, 1: 1, 2: 0}) - self.dw.SetCpuGovernor = mock.Mock(return_value=0) - self.dw.DisableTurbo = mock.Mock(return_value=0) - self.dw.StopUI = mock.Mock(return_value=0) - self.dw.StartUI = mock.Mock(return_value=0) - self.dw.WaitCooldown = mock.Mock(return_value=0) - self.dw.DecreaseWaitTime = mock.Mock(return_value=0) - - self.dw.dut_config['enable_aslr'] = False - self.dw.dut_config['cooldown_time'] = 0 - self.dw.dut_config['governor'] = 'fake_governor' - self.dw.dut_config['cpu_freq_pct'] = 65 - self.dw.dut_config['intel_pstate'] = 'no_hwp' - - SetupMockFunctions() - self.dw.SetupDevice() - - self.dw.KerncmdUpdateNeeded.assert_called_once() - self.dw.UpdateKerncmdIntelPstate.assert_called_once() - self.dw.DisableASLR.assert_called_once() - self.dw.SetupCpuUsage.assert_called_once_with() - self.dw.SetupCpuFreq.assert_called_once_with([0, 1]) - self.dw.GetCpuOnline.assert_called_once_with() - self.dw.SetCpuGovernor.assert_called_once_with('fake_governor') - self.dw.DisableTurbo.assert_called_once_with() - self.dw.DecreaseWaitTime.assert_called_once_with() - self.dw.StopUI.assert_called_once_with() - self.dw.StartUI.assert_called_once_with() - self.dw.WaitCooldown.assert_not_called() - - # Test SetupDevice with cooldown - self.dw.dut_config['cooldown_time'] = 10 - - SetupMockFunctions() - self.dw.GetCpuOnline = mock.Mock(return_value={0: 0, 1: 1}) - - self.dw.SetupDevice() - - self.dw.WaitCooldown.assert_called_once_with() - self.dw.DisableASLR.assert_called_once() - self.dw.DisableTurbo.assert_called_once_with() - self.dw.SetupCpuUsage.assert_called_once_with() - self.dw.SetupCpuFreq.assert_called_once_with([1]) - self.dw.SetCpuGovernor.assert_called() - self.dw.GetCpuOnline.assert_called_once_with() - self.dw.StopUI.assert_called_once_with() - self.dw.StartUI.assert_called_once_with() - self.assertGreater(self.dw.SetCpuGovernor.call_count, 1) - self.assertEqual(self.dw.SetCpuGovernor.call_args, - mock.call('fake_governor')) - - # Test SetupDevice with cooldown - SetupMockFunctions() - self.dw.SetupCpuUsage = mock.Mock(side_effect=RuntimeError()) - - with self.assertRaises(RuntimeError): - self.dw.SetupDevice() - - # This call injected an exception. - self.dw.SetupCpuUsage.assert_called_once_with() - # Calls following the expeption are skipped. - self.dw.WaitCooldown.assert_not_called() - self.dw.DisableTurbo.assert_not_called() - self.dw.SetupCpuFreq.assert_not_called() - self.dw.SetCpuGovernor.assert_not_called() - self.dw.GetCpuOnline.assert_not_called() - # Check that Stop/Start UI are always called. - self.dw.StopUI.assert_called_once_with() - self.dw.StartUI.assert_called_once_with() - - -if __name__ == '__main__': - unittest.main() + """Class of DutWrapper test.""" + + real_logger = logger.GetLogger() + + mock_cmd_exec = mock.Mock(spec=command_executer.CommandExecuter) + mock_logger = mock.Mock(spec=logger.Logger) + + def __init__(self, *args, **kwargs): + super(DutWrapperTest, self).__init__(*args, **kwargs) + + def setUp(self): + self.dw = DutWrapper( + "/tmp/chromeos", + "lumpy.cros2", + log_level="verbose", + logger=self.mock_logger, + ce=self.mock_cmd_exec, + dut_config={}, + ) + + @mock.patch.object( + command_executer.CommandExecuter, "CrosRunCommandWOutput" + ) + def test_run_command_on_dut(self, mock_cros_runcmd): + self.mock_cmd_exec.CrosRunCommandWOutput = mock_cros_runcmd + mock_cros_runcmd.return_value = (0, "", "") + mock_cros_runcmd.assert_not_called() + self.dw.RunCommandOnDut("run command;") + mock_cros_runcmd.assert_called_once_with( + "run command;", chromeos_root="/tmp/chromeos", machine="lumpy.cros2" + ) + + @mock.patch.object( + command_executer.CommandExecuter, "CrosRunCommandWOutput" + ) + def test_dut_wrapper_fatal_error(self, mock_cros_runcmd): + self.mock_cmd_exec.CrosRunCommandWOutput = mock_cros_runcmd + # Command returns error 1. + mock_cros_runcmd.return_value = (1, "", "Error!") + mock_cros_runcmd.assert_not_called() + self.dw.RunCommandOnDut("run command;") + mock_cros_runcmd.assert_called_once_with( + "run command;", chromeos_root="/tmp/chromeos", machine="lumpy.cros2" + ) + # Error status causes log fatal. + self.assertEqual( + self.mock_logger.method_calls[-1], + mock.call.LogFatal( + "Command execution on DUT lumpy.cros2 failed.\n" + "Failing command: run command;\nreturned 1\n" + "Error message: Error!" + ), + ) + + @mock.patch.object( + command_executer.CommandExecuter, "CrosRunCommandWOutput" + ) + def test_dut_wrapper_ignore_error(self, mock_cros_runcmd): + self.mock_cmd_exec.CrosRunCommandWOutput = mock_cros_runcmd + # Command returns error 1. + mock_cros_runcmd.return_value = (1, "", "Error!") + self.dw.RunCommandOnDut("run command;", ignore_status=True) + mock_cros_runcmd.assert_called_once_with( + "run command;", chromeos_root="/tmp/chromeos", machine="lumpy.cros2" + ) + # Error status is not fatal. LogError records the error message. + self.assertEqual( + self.mock_logger.method_calls[-1], + mock.call.LogError( + "Command execution on DUT lumpy.cros2 failed.\n" + "Failing command: run command;\nreturned 1\n" + "Error message: Error!\n" + "(Failure is considered non-fatal. Continue.)" + ), + ) + + def test_disable_aslr(self): + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "", "")) + self.dw.DisableASLR() + # pyformat: disable + set_cpu_cmd = ( + "set -e; " + "if [[ -e /proc/sys/kernel/randomize_va_space ]]; then " + " echo 0 > /proc/sys/kernel/randomize_va_space; " + "fi" + ) + self.dw.RunCommandOnDut.assert_called_once_with(set_cpu_cmd) + + def test_set_cpu_governor(self): + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "", "")) + self.dw.SetCpuGovernor("new_governor", ignore_status=False) + set_cpu_cmd = ( + "for f in `ls -d /sys/devices/system/cpu/cpu*/cpufreq 2>/dev/null`; do " + # Skip writing scaling_governor if cpu is offline. + " [[ -e ${f/cpufreq/online} ]] && grep -q 0 ${f/cpufreq/online} " + " && continue; " + " cd $f; " + " if [[ -e scaling_governor ]]; then " + " echo %s > scaling_governor; fi; " + "done; " + ) + self.dw.RunCommandOnDut.assert_called_once_with( + set_cpu_cmd % "new_governor", ignore_status=False + ) + + def test_set_cpu_governor_propagate_error(self): + self.dw.RunCommandOnDut = mock.Mock(return_value=(1, "", "Error.")) + self.dw.SetCpuGovernor("non-exist_governor") + set_cpu_cmd = ( + "for f in `ls -d /sys/devices/system/cpu/cpu*/cpufreq 2>/dev/null`; do " + # Skip writing scaling_governor if cpu is not online. + " [[ -e ${f/cpufreq/online} ]] && grep -q 0 ${f/cpufreq/online} " + " && continue; " + " cd $f; " + " if [[ -e scaling_governor ]]; then " + " echo %s > scaling_governor; fi; " + "done; " + ) + # By default error status is fatal. + self.dw.RunCommandOnDut.assert_called_once_with( + set_cpu_cmd % "non-exist_governor", ignore_status=False + ) + + def test_set_cpu_governor_ignore_status(self): + self.dw.RunCommandOnDut = mock.Mock(return_value=(1, "", "Error.")) + ret_code = self.dw.SetCpuGovernor( + "non-exist_governor", ignore_status=True + ) + set_cpu_cmd = ( + "for f in `ls -d /sys/devices/system/cpu/cpu*/cpufreq 2>/dev/null`; do " + # Skip writing scaling_governor if cpu is not online. + " [[ -e ${f/cpufreq/online} ]] && grep -q 0 ${f/cpufreq/online} " + " && continue; " + " cd $f; " + " if [[ -e scaling_governor ]]; then " + " echo %s > scaling_governor; fi; " + "done; " + ) + self.dw.RunCommandOnDut.assert_called_once_with( + set_cpu_cmd % "non-exist_governor", ignore_status=True + ) + self.assertEqual(ret_code, 1) + + def test_disable_turbo(self): + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "", "")) + self.dw.DisableTurbo() + set_cpu_cmd = ( + # Disable Turbo in Intel pstate driver + "if [[ -e /sys/devices/system/cpu/intel_pstate/no_turbo ]]; then " + " if grep -q 0 /sys/devices/system/cpu/intel_pstate/no_turbo; then " + " echo -n 1 > /sys/devices/system/cpu/intel_pstate/no_turbo; " + " fi; " + "fi; " + ) + self.dw.RunCommandOnDut.assert_called_once_with(set_cpu_cmd) + + def test_get_cpu_online_two(self): + """Test one digit CPU #.""" + self.dw.RunCommandOnDut = mock.Mock( + return_value=( + 0, + "/sys/devices/system/cpu/cpu0/online 0\n" + "/sys/devices/system/cpu/cpu1/online 1\n", + "", + ) + ) + cpu_online = self.dw.GetCpuOnline() + self.assertEqual(cpu_online, {0: 0, 1: 1}) + + def test_get_cpu_online_twelve(self): + """Test two digit CPU #.""" + self.dw.RunCommandOnDut = mock.Mock( + return_value=( + 0, + "/sys/devices/system/cpu/cpu0/online 1\n" + "/sys/devices/system/cpu/cpu1/online 0\n" + "/sys/devices/system/cpu/cpu10/online 1\n" + "/sys/devices/system/cpu/cpu11/online 1\n" + "/sys/devices/system/cpu/cpu2/online 1\n" + "/sys/devices/system/cpu/cpu3/online 0\n" + "/sys/devices/system/cpu/cpu4/online 1\n" + "/sys/devices/system/cpu/cpu5/online 0\n" + "/sys/devices/system/cpu/cpu6/online 1\n" + "/sys/devices/system/cpu/cpu7/online 0\n" + "/sys/devices/system/cpu/cpu8/online 1\n" + "/sys/devices/system/cpu/cpu9/online 0\n", + "", + ) + ) + cpu_online = self.dw.GetCpuOnline() + self.assertEqual( + cpu_online, + { + 0: 1, + 1: 0, + 2: 1, + 3: 0, + 4: 1, + 5: 0, + 6: 1, + 7: 0, + 8: 1, + 9: 0, + 10: 1, + 11: 1, + }, + ) + + def test_get_cpu_online_no_output(self): + """Test error case, no output.""" + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "", "")) + with self.assertRaises(AssertionError): + self.dw.GetCpuOnline() + + def test_get_cpu_online_command_error(self): + """Test error case, command error.""" + self.dw.RunCommandOnDut = mock.Mock(side_effect=AssertionError) + with self.assertRaises(AssertionError): + self.dw.GetCpuOnline() + + @mock.patch.object(DutWrapper, "SetupArmCores") + def test_setup_cpu_usage_little_on_arm(self, mock_setup_arm): + self.dw.SetupArmCores = mock_setup_arm + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "armv7l", "")) + self.dw.dut_config["cpu_usage"] = "little_only" + self.dw.SetupCpuUsage() + self.dw.SetupArmCores.assert_called_once_with() + + @mock.patch.object(DutWrapper, "SetupArmCores") + def test_setup_cpu_usage_big_on_aarch64(self, mock_setup_arm): + self.dw.SetupArmCores = mock_setup_arm + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "aarch64", "")) + self.dw.dut_config["cpu_usage"] = "big_only" + self.dw.SetupCpuUsage() + self.dw.SetupArmCores.assert_called_once_with() + + @mock.patch.object(DutWrapper, "SetupArmCores") + def test_setup_cpu_usage_big_on_intel(self, mock_setup_arm): + self.dw.SetupArmCores = mock_setup_arm + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "x86_64", "")) + self.dw.dut_config["cpu_usage"] = "big_only" + self.dw.SetupCpuUsage() + # Check that SetupArmCores not called with invalid setup. + self.dw.SetupArmCores.assert_not_called() + + @mock.patch.object(DutWrapper, "SetupArmCores") + def test_setup_cpu_usage_all_on_intel(self, mock_setup_arm): + self.dw.SetupArmCores = mock_setup_arm + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "x86_64", "")) + self.dw.dut_config["cpu_usage"] = "all" + self.dw.SetupCpuUsage() + # Check that SetupArmCores not called in general case. + self.dw.SetupArmCores.assert_not_called() + + def test_setup_arm_cores_big_on_big_little(self): + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + (0, BIG_LITTLE_CPUINFO, ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_usage"] = "big_only" + self.dw.SetupArmCores() + self.dw.RunCommandOnDut.assert_called_with( + "echo 1 | tee /sys/devices/system/cpu/cpu{2}/online; " + "echo 0 | tee /sys/devices/system/cpu/cpu{0,1}/online" + ) + + def test_setup_arm_cores_little_on_big_little(self): + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + (0, BIG_LITTLE_CPUINFO, ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_usage"] = "little_only" + self.dw.SetupArmCores() + self.dw.RunCommandOnDut.assert_called_with( + "echo 1 | tee /sys/devices/system/cpu/cpu{0,1}/online; " + "echo 0 | tee /sys/devices/system/cpu/cpu{2}/online" + ) + + def test_setup_arm_cores_invalid_config(self): + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + (0, LITTLE_ONLY_CPUINFO, ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_usage"] = "big_only" + self.dw.SetupArmCores() + # Check that setup command is not sent when trying + # to use 'big_only' on a platform with all little cores. + self.dw.RunCommandOnDut.assert_called_once_with("cat /proc/cpuinfo") + + def test_setup_arm_cores_not_big_little(self): + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + (0, NOT_BIG_LITTLE_CPUINFO, ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_usage"] = "big_only" + self.dw.SetupArmCores() + # Check that setup command is not sent when trying + # to use 'big_only' on a platform w/o support of big/little. + self.dw.RunCommandOnDut.assert_called_once_with("cat /proc/cpuinfo") + + def test_setup_arm_cores_unsupported_cpu_usage(self): + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + (0, BIG_LITTLE_CPUINFO, ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_usage"] = "exclusive_cores" + self.dw.SetupArmCores() + # Check that setup command is not sent when trying to use + # 'exclusive_cores' on ARM CPU setup. + self.dw.RunCommandOnDut.assert_called_once_with("cat /proc/cpuinfo") + + def test_setup_cpu_freq_single_full(self): + online = [0] + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + ( + 0, + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n", + "", + ), + (0, "1 2 3 4 5 6 7 8 9 10", ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_freq_pct"] = 100 + self.dw.SetupCpuFreq(online) + self.assertGreaterEqual(self.dw.RunCommandOnDut.call_count, 3) + self.assertEqual( + self.dw.RunCommandOnDut.call_args, + mock.call( + "echo 10 | tee " + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq " + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_min_freq" + ), + ) + + def test_setup_cpu_freq_middle(self): + online = [0] + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + ( + 0, + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n", + "", + ), + (0, "1 2 3 4 5 6 7 8 9 10", ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_freq_pct"] = 60 + self.dw.SetupCpuFreq(online) + self.assertGreaterEqual(self.dw.RunCommandOnDut.call_count, 2) + self.assertEqual( + self.dw.RunCommandOnDut.call_args, + mock.call( + "echo 6 | tee " + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq " + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_min_freq" + ), + ) + + def test_setup_cpu_freq_lowest(self): + online = [0] + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + ( + 0, + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n", + "", + ), + (0, "1 2 3 4 5 6 7 8 9 10", ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_freq_pct"] = 0 + self.dw.SetupCpuFreq(online) + self.assertGreaterEqual(self.dw.RunCommandOnDut.call_count, 2) + self.assertEqual( + self.dw.RunCommandOnDut.call_args, + mock.call( + "echo 1 | tee " + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq " + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_min_freq" + ), + ) + + def test_setup_cpu_freq_multiple_middle(self): + online = [0, 1] + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + ( + 0, + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n" + "/sys/devices/system/cpu/cpu1/cpufreq/scaling_available_frequencies\n", + "", + ), + (0, "1 2 3 4 5 6 7 8 9 10", ""), + (0, "", ""), + (0, "1 4 6 8 10 12 14 16 18 20", ""), + (0, "", ""), + ] + ) + self.dw.dut_config["cpu_freq_pct"] = 70 + self.dw.SetupCpuFreq(online) + self.assertEqual(self.dw.RunCommandOnDut.call_count, 5) + self.assertEqual( + self.dw.RunCommandOnDut.call_args_list[2], + mock.call( + "echo 7 | tee " + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq " + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_min_freq" + ), + ) + self.assertEqual( + self.dw.RunCommandOnDut.call_args_list[4], + mock.call( + "echo 14 | tee " + "/sys/devices/system/cpu/cpu1/cpufreq/scaling_max_freq " + "/sys/devices/system/cpu/cpu1/cpufreq/scaling_min_freq" + ), + ) + + def test_setup_cpu_freq_no_scaling_available(self): + online = [0, 1] + self.dw.RunCommandOnDut = mock.Mock( + return_value=(2, "", "No such file or directory") + ) + self.dw.dut_config["cpu_freq_pct"] = 50 + self.dw.SetupCpuFreq(online) + self.dw.RunCommandOnDut.assert_called_once() + self.assertNotRegex( + self.dw.RunCommandOnDut.call_args_list[0][0][0], + "^echo.*scaling_max_freq$", + ) + + def test_setup_cpu_freq_multiple_no_access(self): + online = [0, 1] + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + ( + 0, + "/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies\n" + "/sys/devices/system/cpu/cpu1/cpufreq/scaling_available_frequencies\n", + "", + ), + (0, "1 4 6 8 10 12 14 16 18 20", ""), + AssertionError(), + ] + ) + self.dw.dut_config["cpu_freq_pct"] = 30 + # Error status causes log fatal. + with self.assertRaises(AssertionError): + self.dw.SetupCpuFreq(online) + + @mock.patch.object(time, "sleep") + def test_wait_cooldown_nowait(self, mock_sleep): + mock_sleep.return_value = 0 + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "39000", "")) + self.dw.dut_config["cooldown_time"] = 10 + self.dw.dut_config["cooldown_temp"] = 40 + wait_time = self.dw.WaitCooldown() + # Send command to DUT only once to check temperature + # and make sure it does not exceed the threshold. + self.dw.RunCommandOnDut.assert_called_once() + mock_sleep.assert_not_called() + self.assertEqual(wait_time, 0) + + @mock.patch.object(time, "sleep") + def test_wait_cooldown_needwait_once(self, mock_sleep): + """Wait one iteration for cooldown. + + Set large enough timeout and changing temperature + output. Make sure it exits when expected value + received. + Expect that WaitCooldown check temp twice. + """ + mock_sleep.return_value = 0 + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[(0, "41000", ""), (0, "39999", "")] + ) + self.dw.dut_config["cooldown_time"] = 100 + self.dw.dut_config["cooldown_temp"] = 40 + wait_time = self.dw.WaitCooldown() + self.dw.RunCommandOnDut.assert_called() + self.assertEqual(self.dw.RunCommandOnDut.call_count, 2) + mock_sleep.assert_called() + self.assertGreater(wait_time, 0) + + @mock.patch.object(time, "sleep") + def test_wait_cooldown_needwait(self, mock_sleep): + """Test exit by timeout. + + Send command to DUT checking the temperature and + check repeatedly until timeout goes off. + Output from temperature sensor never changes. + """ + mock_sleep.return_value = 0 + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "41000", "")) + self.dw.dut_config["cooldown_time"] = 60 + self.dw.dut_config["cooldown_temp"] = 40 + wait_time = self.dw.WaitCooldown() + self.dw.RunCommandOnDut.assert_called() + self.assertGreater(self.dw.RunCommandOnDut.call_count, 2) + mock_sleep.assert_called() + self.assertGreater(wait_time, 0) + + @mock.patch.object(time, "sleep") + def test_wait_cooldown_needwait_multtemp(self, mock_sleep): + """Wait until all temps go down. + + Set large enough timeout and changing temperature + output. Make sure it exits when expected value + for all temperatures received. + Expect 3 checks. + """ + mock_sleep.return_value = 0 + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + (0, "41000\n20000\n30000\n45000", ""), + (0, "39000\n20000\n30000\n41000", ""), + (0, "39000\n20000\n30000\n31000", ""), + ] + ) + self.dw.dut_config["cooldown_time"] = 100 + self.dw.dut_config["cooldown_temp"] = 40 + wait_time = self.dw.WaitCooldown() + self.dw.RunCommandOnDut.assert_called() + self.assertEqual(self.dw.RunCommandOnDut.call_count, 3) + mock_sleep.assert_called() + self.assertGreater(wait_time, 0) + + @mock.patch.object(time, "sleep") + def test_wait_cooldown_thermal_error(self, mock_sleep): + """Handle error status. + + Any error should be considered non-fatal. + """ + mock_sleep.return_value = 0 + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[ + (1, "39000\n20000\n30000\n41000", "Thermal error"), + (1, "39000\n20000\n30000\n31000", "Thermal error"), + ] + ) + self.dw.dut_config["cooldown_time"] = 10 + self.dw.dut_config["cooldown_temp"] = 40 + wait_time = self.dw.WaitCooldown() + # Check that errors are ignored. + self.dw.RunCommandOnDut.assert_called_with( + "cat /sys/class/thermal/thermal_zone*/temp", ignore_status=True + ) + self.assertEqual(self.dw.RunCommandOnDut.call_count, 2) + # Check that we are waiting even when an error is returned + # as soon as data is coming. + mock_sleep.assert_called() + self.assertGreater(wait_time, 0) + + @mock.patch.object(time, "sleep") + def test_wait_cooldown_thermal_no_output(self, mock_sleep): + """Handle no output. + + Check handling of empty stdout. + """ + mock_sleep.return_value = 0 + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[(1, "", "Thermal error")] + ) + self.dw.dut_config["cooldown_time"] = 10 + self.dw.dut_config["cooldown_temp"] = 40 + wait_time = self.dw.WaitCooldown() + # Check that errors are ignored. + self.dw.RunCommandOnDut.assert_called_once_with( + "cat /sys/class/thermal/thermal_zone*/temp", ignore_status=True + ) + # No wait. + mock_sleep.assert_not_called() + self.assertEqual(wait_time, 0) + + @mock.patch.object(time, "sleep") + def test_wait_cooldown_thermal_ws_output(self, mock_sleep): + """Handle whitespace output. + + Check handling of whitespace only. + """ + mock_sleep.return_value = 0 + self.dw.RunCommandOnDut = mock.Mock( + side_effect=[(1, "\n", "Thermal error")] + ) + self.dw.dut_config["cooldown_time"] = 10 + self.dw.dut_config["cooldown_temp"] = 40 + wait_time = self.dw.WaitCooldown() + # Check that errors are ignored. + self.dw.RunCommandOnDut.assert_called_once_with( + "cat /sys/class/thermal/thermal_zone*/temp", ignore_status=True + ) + # No wait. + mock_sleep.assert_not_called() + self.assertEqual(wait_time, 0) + + def test_stop_ui(self): + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "", "")) + self.dw.StopUI() + self.dw.RunCommandOnDut.assert_called_once_with( + "stop ui", ignore_status=True + ) + + def test_start_ui(self): + self.dw.RunCommandOnDut = mock.Mock(return_value=(0, "", "")) + self.dw.StartUI() + self.dw.RunCommandOnDut.assert_called_once_with( + "start ui", ignore_status=True + ) + + def test_setup_device(self): + def FakeRunner(command, ignore_status=False): + # pylint fix for unused variable. + del command, ignore_status + return 0, "", "" + + def SetupMockFunctions(): + self.dw.RunCommandOnDut = mock.Mock(return_value=FakeRunner) + self.dw.KerncmdUpdateNeeded = mock.Mock(return_value=True) + self.dw.UpdateKerncmdIntelPstate = mock.Mock(return_value=0) + self.dw.DisableASLR = mock.Mock(return_value=0) + self.dw.SetupCpuUsage = mock.Mock(return_value=0) + self.dw.SetupCpuFreq = mock.Mock(return_value=0) + self.dw.GetCpuOnline = mock.Mock(return_value={0: 1, 1: 1, 2: 0}) + self.dw.SetCpuGovernor = mock.Mock(return_value=0) + self.dw.DisableTurbo = mock.Mock(return_value=0) + self.dw.StopUI = mock.Mock(return_value=0) + self.dw.StartUI = mock.Mock(return_value=0) + self.dw.WaitCooldown = mock.Mock(return_value=0) + self.dw.DecreaseWaitTime = mock.Mock(return_value=0) + + self.dw.dut_config["enable_aslr"] = False + self.dw.dut_config["cooldown_time"] = 0 + self.dw.dut_config["governor"] = "fake_governor" + self.dw.dut_config["cpu_freq_pct"] = 65 + self.dw.dut_config["intel_pstate"] = "no_hwp" + + SetupMockFunctions() + self.dw.SetupDevice() + + self.dw.KerncmdUpdateNeeded.assert_called_once() + self.dw.UpdateKerncmdIntelPstate.assert_called_once() + self.dw.DisableASLR.assert_called_once() + self.dw.SetupCpuUsage.assert_called_once_with() + self.dw.SetupCpuFreq.assert_called_once_with([0, 1]) + self.dw.GetCpuOnline.assert_called_once_with() + self.dw.SetCpuGovernor.assert_called_once_with("fake_governor") + self.dw.DisableTurbo.assert_called_once_with() + self.dw.DecreaseWaitTime.assert_called_once_with() + self.dw.StopUI.assert_called_once_with() + self.dw.StartUI.assert_called_once_with() + self.dw.WaitCooldown.assert_not_called() + + # Test SetupDevice with cooldown + self.dw.dut_config["cooldown_time"] = 10 + + SetupMockFunctions() + self.dw.GetCpuOnline = mock.Mock(return_value={0: 0, 1: 1}) + + self.dw.SetupDevice() + + self.dw.WaitCooldown.assert_called_once_with() + self.dw.DisableASLR.assert_called_once() + self.dw.DisableTurbo.assert_called_once_with() + self.dw.SetupCpuUsage.assert_called_once_with() + self.dw.SetupCpuFreq.assert_called_once_with([1]) + self.dw.SetCpuGovernor.assert_called() + self.dw.GetCpuOnline.assert_called_once_with() + self.dw.StopUI.assert_called_once_with() + self.dw.StartUI.assert_called_once_with() + self.assertGreater(self.dw.SetCpuGovernor.call_count, 1) + self.assertEqual( + self.dw.SetCpuGovernor.call_args, mock.call("fake_governor") + ) + + # Test SetupDevice with cooldown + SetupMockFunctions() + self.dw.SetupCpuUsage = mock.Mock(side_effect=RuntimeError()) + + with self.assertRaises(RuntimeError): + self.dw.SetupDevice() + + # This call injected an exception. + self.dw.SetupCpuUsage.assert_called_once_with() + # Calls following the expeption are skipped. + self.dw.WaitCooldown.assert_not_called() + self.dw.DisableTurbo.assert_not_called() + self.dw.SetupCpuFreq.assert_not_called() + self.dw.SetCpuGovernor.assert_not_called() + self.dw.GetCpuOnline.assert_not_called() + # Check that Stop/Start UI are always called. + self.dw.StopUI.assert_called_once_with() + self.dw.StartUI.assert_called_once_with() + + +if __name__ == "__main__": + unittest.main() |