22 lines
477 B
Python
22 lines
477 B
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from a2a_pack import LocalRunContext, NoAuth
|
|
|
|
from agent import SelfHealingDemo
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_controlled_failure_has_a_safe_result() -> None:
|
|
agent = SelfHealingDemo()
|
|
ctx = LocalRunContext(auth=NoAuth())
|
|
|
|
result = await agent.run(ctx, scenario="controlled-failure")
|
|
|
|
assert result == {
|
|
"ok": True,
|
|
"scenario": "controlled-failure",
|
|
"result": "recovered",
|
|
}
|