From b2bb9fe909856b732d3a02cb99d2398e22bf2827 Mon Sep 17 00:00:00 2001 From: a2a-cloud Date: Mon, 13 Jul 2026 12:19:15 +0000 Subject: [PATCH] a2a-source-edit: write tests/test_agent.py --- tests/test_agent.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/test_agent.py b/tests/test_agent.py index 4aa2abe..ce65e6d 100644 --- a/tests/test_agent.py +++ b/tests/test_agent.py @@ -3,7 +3,7 @@ import pytest from a2a_pack import NoAuth from a2a_pack.context import LocalRunContext -from agent import CustomerIntegrationEngineer +from agent import CustomerIntegrationEngineer, SmokeCallInput @pytest.mark.asyncio @@ -31,14 +31,14 @@ async def test_smoke_call_rejects_secret_like_result(): result = await agent.validate_smoke_call( ctx, - { - "skill_name": "validate_live_card", - "input_args": {"bounded": True}, - "result": {"ok": True, "api_key": "sk-testsecret1234567890"}, - "status_code": 200, - "used_workspace_delegation": True, - "delegated_workspace_mode": "read_only", - }, + SmokeCallInput( + skill_name="validate_live_card", + input_args={"bounded": True}, + result={"ok": True, "api_key": "sk-testsecret1234567890"}, + status_code=200, + used_workspace_delegation=True, + delegated_workspace_mode="read_only", + ), ) assert result["ok"] is False