mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
feat(sandbox): on-demand provisioning via request_sandbox verb (#338)
* feat(sandbox): on-demand request_sandbox verb replaces eager provisioning Sandboxes were provisioned at every agent spawn for opted-in projects, so every role paid the sidecar spin-up and a provisioning failure refused the spawn. Provisioning now happens when an agent asks: the request_sandbox do-verb (dev + QA) reaches the orchestrator through ContentActionsDeps, ensure_sandbox provisions idempotently with an in-memory per-agent cache (evicted at teardown and janitor sweep), and creds return in the envelope payload including ready-to-export ROBOCO_TEST_* values. Spawn now only injects a marker env naming the available services plus a briefing line; sandbox failures can no longer refuse a spawn. Teardown lifecycle unchanged. * feat(sandbox): harden request_sandbox + Phase 3 wiring proof and docs Hardening from adversarial review: ensure_sandbox now provisions the project's full opted-in set on first request (a later superset can never tear down a live sandbox mid-use), serializes per-agent behind an asyncio lock (a client timeout-retry no longer races its own in-flight provision), and verifies container liveness on every cache hit (a dead sandbox evicts and re-provisions instead of serving dead creds). MCP client budget 720->1080s for the full-set cold case. Phase 3: e2e smoke wiring test (manifest grants + guard-chain envelopes over the real API), sandbox-db/tools/map docs and CLAUDE.md rewritten for on-demand. * feat(sandbox): release sandboxes when the agent's work ends CEO directive: sidecars must not dangle once the agent is done. The six work-ending verbs (i_am_done, unclaim, i_am_idle, pass_review, fail_review, i_documented) now release the caller's sandbox best-effort on their success path via release_sandbox (lock + teardown + cache evict; a no-sandbox agent costs a dict lookup). Container removal and the janitor remain the backstop; a re-request provisions fresh. * test(sandbox): monkeypatch the release hook instead of method assignment mypy method-assign rejected the direct AsyncMock assignments; the prior static gate ran before this test file landed. * test(sandbox): guard envelope evidence for mypy in verb tests * chore(prompts): regenerate verb tables for request_sandbox * chore: resolve merge with master (breadcrumbs + statement budget) --------- Co-authored-by: Renn F <rennf93@users.noreply.github.com>
This commit is contained in:
@@ -118,3 +118,98 @@ async def test_sandbox_janitor_sweep_swallows_errors(
|
||||
sandbox.janitor_sweep.side_effect = RuntimeError("boom")
|
||||
|
||||
await orch._sandbox_janitor_sweep() # must not raise
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# ensure_sandbox cache eviction (request_sandbox on-demand provisioning)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_remove_container_evicts_ensure_sandbox_cache(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
monkeypatch.setattr(settings, "sandbox_db_enabled", True)
|
||||
monkeypatch.setattr(asyncio, "create_subprocess_exec", _fake_create_subprocess_exec)
|
||||
orch, _sandbox = _make_orchestrator()
|
||||
orch._sandbox_info = {"dev-1": MagicMock(), "dev-2": MagicMock()}
|
||||
|
||||
await orch._remove_container("roboco-agent-dev-1")
|
||||
|
||||
assert "dev-1" not in orch._sandbox_info
|
||||
assert "dev-2" in orch._sandbox_info
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_remove_container_teardown_false_spares_cache(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
monkeypatch.setattr(settings, "sandbox_db_enabled", True)
|
||||
monkeypatch.setattr(asyncio, "create_subprocess_exec", _fake_create_subprocess_exec)
|
||||
orch, _sandbox = _make_orchestrator()
|
||||
orch._sandbox_info = {"dev-1": MagicMock()}
|
||||
|
||||
await orch._remove_container("roboco-agent-dev-1", teardown_sandbox=False)
|
||||
|
||||
assert "dev-1" in orch._sandbox_info
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_janitor_sweep_evicts_cache_for_reaped_agents(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
monkeypatch.setattr(settings, "sandbox_db_enabled", True)
|
||||
orch, _sandbox = _make_orchestrator()
|
||||
orch._sandbox_info = {"dev-1": MagicMock(), "dev-2": MagicMock()}
|
||||
orch._instances = {"dev-2": MagicMock()} # dev-1's agent instance is gone
|
||||
|
||||
await orch._sandbox_janitor_sweep()
|
||||
|
||||
assert "dev-1" not in orch._sandbox_info
|
||||
assert "dev-2" in orch._sandbox_info
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# release_sandbox (end-of-engagement teardown, called by the Choreographer's
|
||||
# post-verb hook — i_am_done / unclaim / i_am_idle / pass_review / fail_review
|
||||
# / i_documented — instead of only at container removal)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_release_sandbox_no_cache_entry_is_fast_noop() -> None:
|
||||
"""The overwhelmingly common case: no sandbox for this agent. Must not
|
||||
take the per-agent lock or call docker — the cache dict check alone
|
||||
decides, before any lock is even allocated."""
|
||||
orch, sandbox = _make_orchestrator()
|
||||
orch._sandbox_info = {}
|
||||
|
||||
await orch.release_sandbox("dev-1")
|
||||
|
||||
sandbox.teardown.assert_not_called()
|
||||
assert not hasattr(orch, "_sandbox_locks")
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_release_sandbox_tears_down_and_evicts_cache() -> None:
|
||||
orch, sandbox = _make_orchestrator()
|
||||
orch._sandbox_info = {"dev-1": MagicMock(), "dev-2": MagicMock()}
|
||||
|
||||
await orch.release_sandbox("dev-1")
|
||||
|
||||
sandbox.teardown.assert_awaited_once_with("dev-1")
|
||||
assert "dev-1" not in orch._sandbox_info
|
||||
assert "dev-2" in orch._sandbox_info
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_release_sandbox_is_idempotent() -> None:
|
||||
"""A second release for the same slug (e.g. unclaim right after
|
||||
i_am_idle) is a no-op — the first call already evicted the cache."""
|
||||
orch, sandbox = _make_orchestrator()
|
||||
orch._sandbox_info = {"dev-1": MagicMock()}
|
||||
|
||||
await orch.release_sandbox("dev-1")
|
||||
await orch.release_sandbox("dev-1")
|
||||
|
||||
sandbox.teardown.assert_awaited_once_with("dev-1")
|
||||
|
||||
Reference in New Issue
Block a user