import asyncio from app.main import run async def test_run_returns_when_stop_is_set() -> None: stop = asyncio.Event() task = asyncio.create_task(run(stop)) await asyncio.sleep(0) assert not task.done() stop.set() await asyncio.wait_for(task, timeout=1)