Files
yao-meta-skill/tests/verify_quality_checks.py
T
2026-06-14 10:16:09 +08:00

200 lines
7.6 KiB
Python

#!/usr/bin/env python3
import json
import subprocess
import sys
from pathlib import Path
from tempfile import TemporaryDirectory
ROOT = Path(__file__).resolve().parent.parent
def run(name: str, cmd: list[str], expect_ok: bool = True, expected_substrings: list[str] | None = None) -> dict:
proc = subprocess.run(cmd, cwd=ROOT, capture_output=True, text=True)
payload = {}
if proc.stdout.strip():
try:
payload = json.loads(proc.stdout)
except json.JSONDecodeError:
payload = {"raw_stdout": proc.stdout}
joined = proc.stdout + "\n" + proc.stderr
passed = proc.returncode == 0 if expect_ok else proc.returncode == 2
if expected_substrings:
passed = passed and all(fragment in joined for fragment in expected_substrings)
return {
"name": name,
"passed": passed,
"returncode": proc.returncode,
"stdout": proc.stdout,
"stderr": proc.stderr,
"payload": payload,
}
def require_min_score(case: dict, minimum: int) -> None:
score = case.get("payload", {}).get("details", {}).get("governance_score")
case["minimum_score"] = minimum
case["observed_score"] = score
case["passed"] = case["passed"] and score is not None and score >= minimum
def require_context_targets(case: dict, max_initial: int, min_density: float) -> None:
stats = case.get("payload", {}).get("stats", {})
initial = stats.get("estimated_initial_load_tokens")
density = stats.get("quality_density")
deferred = stats.get("deferred_resource_tokens")
deferred_threshold = stats.get("deferred_resource_warn_threshold")
deferred_dirs = stats.get("deferred_resource_dirs", [])
deferred_governance = stats.get("deferred_resource_governance", {})
unused = stats.get("unused_resource_dirs", [])
case["max_initial_load"] = max_initial
case["observed_initial_load"] = initial
case["minimum_quality_density"] = min_density
case["observed_quality_density"] = density
case["observed_deferred_resource_tokens"] = deferred
case["deferred_resource_warn_threshold"] = deferred_threshold
case["deferred_resource_dir_count"] = len(deferred_dirs) if isinstance(deferred_dirs, list) else None
case["deferred_resource_governance_status"] = deferred_governance.get("status") if isinstance(deferred_governance, dict) else None
case["unused_resource_dirs"] = unused
case["passed"] = (
case["passed"]
and initial is not None
and density is not None
and deferred is not None
and deferred_threshold is not None
and isinstance(deferred_dirs, list)
and isinstance(deferred_governance, dict)
and deferred_governance.get("status") in {"governed", "not-required", "needs-review"}
and initial <= max_initial
and density >= min_density
and not unused
)
def verify_private_report_ignored() -> dict:
with TemporaryDirectory(prefix="yao-private-report-ignore-") as temp_dir:
skill_dir = Path(temp_dir) / "skill"
(skill_dir / "reports").mkdir(parents=True)
(skill_dir / "SKILL.md").write_text(
"---\n"
"name: report-ignore-fixture\n"
"description: Test fixture for ignored private reports.\n"
"---\n"
"Use this fixture to validate context sizing.\n",
encoding="utf-8",
)
(skill_dir / "reports" / "kept.md").write_text("stable evidence\n", encoding="utf-8")
baseline = run(
"private_report_ignore_baseline",
[sys.executable, "scripts/context_sizer.py", str(skill_dir), "--json"],
)
(skill_dir / "reports" / "client-research-plan.md").write_text(
"private notes\n" * 500,
encoding="utf-8",
)
context_case = run(
"private_report_ignore_context_sizer",
[sys.executable, "scripts/context_sizer.py", str(skill_dir), "--json"],
)
boundary_case = run(
"private_report_ignore_resource_boundary",
[sys.executable, "scripts/resource_boundary_check.py", str(skill_dir), "--max-initial-tokens", "5000"],
)
baseline_total = baseline.get("payload", {}).get("estimated_total_text_tokens")
context_total = context_case.get("payload", {}).get("estimated_total_text_tokens")
boundary_files = boundary_case.get("payload", {}).get("stats", {}).get("relevant_file_count")
passed = (
baseline["passed"]
and context_case["passed"]
and boundary_case["passed"]
and baseline_total == context_total
and boundary_files == 2
)
return {
"name": "private_report_ignore",
"passed": passed,
"baseline_total_tokens": baseline_total,
"observed_total_tokens": context_total,
"resource_relevant_file_count": boundary_files,
"context_stdout": context_case["stdout"],
"boundary_stdout": boundary_case["stdout"],
}
def main() -> None:
python = sys.executable
cases = []
root_governance = run(
"root_governance",
[python, "scripts/governance_check.py", str(ROOT), "--require-manifest"],
)
require_min_score(root_governance, 90)
cases.append(root_governance)
root_resource = run(
"root_resource_boundaries",
[python, "scripts/resource_boundary_check.py", str(ROOT)],
)
require_context_targets(root_resource, 1000, 100.0)
root_payload = root_resource.get("payload", {})
root_governance_status = root_resource.get("deferred_resource_governance_status")
root_resource["passed"] = (
root_resource["passed"]
and root_governance_status == "governed"
and not any("Deferred resource footprint is high" in item for item in root_payload.get("warnings", []))
)
cases.append(root_resource)
complex_resource = run(
"complex_example_resource_boundaries",
[python, "scripts/resource_boundary_check.py", str(ROOT / "examples" / "complex-release-orchestrator" / "generated-skill")],
)
require_context_targets(complex_resource, 1000, 120.0)
governed_resource = run(
"governed_example_resource_boundaries",
[python, "scripts/resource_boundary_check.py", str(ROOT / "examples" / "governed-incident-command" / "generated-skill")],
)
require_context_targets(governed_resource, 1000, 120.0)
cases.extend([
run(
"complex_example_governance",
[python, "scripts/governance_check.py", str(ROOT / "examples" / "complex-release-orchestrator" / "generated-skill"), "--require-manifest"],
),
complex_resource,
run(
"invalid_governance_manifest",
[python, "scripts/governance_check.py", str(ROOT / "tests" / "fixtures" / "governance_invalid_manifest"), "--require-manifest"],
expect_ok=False,
expected_substrings=[
"Missing manifest fields",
"Invalid status",
"updated_at must use YYYY-MM-DD",
"manifest name does not match",
],
),
])
governed_example = run(
"governed_example_governance",
[python, "scripts/governance_check.py", str(ROOT / "examples" / "governed-incident-command" / "generated-skill"), "--require-manifest"],
)
require_min_score(governed_example, 90)
cases.insert(4, governed_example)
cases.insert(5, governed_resource)
cases.append(verify_private_report_ignored())
report = {"ok": all(case["passed"] for case in cases), "cases": cases}
print(json.dumps(report, ensure_ascii=False, indent=2))
if not report["ok"]:
raise SystemExit(2)
if __name__ == "__main__":
main()