removed error

sonic-net · Dec 31, 2024 · bd2e854 · bd2e854
1 parent 3a73d97
commit bd2e854
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_config_stp.py b/tests/test_config_stp.py
@@ -64,8 +64,9 @@ def test_is_valid_root_guard_timeout():
         pytest.fail("Unexpected failure on valid root guard timeout")
 
     # Invalid case
-    with pytest.raises(SystemExit):
-        is_valid_root_guard_timeout(mock_ctx, 700)
+    mock_ctx.fail = MagicMock()  # Mocking the fail method to prevent actual exit
+    is_valid_root_guard_timeout(mock_ctx, 700)
+    mock_ctx.fail.assert_called_once_with("STP root guard timeout must be in range 5-600")
 
 
 def test_is_valid_forward_delay():