chore: initialize project repository

This commit is contained in:
Codex
2026-09-09 21:55:37 +08:00
commit b1497fd2c6
167 changed files with 17690 additions and 0 deletions
@@ -0,0 +1,32 @@
import pytest
from app.core.errors import RecoverableAgentError
from app.worker.agent_run_worker import AgentRunWorker
@pytest.mark.asyncio
async def test_worker_translates_recoverable_failure() -> None:
class FakeRepository:
class Session:
def begin(self):
class Context:
async def __aenter__(self): return self
async def __aexit__(self, *args): return False
return Context()
session = Session()
async def claim(self, run_id, worker_id, lease_seconds):
return True
async def record_failure(self, run_id, error_code, retry_limit):
return True
worker = object.__new__(AgentRunWorker)
worker.repository = FakeRepository()
worker.worker_id = "worker-1"
worker.lease_seconds = 60
worker.retry_limit = 3
async def execute(_: str) -> None:
raise RecoverableAgentError("temporary")
worker.execute_run = execute
assert await worker.claim_and_execute("run-1") is False