rune/tests/test_operator_probe_isolation.py

"""Verification probes have a task budget, not a global chat lifetime budget."""
from pathlib import Path
from types import SimpleNamespace
import hashlib

from rune import evidence_probe, rune_operator


def test_probe_uses_distinct_task_directories(tmp_path, monkeypatch):
    monkeypatch.setenv("MIMO_PROBE_ENABLED", "1")
    operator = object.__new__(rune_operator.RuneOperator)
    operator.storage_path = str(tmp_path / "operator.json")
    called = []

    def probe(_question, artifact_dir, model):
        called.append(Path(artifact_dir))
        return SimpleNamespace(status="GROUNDED")

    # The operator binds run_probe at module level (_run_probe) — the real
    # interception seam is the operator module's binding, not evidence_probe's.
    monkeypatch.setattr(rune_operator, "_run_probe", probe)
    monkeypatch.setattr(operator, "_route_task_class", lambda *_: None)
    for task_id in ("task-a", "task-b"):
        task = SimpleNamespace(task_id=task_id, description="verify")
        assert operator._probe_and_escalate(task, SimpleNamespace(error="failure"), "chat") is False
    assert called == [tmp_path / "probes" / hashlib.sha256(t.encode()).hexdigest()
                      for t in ("task-a", "task-b")]


def test_exhausted_task_budget_never_escapes_into_chat(tmp_path, monkeypatch):
    monkeypatch.setenv("MIMO_PROBE_ENABLED", "1")
    operator = object.__new__(rune_operator.RuneOperator)
    operator.storage_path = str(tmp_path / "operator.json")

    def exhausted(*_args, **_kwargs):
        raise evidence_probe.ProbeBudgetExceeded("probe budget exhausted (2)")

    monkeypatch.setattr(rune_operator, "_run_probe", exhausted)
    task = SimpleNamespace(task_id="task-a", description="verify")
    assert operator._probe_and_escalate(task, SimpleNamespace(error="failure"), "chat") is False