732642856
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Keep README capability badges aligned with the current registry."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import re
|
|
from collections import Counter
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
README = ROOT / "README.md"
|
|
REGISTRY = ROOT / "references" / "technique_registry.json"
|
|
|
|
|
|
def _readme_badge_value(name: str) -> int:
|
|
text = README.read_text(encoding="utf-8")
|
|
match = re.search(rf"\[!\[{re.escape(name)}\]\(https://img\.shields\.io/badge/{name.lower()}-(\d+)-", text)
|
|
assert match, f"Missing README badge for {name}"
|
|
return int(match.group(1))
|
|
|
|
|
|
def test_readme_badges_match_technique_registry_counts() -> None:
|
|
registry = json.loads(REGISTRY.read_text(encoding="utf-8"))
|
|
techniques = registry["techniques"].values()
|
|
counts = Counter(item["status"] for item in techniques)
|
|
total = len(registry["techniques"])
|
|
|
|
assert _readme_badge_value("Capabilities") == total
|
|
assert _readme_badge_value("Covered") == counts["covered"]
|
|
assert _readme_badge_value("Complete") == counts["complete"]
|
|
assert _readme_badge_value("Partial") == counts["partial"]
|