Class SlowMockToolManager

tests/tools/test_execution.py:503–528

Mock ToolManager with configurable per-tool delays for batch timeout tests.

class SlowMockToolManager:
    """Mock ToolManager with configurable per-tool delays for batch timeout tests."""

    def __init__(
        self, delays: dict[str, float] | None = None, default_delay: float = 0.0
    ):
        self.tool_timeout = 30.0
        self.delays = delays or {}
        self.default_delay = default_delay
        self.executed_tools: list[str] = []

    async def execute_tool(
        self,
        tool_name: str,
        arguments: dict,
        namespace: str | None = None,
        timeout: float | None = None,
    ) -> ToolCallResult:
        delay = self.delays.get(tool_name, self.default_delay)
        await asyncio.sleep(delay)
        self.executed_tools.append(tool_name)
        return ToolCallResult(
            tool_name=tool_name,
            success=True,
            result={"output": f"result from {tool_name}"},
        )


@pytest.mark.asyncio