Fix and add hardware reboot cause determination tests

sonic-net · Aug 18, 2022 · 8720561 · 8720561
1 parent 0dcc7fe
commit 8720561
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/determine-reboot-cause_test.py b/tests/determine-reboot-cause_test.py
@@ -109,7 +109,12 @@ def test_find_proc_cmdline_reboot_cause(self):
     def test_find_hardware_reboot_cause(self):
         with mock.patch("determine_reboot_cause.get_reboot_cause_from_platform", return_value=("Powerloss", None)):
             result = determine_reboot_cause.find_hardware_reboot_cause()
-            assert result == "Powerloss (None)"
+            assert result == "Powerloss"
+
+    def test_find_hardware_reboot_cause_with_minor(self):
+        with mock.patch("determine_reboot_cause.get_reboot_cause_from_platform", return_value=("Powerloss", "under-voltage")):
+            result = determine_reboot_cause.find_hardware_reboot_cause()
+            assert result == "Powerloss (under-voltage)"
 
     def test_get_reboot_cause_dict_watchdog(self):
         reboot_cause_dict = determine_reboot_cause.get_reboot_cause_dict(REBOOT_CAUSE_WATCHDOG, "", GEN_TIME_WATCHDOG)