14452bf88a
Wait on the watchdog and delegation transition events instead of racing wall-clock polling bounds. Keep the stalling record alive until the assertion owns its teardown, and resolve clarify text replies directly from the fallback send boundary so scheduler load cannot turn a delivered prompt into a timeout.
83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
"""Gateway /agents surfaces background delegations with live activity (#51690).
|
|
|
|
Drives the REAL GatewayRunner._handle_agents_command against a REAL
|
|
async-delegation registry dispatch (no mocked list function), so the test
|
|
covers the whole projection: registry record → list_async_delegations()
|
|
live sampling → /agents rendering.
|
|
"""
|
|
|
|
import threading
|
|
import time
|
|
|
|
import pytest
|
|
|
|
from tools import async_delegation as ad
|
|
from tools.process_registry import process_registry
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _clean_state():
|
|
ad._reset_for_tests()
|
|
while not process_registry.completion_queue.empty():
|
|
process_registry.completion_queue.get_nowait()
|
|
yield
|
|
deadline = time.monotonic() + 2.0
|
|
while ad.active_count() and time.monotonic() < deadline:
|
|
time.sleep(0.02)
|
|
ad._reset_for_tests()
|
|
while not process_registry.completion_queue.empty():
|
|
process_registry.completion_queue.get_nowait()
|
|
|
|
|
|
def _make_runner():
|
|
from gateway.run import GatewayRunner
|
|
|
|
runner = object.__new__(GatewayRunner)
|
|
runner._running_agents = {}
|
|
runner._running_agents_ts = {}
|
|
runner._background_tasks = set()
|
|
runner._session_key_for_source = lambda source: "agent:main:test:dm:1"
|
|
return runner
|
|
|
|
|
|
class _Event:
|
|
source = None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_agents_command_marks_stalling_delegation(monkeypatch):
|
|
monkeypatch.setattr(ad, "_STALE_CHECK_INTERVAL", 0.03)
|
|
monkeypatch.setattr(ad, "_STALE_IDLE_SECONDS", 0.1)
|
|
# Force-finalization is a separate contract. Disable it so the stalling
|
|
# projection stays observable until this test releases the worker.
|
|
monkeypatch.setattr(ad, "_STALL_GRACE_SECONDS", float("inf"))
|
|
gate = threading.Event()
|
|
stalling = threading.Event()
|
|
|
|
def blocked_runner():
|
|
gate.wait()
|
|
return {}
|
|
|
|
res = ad.dispatch_async_delegation(
|
|
goal="wedged child", context=None, toolsets=None, role="leaf",
|
|
model="m", session_key="agent:main:test:dm:1", max_async_children=1,
|
|
runner=blocked_runner,
|
|
interrupt_fn=stalling.set,
|
|
progress_fn=lambda: ((0, None), False),
|
|
)
|
|
assert res["status"] == "dispatched"
|
|
|
|
try:
|
|
assert stalling.wait(30.0), "delegation never reached stalling state"
|
|
|
|
runner = _make_runner()
|
|
out = await runner._handle_agents_command(_Event())
|
|
finally:
|
|
gate.set()
|
|
|
|
assert res["delegation_id"] in out
|
|
assert "stalling" in out
|
|
assert "no progress" in out
|
|
|
|
|