fix run manager test cleanup await

This commit is contained in:
rayhpeng
2026-05-22 18:01:55 +08:00
parent f3e3a350ce
commit 43eb643910
+3 -3
View File
@@ -2,7 +2,6 @@
import asyncio
import re
from contextlib import suppress
import pytest
@@ -298,13 +297,14 @@ async def test_create_does_not_expose_run_until_store_persist_completes():
assert [run.run_id for run in runs] == [record.run_id]
finally:
allow_put.set()
cleanup_tasks = []
for task in (list_task, create_task):
if task is None:
continue
if not task.done():
task.cancel()
with suppress(asyncio.CancelledError):
await task
cleanup_tasks.append(task)
await asyncio.gather(*cleanup_tasks, return_exceptions=True)
@pytest.mark.anyio