gnoi_system_reboot_status_after_startup

Author: hdwhdw

tests/gnmi/test_gnoi_system.py

@@ -4,6 +4,7 @@
 
 from .helper import gnoi_request
 from tests.common.helpers.assertions import pytest_assert
+from tests.common.reboot import wait_for_startup
 import re
 
 pytestmark = [
@@ -84,6 +85,35 @@ def test_gnoi_system_reboot_status_immediately(duthosts, rand_one_dut_hostname,
     pytest_assert(msg_json["active"] is True, "System.RebootStatus API did not return active = true")
 
 
+def gnoi_system_reboot_status_after_startup(duthosts, rand_one_dut_hostname, localhost):
+    """
+    Verify the gNOI System RebootStatus API returns the correct status after the device has started up.
+    """
+    duthost = duthosts[rand_one_dut_hostname]
+
+    # Trigger reboot
+    ret, msg = gnoi_request(duthost, localhost, "Reboot", '{"method": 1, "message": "test"}')
+    pytest_assert(ret == 0, "System.Reboot API reported failure (rc = {}) with message: {}".format(ret, msg))
+    logging.info("System.Reboot API returned msg: {}".format(msg))
+
+    # Wait for device to come back online
+    wait_for_startup(duthost)
+
+    # Get reboot status
+    ret, msg = gnoi_request(duthost, localhost, "RebootStatus", "")
+    pytest_assert(ret == 0, "System.RebootStatus API reported failure (rc = {}) with message: {}".format(ret, msg))
+    logging.info("System.RebootStatus API returned msg: {}".format(msg))
+    # Message should contain a json substring like this
+    # {"active":false,"wait":0,"when":0,"reason":"test","count":1,"method":1,"status":1}
+    # Extract JSON part from the message
+    msg_json = extract_first_json_substring(msg)
+    if not msg_json:
+        pytest.fail("Failed to extract JSON from System.RebootStatus API response")
+    logging.info("Extracted JSON: {}".format(msg_json))
+    pytest_assert("active" in msg_json, "System.RebootStatus API did not return active")
+    pytest_assert(msg_json["active"] is False, "System.RebootStatus API did not return active = false")
+
+
 def extract_first_json_substring(s):
     """
     Extract the first JSON substring from a given string.