Initial commit

tests/test_findings_breakdown.py

@@ -0,0 +1,48 @@
+"""Severity breakdown helper for console output."""
+
+from __future__ import annotations
+
+from applepy.findings import Finding, Severity, severity_breakdown_line
+
+
+def test_severity_breakdown_line_all_levels() -> None:
+    findings = [
+        Finding(
+            id="1",
+            title="a",
+            category="c",
+            severity=Severity.CRITICAL,
+            description="d",
+            evidence="e",
+            worksheet="W",
+        ),
+        Finding(
+            id="2",
+            title="b",
+            category="c",
+            severity=Severity.HIGH,
+            description="d",
+            evidence="e",
+            worksheet="W",
+        ),
+        Finding(
+            id="3",
+            title="c",
+            category="c",
+            severity=Severity.INFORMATIONAL,
+            description="d",
+            evidence="e",
+            worksheet="W",
+        ),
+    ]
+    line = severity_breakdown_line(findings)
+    assert "critical: 1" in line
+    assert "high: 1" in line
+    assert "medium: 0" in line
+    assert "low: 0" in line
+    assert "informational: 1" in line
+
+
+def test_severity_breakdown_line_empty() -> None:
+    line = severity_breakdown_line([])
+    assert line == "critical: 0 · high: 0 · medium: 0 · low: 0 · informational: 0"