65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
import pytest
|
|
|
|
from a2a_pack import NoAuth
|
|
from a2a_pack.context import LocalRunContext
|
|
|
|
from agent import CustomerIntegrationEngineer, SmokeCallInput
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_source_policy_rejects_code_editor_agent():
|
|
agent = CustomerIntegrationEngineer()
|
|
ctx = LocalRunContext(auth=NoAuth())
|
|
|
|
result = await agent.review_source_change_policy(
|
|
ctx,
|
|
mutated_source=False,
|
|
critical_defect_remaining=False,
|
|
used_code_editor_agent=True,
|
|
improvement_iterations=0,
|
|
)
|
|
|
|
assert result["ok"] is False
|
|
failed = {check["name"] for check in result["checks"] if not check["ok"]}
|
|
assert "no_code_editor_agent" in failed
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_smoke_call_rejects_secret_like_result():
|
|
agent = CustomerIntegrationEngineer()
|
|
ctx = LocalRunContext(auth=NoAuth())
|
|
|
|
result = await agent.validate_smoke_call(
|
|
ctx,
|
|
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
|
|
failed = {check["name"] for check in result["checks"] if not check["ok"]}
|
|
assert "no_secrets_returned" in failed
|
|
assert "$.api_key" in result["secret_like_paths"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_final_report_passes_with_clean_validations():
|
|
agent = CustomerIntegrationEngineer()
|
|
ctx = LocalRunContext(auth=NoAuth())
|
|
|
|
result = await agent.final_release_report(
|
|
ctx,
|
|
validation_results=[{"checks": [{"name": "sample", "ok": True}]}],
|
|
reviewer_findings=[],
|
|
warnings_to_accept=[],
|
|
)
|
|
|
|
assert result["ok"] is True
|
|
assert result["critical_findings"] == []
|
|
assert result["secrets_included"] is False
|