| #!/usr/bin/env python3 |
| # -*- coding: utf-8 -*- |
| # |
| # 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.""" |
| |
| |
| import time |
| import unittest |
| from unittest import mock |
| |
| 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 |
| Features : half thumb fastmult vfp edsp neon vfpv3 tls vfpv4 |
| CPU implementer : 0x41 |
| CPU architecture: 8 |
| CPU variant : 0x0 |
| CPU part : 0xd03 |
| CPU revision : 4 |
| |
| processor : 1 |
| model name : ARMv8 Processor rev 4 (v8l) |
| BogoMIPS : 48.00 |
| Features : half thumb fastmult vfp edsp neon vfpv3 tls vfpv4 |
| CPU implementer : 0x41 |
| CPU architecture: 8 |
| CPU variant : 0x0 |
| CPU part : 0xd03 |
| CPU revision : 4 |
| |
| processor : 2 |
| model name : ARMv8 Processor rev 2 (v8l) |
| BogoMIPS : 48.00 |
| Features : half thumb fastmult vfp edsp neon vfpv3 tls vfpv4 |
| CPU implementer : 0x41 |
| CPU architecture: 8 |
| CPU variant : 0x0 |
| CPU part : 0xd08 |
| CPU revision : 2 |
| """ |
| LITTLE_ONLY_CPUINFO = """processor : 0 |
| model name : ARMv8 Processor rev 4 (v8l) |
| BogoMIPS : 48.00 |
| Features : half thumb fastmult vfp edsp neon vfpv3 tls vfpv4 |
| CPU implementer : 0x41 |
| CPU architecture: 8 |
| CPU variant : 0x0 |
| CPU part : 0xd03 |
| CPU revision : 4 |
| |
| processor : 1 |
| model name : ARMv8 Processor rev 4 (v8l) |
| BogoMIPS : 48.00 |
| Features : half thumb fastmult vfp edsp neon vfpv3 tls vfpv4 |
| CPU implementer : 0x41 |
| CPU architecture: 8 |
| CPU variant : 0x0 |
| CPU part : 0xd03 |
| CPU revision : 4 |
| """ |
| |
| NOT_BIG_LITTLE_CPUINFO = """processor : 0 |
| model name : ARMv7 Processor rev 1 (v7l) |
| Features : swp half thumb fastmult vfp edsp thumbee neon vfpv3 tls vfpv4 |
| CPU implementer : 0x41 |
| CPU architecture: 7 |
| CPU variant : 0x0 |
| CPU part : 0xc0d |
| CPU revision : 1 |
| |
| processor : 1 |
| model name : ARMv7 Processor rev 1 (v7l) |
| Features : swp half thumb fastmult vfp edsp thumbee neon vfpv3 tls vfpv4 |
| CPU implementer : 0x41 |
| CPU architecture: 7 |
| CPU variant : 0x0 |
| CPU part : 0xc0d |
| CPU revision : 1 |
| |
| Hardware : Rockchip (Device Tree) |
| Revision : 0000 |
| 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() |
