Class MockToolManager

tests/tools/test_execution.py:15–37

Mock ToolManager for testing execution functions.

class MockToolManager:
    """Mock ToolManager for testing execution functions."""

    def __init__(self, results: dict[str, ToolCallResult] | None = None):
        self.tool_timeout = 30.0
        self.results = results or {}
        self.executed_tools: list[str] = []

    async def execute_tool(
        self,
        tool_name: str,
        arguments: dict,
        namespace: str | None = None,
        timeout: float | None = None,
    ) -> ToolCallResult:
        self.executed_tools.append(tool_name)
        if tool_name in self.results:
            return self.results[tool_name]
        return ToolCallResult(
            tool_name=tool_name,
            success=True,
            result={"output": f"result from {tool_name}"},
        )


# ----------------------------------------------------------------------------