[Smartswitch] Update the type of reboot status variable for DPU (#311)

Author: gpunathilell

host_modules/reboot.py

@@ -61,7 +61,11 @@ def populate_reboot_status_flag(self, active = False, when = 0, reason = "", met
         self.reboot_status_flag["reason"] = reason
         self.reboot_status_flag["count"] = self.reboot_count
         self.reboot_status_flag["method"] = method
-        self.reboot_status_flag["status"] = status.value
+        # status value should be of type RebootStatus
+        self.reboot_status_flag["status"] = {
+            "status": status.value,
+            "message": ""
+        }
         self.lock.release()
         return
 

tests/host_modules/reboot_test.py

@@ -73,15 +73,21 @@ def test_populate_reboot_status_flag(self):
             assert get_reboot_status_flag_data["reason"] == ""
             assert get_reboot_status_flag_data["count"] == 0
             assert get_reboot_status_flag_data["method"] == ""
-            assert get_reboot_status_flag_data["status"] == RebootStatus.STATUS_UNKNOWN.value
+            assert get_reboot_status_flag_data["status"] == {
+                "status": RebootStatus.STATUS_UNKNOWN.value,
+                "message": ""
+            }
 
     def test_populate_reboot_status_flag_with_status(self):
         with mock.patch("time.time", return_value=1617811205.25):
             self.reboot_module.populate_reboot_status_flag(status=RebootStatus.STATUS_SUCCESS)
             return_value, get_reboot_status_flag_data = self.reboot_module.get_reboot_status()
             assert return_value == 0
             get_reboot_status_flag_data = json.loads(get_reboot_status_flag_data)
-            assert get_reboot_status_flag_data["status"] == RebootStatus.STATUS_SUCCESS.value
+            assert get_reboot_status_flag_data["status"] == {
+                "status": RebootStatus.STATUS_SUCCESS.value,
+                "message": ""
+            }
 
     def test_validate_reboot_request_success_cold_boot_enum_method(self):
         reboot_request = {"method": REBOOT_METHOD_COLD_BOOT_ENUM, "reason": "test reboot request reason"}
@@ -391,7 +397,10 @@ def test_get_reboot_status_active(self):
         assert response_data["when"] == TIME
         assert response_data["reason"] == MSG
         assert response_data["method"] == REBOOT_METHOD_COLD_BOOT_ENUM
-        assert response_data["status"] == RebootStatus.STATUS_SUCCESS.value
+        assert response_data["status"] == {
+            "status": RebootStatus.STATUS_SUCCESS.value,
+            "message": ""
+        }
 
     def test_get_reboot_status_inactive(self):
         self.reboot_module.populate_reboot_status_flag(False, 0, "", REBOOT_METHOD_COLD_BOOT_ENUM, RebootStatus.STATUS_SUCCESS)
@@ -402,7 +411,10 @@ def test_get_reboot_status_inactive(self):
         assert response_data["when"] == 0
         assert response_data["reason"] == ""
         assert response_data["method"] == REBOOT_METHOD_COLD_BOOT_ENUM
-        assert response_data["status"] == RebootStatus.STATUS_SUCCESS.value
+        assert response_data["status"] == {
+            "status": RebootStatus.STATUS_SUCCESS.value,
+            "message": ""
+        }
 
 #        assert result[1] == TEST_INACTIVE_RESPONSE_DATA