Function test_batch_timeout_cancels_remaining

tests/tools/test_execution.py:532–548

When batch timeout fires, remaining tasks are cancelled.

()

@pytest.mark.asyncio
async def test_batch_timeout_cancels_remaining():
    """When batch timeout fires, remaining tasks are cancelled."""
    # fast_tool completes quickly, slow_tool takes much longer than the batch timeout
    manager = SlowMockToolManager(delays={"fast_tool": 0.01, "slow_tool": 10.0})
    calls = [
        CTPToolCall(id="call_1", tool="fast_tool", arguments={}),
        CTPToolCall(id="call_2", tool="slow_tool", arguments={}),
    ]

    results = await execute_tools_parallel(manager, calls, batch_timeout=0.5)

    # fast_tool should have completed; slow_tool should have been cancelled
    completed_tools = {r.tool for r in results}
    assert "fast_tool" in completed_tools
    # slow_tool should NOT have completed (it sleeps for 10s, batch timeout is 0.5s)
    assert "slow_tool" not in completed_tools
    assert len(results) < len(calls)


@pytest.mark.asyncio