fix(security): update tests for verdict and --force changes

This commit is contained in:
sprmn24 2026-03-27 00:27:30 +03:00 committed by Teknium
parent 0f8215f633
commit 789043b691
2 changed files with 11 additions and 11 deletions

View file

@ -84,13 +84,13 @@ class TestDetermineVerdict:
f = Finding("x", "high", "network", "f.py", 1, "m", "d")
assert _determine_verdict([f]) == "caution"
def test_medium_finding_caution(self):
def test_medium_finding_safe(self):
f = Finding("x", "medium", "structural", "f.py", 1, "m", "d")
assert _determine_verdict([f]) == "caution"
assert _determine_verdict([f]) == "safe"
def test_low_finding_caution(self):
def test_low_finding_safe(self):
f = Finding("x", "low", "obfuscation", "f.py", 1, "m", "d")
assert _determine_verdict([f]) == "caution"
assert _determine_verdict([f]) == "safe"
# ---------------------------------------------------------------------------
@ -145,21 +145,21 @@ class TestShouldAllowInstall:
allowed, _ = should_allow_install(self._result("community", "dangerous", f), force=False)
assert allowed is False
def test_force_overrides_dangerous_for_community(self):
def test_force_does_not_override_dangerous_for_community(self):
f = [Finding("x", "critical", "c", "f", 1, "m", "d")]
allowed, reason = should_allow_install(
self._result("community", "dangerous", f), force=True
)
assert allowed is True
assert "Force-installed" in reason
assert allowed is False
assert "Blocked" in reason
def test_force_overrides_dangerous_for_trusted(self):
def test_force_does_not_override_dangerous_for_trusted(self):
f = [Finding("x", "critical", "c", "f", 1, "m", "d")]
allowed, reason = should_allow_install(
self._result("trusted", "dangerous", f), force=True
)
assert allowed is True
assert "Force-installed" in reason
assert allowed is False
assert "Blocked" in reason
# -- agent-created policy --

View file

@ -661,7 +661,7 @@ def should_allow_install(result: ScanResult, force: bool = False) -> Tuple[bool,
if decision == "allow":
return True, f"Allowed ({result.trust_level} source, {result.verdict} verdict)"
if force and result.verdict != "dangerous":
if force and not (result.verdict == "dangerous" and result.trust_level in ("community", "trusted")):
return True, (
f"Force-installed despite {result.verdict} verdict "
f"({len(result.findings)} findings)"