diff --git a/tests/test_cli.py b/tests/test_cli.py index 1aad2183..14082564 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -18,7 +18,7 @@ from safety import cli from safety.models import CVE, SafetyRequirement, Severity, Vulnerability -from safety.util import Package, SafetyContext, get_safety_version +from safety.util import Package, SafetyContext from safety.auth.models import Auth from safety_schemas.models.base import AuthenticationType @@ -533,7 +533,7 @@ def test_debug_flag(self, mock_get_auth_info, mock_is_valid, mock_get_auth_type, assert result.exit_code == 0, ( f"CLI exited with code {result.exit_code} and output: {result.output} and error: {result.stderr}" ) - expected_output_snippet = f"{get_safety_version()} scanning" + expected_output_snippet = "Safety 3.2.8 scanning" assert expected_output_snippet in result.output, ( f"Expected output to contain: {expected_output_snippet}, but got: {result.output}" )