llama_deploy/e2e_tests/message_queues/simple/test_server.py

import asyncio

import pytest

from llama_deploy.message_queues import (
    SimpleMessageQueueConfig,
    SimpleMessageQueueServer,
)


@pytest.mark.asyncio
async def test_cancel_launch_server():
    mq = SimpleMessageQueueServer(SimpleMessageQueueConfig(port=8009))
    t = asyncio.create_task(mq.launch_server())

    # Make sure the queue starts
    await asyncio.sleep(1)

    # Cancel
    t.cancel()
    await t