mirror of
https://github.com/hoshikawa2/agent_platform_oci.git
synced 2026-09-07 18:23:46 +00:00
99 lines
3.3 KiB
Python
99 lines
3.3 KiB
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
|
|
from agent_framework.mcp.tool_policy import ToolPolicyRegistry
|
|
from agent_framework.mcp.tool_router import MCPToolRouter
|
|
|
|
|
|
class _Registry:
|
|
def __init__(self, tool=None):
|
|
self.tool = tool
|
|
|
|
def get_tool(self, _name):
|
|
return self.tool
|
|
|
|
|
|
def _router(policy_registry, legacy=None):
|
|
router = MCPToolRouter.__new__(MCPToolRouter)
|
|
router.tool_policies = policy_registry
|
|
router.registry = _Registry(legacy)
|
|
return router
|
|
|
|
|
|
def test_missing_policy_file_preserves_legacy_behavior(tmp_path):
|
|
policies = ToolPolicyRegistry(str(tmp_path / "missing.yaml"))
|
|
legacy = SimpleNamespace(
|
|
tool_type="action",
|
|
requires=["order_id"],
|
|
confirmation_required=True,
|
|
execution_policy={},
|
|
)
|
|
router = _router(policies, legacy)
|
|
|
|
allowed, reason, metadata = router.validate_execution_policy("alterar", {"order_id": "42"})
|
|
|
|
assert allowed is False
|
|
assert "confirmação" in reason
|
|
assert metadata["operation_type"] == "transactional"
|
|
assert metadata["policy_source"] == "tools.yaml"
|
|
|
|
|
|
def test_read_only_policy_executes_without_confirmation(tmp_path):
|
|
path = tmp_path / "tool_policies.yaml"
|
|
path.write_text(
|
|
"tool_policies:\n consultar:\n operation_type: read_only\n",
|
|
encoding="utf-8",
|
|
)
|
|
router = _router(ToolPolicyRegistry(str(path)))
|
|
|
|
allowed, reason, metadata = router.validate_execution_policy("consultar", {})
|
|
|
|
assert allowed is True
|
|
assert reason is None
|
|
assert metadata["operation_type"] == "read_only"
|
|
|
|
|
|
def test_transactional_policy_requires_literal_boolean_confirmation(tmp_path):
|
|
path = tmp_path / "tool_policies.yaml"
|
|
path.write_text(
|
|
"tool_policies:\n cancelar:\n operation_type: transactional\n require_confirmation: true\n",
|
|
encoding="utf-8",
|
|
)
|
|
router = _router(ToolPolicyRegistry(str(path)))
|
|
|
|
denied, _, _ = router.validate_execution_policy("cancelar", {"confirmed": "true"})
|
|
allowed, reason, metadata = router.validate_execution_policy("cancelar", {"confirmed": True})
|
|
|
|
assert denied is False
|
|
assert allowed is True
|
|
assert reason is None
|
|
assert metadata["policy_source"] == "tool_policies.yaml"
|
|
|
|
|
|
def test_requires_confirmation_alias_is_supported(tmp_path):
|
|
path = tmp_path / "tool_policies.yaml"
|
|
path.write_text(
|
|
"tool_policies:\n alterar:\n type: transactional\n requires_confirmation: true\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
policy = ToolPolicyRegistry(str(path)).get("alterar")
|
|
|
|
assert policy.operation_type == "transactional"
|
|
assert policy.require_confirmation is True
|
|
|
|
|
|
def test_workflow_execution_policy_is_exposed(tmp_path):
|
|
path = tmp_path / "tool_policies.yaml"
|
|
path.write_text(
|
|
"""defaults:\n operation_type: read_only\ntool_policies:\n refund:\n operation_type: transactional\n require_confirmation: true\n execution:\n mode: workflow\n workflow: refund_order\n version: 2\n""",
|
|
encoding="utf-8",
|
|
)
|
|
registry = ToolPolicyRegistry(str(path))
|
|
policy = registry.get("refund")
|
|
assert policy is not None
|
|
assert policy.execution.mode == "workflow"
|
|
assert policy.execution.workflow == "refund_order"
|
|
assert policy.execution.version == 2
|