张胜宇
e239eb778b
1) 客服 Agent 四份交付文档 + 构建脚手架：品牌由包装占位 XX科技 / 旧名 南方财富 统一为南方基金（热线 400-889-8899 / 官网 nffund.com），系统名改为「智能服务系统」； 同步追加 §0.4 修订记录行，工程记录行保留原占位字面以支撑硬编码扫描验收。 2) 开发文档：清理 28 份已作废/残留文档（14 份移出归档 + 14 份仓库副本）， 新增《文档规整方案与开发前待决事项-2026-09-17》。 3) 客服agent 四份交付文档首次纳入本分支。
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
import pytest
|
|
|
|
from app.core.errors import AgentError, RecoverableAgentError
|
|
from app.worker.agent_run_worker import AgentRunWorker
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_worker_retries_recoverable_error() -> None:
|
|
class FakeSession:
|
|
def begin(self):
|
|
class Context:
|
|
async def __aenter__(self): return self
|
|
async def __aexit__(self, *args): return False
|
|
return Context()
|
|
|
|
class FakeRepository:
|
|
session = FakeSession()
|
|
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
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_worker_stops_non_recoverable_error() -> None:
|
|
class FakeSession:
|
|
def begin(self):
|
|
class Context:
|
|
async def __aenter__(self): return self
|
|
async def __aexit__(self, *args): return False
|
|
return Context()
|
|
|
|
class FakeRepository:
|
|
session = FakeSession()
|
|
async def claim(self, run_id, worker_id, lease_seconds): return True
|
|
async def record_failure(self, run_id, error_code, retry_limit): return False
|
|
|
|
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 AgentError("fatal")
|
|
worker.execute_run = execute
|
|
assert await worker.claim_and_execute("run-1") is False
|