diff --git a/tests/gnmi/conftest.py b/tests/gnmi/conftest.py index 6a74fc580b9..859ba20aa6f 100644 --- a/tests/gnmi/conftest.py +++ b/tests/gnmi/conftest.py @@ -210,7 +210,13 @@ def setup_gnmi_server(duthosts, rand_one_dut_hostname, localhost, ptfhost): # Rollback configuration rollback(duthost, SETUP_ENV_CP) - recover_cert_config(duthost) + # Get the skip_gnmi_check flag from duthost options + skip_gnmi_check = duthost.host.options.get('skip_gnmi_check', False) + # Skip GNMI restart if the reboot flag is set + if not skip_gnmi_check: + recover_cert_config(duthost) + else: + logging.info("Skipping GNMI restart due to skip_gnmi_check flag") @pytest.fixture(scope="module", autouse=True) diff --git a/tests/gnmi/test_gnoi_system.py b/tests/gnmi/test_gnoi_system.py index e19b5dd1645..aa775ea9857 100644 --- a/tests/gnmi/test_gnoi_system.py +++ b/tests/gnmi/test_gnoi_system.py @@ -11,6 +11,7 @@ pytest.mark.topology('any') ] +MAX_TIME_TO_REBOOT = 300 """ This module contains tests for the gNOI System API. @@ -42,45 +43,58 @@ def test_gnoi_system_reboot(duthosts, rand_one_dut_hostname, localhost): """ duthost = duthosts[rand_one_dut_hostname] + # Set flag to indicate that this test involves reboot + duthost.host.options['skip_gnmi_check'] = True + # Trigger reboot - ret, msg = gnoi_request(duthost, localhost, "Reboot", '{"method": 1}') + ret, msg = gnoi_request(duthost, localhost, "Reboot", '{"method": 1,"delay":0,"message":"Cold Reboot"}') pytest_assert(ret == 0, "System.Reboot API reported failure (rc = {}) with message: {}".format(ret, msg)) logging.info("System.Reboot API returned msg: {}".format(msg)) - +@pytest.mark.disable_loganalyzer def test_gnoi_system_reboot_fail_invalid_method(duthosts, rand_one_dut_hostname, localhost): """ Verify the gNOI System Reboot API fails with invalid method. """ duthost = duthosts[rand_one_dut_hostname] + # Set flag to indicate that this test involves reboot + duthost.host.options['skip_gnmi_check'] = True + # Trigger reboot with invalid method ret, msg = gnoi_request(duthost, localhost, "Reboot", '{"method": 99}') pytest_assert(ret != 0, "System.Reboot API did not report failure with invalid method") - +@pytest.mark.disable_loganalyzer def test_gnoi_system_reboot_when_reboot_active(duthosts, rand_one_dut_hostname, localhost): """ Verify the gNOI System Reboot API fails if a reboot is already active. """ duthost = duthosts[rand_one_dut_hostname] + # Set flag to indicate that this test involves reboot + duthost.host.options['skip_gnmi_check'] = True + # Trigger first reboot - ret, msg = gnoi_request(duthost, localhost, "Reboot", '{"method": 1}') + ret, msg = gnoi_request(duthost, localhost, "Reboot", '{"method": 1,"delay":0,"message":"Cold Reboot"}') pytest_assert(ret == 0, "System.Reboot API reported failure (rc = {}) with message: {}".format(ret, msg)) logging.info("System.Reboot API returned msg: {}".format(msg)) # Trigger second reboot while the first one is still active - ret, msg = gnoi_request(duthost, localhost, "Reboot", '{"method": 1}') + ret, msg = gnoi_request(duthost, localhost, "Reboot", '{"method": 1,"delay":0,"message":"Cold Reboot"}') pytest_assert(ret != 0, "System.Reboot API did not report failure when reboot is already active") +@pytest.mark.disable_loganalyzer def test_gnoi_system_reboot_status_immediately(duthosts, rand_one_dut_hostname, localhost): """ Verify the gNOI System RebootStatus API returns the correct status immediately after reboot. """ duthost = duthosts[rand_one_dut_hostname] + # Set flag to indicate that this test involves reboot + duthost.host.options['skip_gnmi_check'] = True + # 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)) @@ -107,6 +121,9 @@ def gnoi_system_reboot_status_after_startup(duthosts, rand_one_dut_hostname, loc """ duthost = duthosts[rand_one_dut_hostname] + # Set flag to indicate that this test involves reboot + duthost.host.options['skip_gnmi_check'] = True + # 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)) @@ -138,12 +155,18 @@ def extract_first_json_substring(s): :return: The first JSON substring if found, otherwise None. """ - json_pattern = re.compile(r'\{.*?\}') - match = json_pattern.search(s) - if match: - try: - return json.loads(match.group()) - except json.JSONDecodeError: - logging.error("Failed to parse JSON: {}".format(match.group())) - return None - return None + start_index = s.find('{') # Find the first '{' in the string + if start_index == -1: + logging.error("No JSON found in response: {}".format(s)) + return None + json_str = s[start_index:] # Extract substring starting from '{' + try: + parsed_json = json.loads(json_str) # Attempt to parse the JSON + # Handle cases where "status": {} is empty + if "status" in parsed_json and parsed_json["status"] == {}: + logging.warning("Replacing empty 'status' field with a default value.") + parsed_json["status"] = {"unknown": "empty_status"} + return parsed_json + except json.JSONDecodeError as e: + logging.error("Failed to parse JSON: {} | Error: {}".format(json_str, e)) + return None