Skip to content

Commit 53ad7cd

Browse files
committed
fix the mock patch function
1 parent 7c8003d commit 53ad7cd

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

tests/determine-reboot-cause_test.py

+12-12
Original file line numberDiff line numberDiff line change
@@ -119,33 +119,33 @@ def test_get_reboot_cause_dict_kernel_panic(self):
119119
assert reboot_cause_dict == EXPECTED_KERNEL_PANIC_REBOOT_CAUSE_DICT
120120

121121
def test_determine_reboot_cause_hardware(self):
122-
with mock.patch("determine_reboot_cause.find_proc_cmdline_reboot_cause", return_value="Unknown"):
123-
with mock.patch("determine_reboot_cause.find_software_reboot_cause", return_value="Power Cycle"):
124-
with mock.patch("determine_reboot_cause.find_hardware_reboot_cause", return_value="Unknown"):
122+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_proc_cmdline_reboot_cause", return_value="Unknown"):
123+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_software_reboot_cause", return_value="Power Cycle"):
124+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_hardware_reboot_cause", return_value="Unknown"):
125125
previous_reboot_cause, additional_info = determine_reboot_cause.determine_reboot_cause()
126126
assert previous_reboot_cause == "Power Cycle"
127127
assert additional_info == "N/A"
128128

129129
def test_determine_reboot_cause_software(self):
130-
with mock.patch("determine_reboot_cause.find_proc_cmdline_reboot_cause", return_value="Unknown"):
131-
with mock.patch("determine_reboot_cause.find_software_reboot_cause", return_value=EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER):
132-
with mock.patch("determine_reboot_cause.find_hardware_reboot_cause", return_value="Unknown"):
130+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_proc_cmdline_reboot_cause", return_value="Unknown"):
131+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_software_reboot_cause", return_value=EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER):
132+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_hardware_reboot_cause", return_value="Unknown"):
133133
previous_reboot_cause, additional_info = determine_reboot_cause.determine_reboot_cause()
134134
assert previous_reboot_cause == EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER
135135
assert additional_info == "N/A"
136136

137137
def test_determine_reboot_cause_cmdline(self):
138-
with mock.patch("determine_reboot_cause.find_proc_cmdline_reboot_cause", return_value=EXPECTED_PARSE_WARMFAST_REBOOT_FROM_PROC_CMDLINE):
139-
with mock.patch("determine_reboot_cause.find_software_reboot_cause", return_value=EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER):
140-
with mock.patch("determine_reboot_cause.find_hardware_reboot_cause", return_value="Unknown"):
138+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_proc_cmdline_reboot_cause", return_value=EXPECTED_PARSE_WARMFAST_REBOOT_FROM_PROC_CMDLINE):
139+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_software_reboot_cause", return_value=EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER):
140+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_hardware_reboot_cause", return_value="Unknown"):
141141
previous_reboot_cause, additional_info = determine_reboot_cause.determine_reboot_cause()
142142
assert previous_reboot_cause == EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER
143143
assert additional_info == "N/A"
144144

145145
def test_determine_reboot_cause_cmdline_hardware(self):
146-
with mock.patch("determine_reboot_cause.find_proc_cmdline_reboot_cause", return_value=EXPECTED_PARSE_WARMFAST_REBOOT_FROM_PROC_CMDLINE):
147-
with mock.patch("determine_reboot_cause.find_software_reboot_cause", return_value=EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER):
148-
with mock.patch("determine_reboot_cause.find_hardware_reboot_cause", return_value=REBOOT_CAUSE_WATCHDOG):
146+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_proc_cmdline_reboot_cause", return_value=EXPECTED_PARSE_WARMFAST_REBOOT_FROM_PROC_CMDLINE):
147+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_software_reboot_cause", return_value=EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER):
148+
with mock.patch("determine_reboot_cause.determine_reboot_cause.find_hardware_reboot_cause", return_value=REBOOT_CAUSE_WATCHDOG):
149149
previous_reboot_cause, additional_info = determine_reboot_cause.determine_reboot_cause()
150150
assert previous_reboot_cause == REBOOT_CAUSE_WATCHDOG
151151
assert additional_info == EXPECTED_FIND_SOFTWARE_REBOOT_CAUSE_USER

0 commit comments

Comments
 (0)