chore: initialize project repository
This commit is contained in:
@@ -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
|
||||
Reference in New Issue
Block a user