proof-aware-crypto-tooling-agent/tests/test_web_and_witness.py

import json
import threading
import urllib.request
from pathlib import Path

from pacta.signing import generate_ed25519_keypair
from pacta.transparency import leaf_bytes_for_attestation, verify_inclusion
from pacta.witness import audit_published_log
from pacta_provider.transparency_log import TransparencyLog
from pacta_provider.web import serve


def _make_log(tmp_path, n=3):
    generate_ed25519_keypair(tmp_path / "k.key", tmp_path / "k.pub")
    log = TransparencyLog(tmp_path / "log")
    log.init("test-provider", tmp_path / "k.pub")
    from pacta.yamlio import dump_data

    for i in range(n):
        att = {
            "schema_version": 1,
            "provider": "test-provider",
            "issued_at": "2026-07-07T00:00:00Z",
            "subject": {"component": f"component-{i}", "repo_commit": f"commit-{i}"},
            "certificates": [{"name": "T.cert", "status": "proven", "axiom_status": "clean"}],
        }
        dump_data(att, tmp_path / f"a{i}.yaml")
        log.append_attestation(tmp_path / f"a{i}.yaml", tmp_path / "k.key", tmp_path / "k.pub")
    return log


def test_web_endpoints_and_online_proof_roundtrip(tmp_path):
    # root mount: the production shape (ltl.zkdefi.org serves from /)
    _make_log(tmp_path)
    server = serve(str(tmp_path / "log"), port=0)
    port = server.server_address[1]
    threading.Thread(target=server.serve_forever, daemon=True).start()
    base = f"http://127.0.0.1:{port}"
    try:
        def get(path):
            with urllib.request.urlopen(base + path, timeout=10) as r:
                return json.loads(r.read())

        assert get("/healthz")["tree_size"] == 3
        sth = get("/v1/sth")
        assert sth["tree_size"] == 3
        att = get("/v1/attestation?component=component-1")["attestation"]
        proof = get("/v1/proof?component=component-1")
        ok = verify_inclusion(
            leaf_bytes_for_attestation(att), proof["leaf_index"], proof["tree_size"],
            [bytes.fromhex(h) for h in proof["inclusion_proof"]],
            bytes.fromhex(proof["sth"]["root_hash"]),
        )
        assert ok
        consistency = get("/v1/sth-consistency?first=2")
        assert consistency["from_tree_size"] == 2 and consistency["proof"]
        history = get("/v1/sth-history")["sth_history"]
        assert len(history) == 3  # one head per append
        with urllib.request.urlopen(base + "/paper", timeout=10) as r:
            assert r.headers["Content-Type"] == "application/pdf"
            assert r.read(5) == b"%PDF-"
    finally:
        server.shutdown()


def test_logclient_fetch_and_refresh_pin(tmp_path):
    # path mount still supported for anyone proxying under a prefix
    _make_log(tmp_path)
    server = serve(str(tmp_path / "log"), base_path="lean-transparency-log", port=0)
    port = server.server_address[1]
    threading.Thread(target=server.serve_forever, daemon=True).start()
    base = f"http://127.0.0.1:{port}/lean-transparency-log"
    try:
        from pacta.logclient import fetch_evidence, refresh_pin

        paths = fetch_evidence(base, "component-2", tmp_path / "fetched")
        assert paths["attestation"].exists() and paths["receipt"].exists()
        ok, diagnostics = refresh_pin(base, tmp_path / "pins.json", tmp_path / "k.pub")
        assert ok, diagnostics
        # second refresh: matched, still ok
        ok, _ = refresh_pin(base, tmp_path / "pins.json", tmp_path / "k.pub")
        assert ok
    finally:
        server.shutdown()


def test_publish_and_witness_audit_catches_tampering(tmp_path):
    log = _make_log(tmp_path)
    published = tmp_path / "published"
    report = log.publish(published, public_key_path=tmp_path / "k.pub")
    assert report["entries"] == 3
    assert (published / "verify.py").exists() and (published / "README.md").exists()

    clean = audit_published_log(published, tmp_path / "k.pub")
    assert clean.ok and clean.heads_checked == 3

    # tamper one entry: structural audit must fail loudly
    victim = published / "entries" / "000001.json"
    record = json.loads(victim.read_text())
    record["leaf"]["attestation"]["subject"]["repo_commit"] = "EVIL"
    victim.write_text(json.dumps(record))
    dirty = audit_published_log(published, tmp_path / "k.pub")
    assert not dirty.ok
    assert any("leaf_hash" in problem for problem in dirty.problems)
    assert any("EQUIVOCATION or tampered" in problem for problem in dirty.problems)


def test_standalone_verify_py_runs(tmp_path):
    import subprocess
    import sys

    log = _make_log(tmp_path)
    published = tmp_path / "published"
    log.publish(published, public_key_path=tmp_path / "k.pub")
    result = subprocess.run([sys.executable, "verify.py", "--all"], cwd=published, capture_output=True, text=True)
    assert result.returncode == 0, result.stdout + result.stderr
    assert "OK - the log is internally consistent" in result.stdout