a2a-source-edit: write tests/test_agent.py
This commit is contained in:
@@ -124,13 +124,15 @@ def test_process_email_proposes_refund_and_dedupes(monkeypatch: Any) -> None:
|
|||||||
assert "APPROVE-REFUND-" in first["approval_instructions"]
|
assert "APPROVE-REFUND-" in first["approval_instructions"]
|
||||||
assert second["status"] == "duplicate_email"
|
assert second["status"] == "duplicate_email"
|
||||||
assert second["previous_action_digest"] == first["action_digest"]
|
assert second["previous_action_digest"] == first["action_digest"]
|
||||||
assert all("sk_test_x" not in str(call) for call in MockAsyncClient.calls)
|
assert "sk_test_x" not in str(first)
|
||||||
|
assert "xoxb-test" not in str(first)
|
||||||
|
|
||||||
asyncio.run(run())
|
asyncio.run(run())
|
||||||
|
|
||||||
|
|
||||||
def test_execute_requires_digest_bound_approval(monkeypatch: Any) -> None:
|
def test_execute_requires_digest_bound_approval(monkeypatch: Any) -> None:
|
||||||
async def run() -> None:
|
async def run() -> None:
|
||||||
|
MockAsyncClient.calls = []
|
||||||
monkeypatch.setattr("agent.httpx.AsyncClient", MockAsyncClient)
|
monkeypatch.setattr("agent.httpx.AsyncClient", MockAsyncClient)
|
||||||
agent = ReceivesSupportEmailChecks1919313()
|
agent = ReceivesSupportEmailChecks1919313()
|
||||||
ctx = make_ctx()
|
ctx = make_ctx()
|
||||||
@@ -169,6 +171,8 @@ def test_execute_requires_digest_bound_approval(monkeypatch: Any) -> None:
|
|||||||
assert approved["idempotency_key"].startswith("a2a-refund-")
|
assert approved["idempotency_key"].startswith("a2a-refund-")
|
||||||
refund_call = next(call for call in MockAsyncClient.calls if call["url"].endswith("/v1/refunds"))
|
refund_call = next(call for call in MockAsyncClient.calls if call["url"].endswith("/v1/refunds"))
|
||||||
assert refund_call["headers"]["Idempotency-Key"] == approved["idempotency_key"]
|
assert refund_call["headers"]["Idempotency-Key"] == approved["idempotency_key"]
|
||||||
|
assert "sk_test_x" not in str(approved)
|
||||||
|
assert "xoxb-test" not in str(approved)
|
||||||
|
|
||||||
asyncio.run(run())
|
asyncio.run(run())
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user