Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,112 @@
|
||||
"""Tests for typed block reasons + the unblock-loop breaker.
|
||||
|
||||
Covers the built-in fix for the kanban "blocked loop" — a worker blocks a
|
||||
task, a cron unblocks it, the worker re-blocks for the same reason, repeat
|
||||
forever. The fix gives ``block_task`` a typed ``kind`` and a persistent
|
||||
``block_recurrences`` counter:
|
||||
|
||||
* ``dependency`` blocks route to ``todo`` (parent-gated, auto-resumed) and
|
||||
never enter the human ``blocked`` bucket a cron would keep unblocking.
|
||||
* ``needs_input`` / ``capability`` / un-typed blocks land in ``blocked``;
|
||||
each same-cause re-block after an unblock increments ``block_recurrences``,
|
||||
and at ``BLOCK_RECURRENCE_LIMIT`` the task routes to ``triage`` for a human.
|
||||
* ``unblock_task`` deliberately does NOT reset ``block_recurrences`` (the
|
||||
amnesia that let the loop run unbounded).
|
||||
* A successful ``complete_task`` resets the loop memory.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from hermes_cli import kanban_db as kb
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def kanban_home(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> Path:
|
||||
home = tmp_path / ".hermes"
|
||||
home.mkdir()
|
||||
monkeypatch.setenv("HERMES_HOME", str(home))
|
||||
monkeypatch.setattr(Path, "home", lambda: tmp_path)
|
||||
kb.init_db()
|
||||
return home
|
||||
|
||||
|
||||
def _running_task(conn, title="t"):
|
||||
"""Create a task and drive it to ``running`` so block_task can act."""
|
||||
tid = kb.create_task(conn, title=title, assignee="worker")
|
||||
with kb.write_txn(conn):
|
||||
conn.execute("UPDATE tasks SET status='ready' WHERE id=?", (tid,))
|
||||
claimed = kb.claim_task(conn, tid, claimer="worker")
|
||||
assert claimed is not None
|
||||
return tid
|
||||
|
||||
|
||||
def _make_running_again(conn, tid):
|
||||
with kb.write_txn(conn):
|
||||
conn.execute("UPDATE tasks SET status='ready' WHERE id=?", (tid,))
|
||||
assert kb.claim_task(conn, tid, claimer="worker") is not None
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Loop breaker
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
def test_block_loop_detected_event_emitted(kanban_home: Path) -> None:
|
||||
with kb.connect_closing() as conn:
|
||||
tid = _running_task(conn)
|
||||
kb.block_task(conn, tid, reason="x", kind="capability")
|
||||
kb.unblock_task(conn, tid)
|
||||
_make_running_again(conn, tid)
|
||||
kb.block_task(conn, tid, reason="x", kind="capability")
|
||||
events = [e for e in kb.list_events(conn, tid)
|
||||
if e.kind == "block_loop_detected"]
|
||||
assert events, "expected a block_loop_detected event"
|
||||
payload = events[-1].payload or {}
|
||||
assert payload.get("recurrences") == 2
|
||||
assert payload.get("kind") == "capability"
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Dependency routing
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
def test_dependency_then_parent_done_promotes(kanban_home: Path) -> None:
|
||||
"""A dependency-parked child becomes ready once its parent completes."""
|
||||
with kb.connect_closing() as conn:
|
||||
parent = kb.create_task(conn, title="parent", assignee="worker")
|
||||
child = _running_task(conn, title="child")
|
||||
kb.link_tasks(conn, parent_id=parent, child_id=child)
|
||||
kb.block_task(conn, child, reason="wait", kind="dependency")
|
||||
assert kb.get_task(conn, child).status == "todo"
|
||||
# Finish the parent, then let recompute_ready run.
|
||||
with kb.write_txn(conn):
|
||||
conn.execute("UPDATE tasks SET status='ready' WHERE id=?", (parent,))
|
||||
kb.claim_task(conn, parent, claimer="worker")
|
||||
kb.complete_task(conn, parent, result="done")
|
||||
kb.recompute_ready(conn)
|
||||
assert kb.get_task(conn, child).status == "ready"
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Completion resets loop memory
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Validation + back-compat
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
Reference in New Issue
Block a user