33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
|
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
|