Function test_stream_cancelled_error_path

tests/tools/test_execution.py:724–754

Cover lines 251-256: CancelledError inside queue.get() wait. We cancel the outer consuming task while stream_execute_tools is blocked waiting for a result. The CancelledError propagates into the except branch, which cancels remaining tasks and breaks.

()

@pytest.mark.asyncio
async def test_stream_cancelled_error_path():
    """Cover lines 251-256: CancelledError inside queue.get() wait.

    We cancel the outer consuming task while stream_execute_tools is blocked
    waiting for a result.  The CancelledError propagates into the except branch,
    which cancels remaining tasks and breaks.
    """
    manager = SlowMockToolManager(default_delay=10.0)  # all tools are very slow
    calls = [
        CTPToolCall(id="call_1", tool="slow_tool_1", arguments={}),
        CTPToolCall(id="call_2", tool="slow_tool_2", arguments={}),
    ]

    results = []

    async def consumer():
        async for result in stream_execute_tools(manager, calls, batch_timeout=100.0):
            results.append(result)

    task = asyncio.create_task(consumer())
    # Give the generator time to start and block on queue.get()
    await asyncio.sleep(0.05)
    task.cancel()

    try:
        await task
    except asyncio.CancelledError:
        pass  # expected

    # No results should have been received (tools are too slow)
    assert results == []


@pytest.mark.asyncio