chore(ci): Apply pytest-timeout to test_bgtask (#3226)

Backported-from: main (24.12) Backported-to: 24.09 Backport-of: 3226
lablup · Dec 9, 2024 · 17368f8 · 17368f8
1 parent eec807b
commit 17368f8
Show file tree

Hide file tree

Showing 3 changed files with 284 additions and 164 deletions.
diff --git a/tests/manager/api/test_bgtask.py b/tests/manager/api/test_bgtask.py
@@ -19,6 +19,7 @@
 from ai.backend.manager.server import background_task_ctx, event_dispatcher_ctx, shared_config_ctx
 
 
+@pytest.mark.timeout(60)
 @pytest.mark.asyncio
 async def test_background_task(etcd_fixture, create_app_and_client) -> None:
     app, client = await create_app_and_client(
@@ -84,6 +85,7 @@ async def _mock_task(reporter):
         await dispatcher.close()
 
 
+@pytest.mark.timeout(60)
 @pytest.mark.asyncio
 async def test_background_task_fail(etcd_fixture, create_app_and_client) -> None:
     app, client = await create_app_and_client(

diff --git a/tools/pytest-requirements.txt b/tools/pytest-requirements.txt
@@ -5,4 +5,5 @@ pytest-cov>=5.0
 pytest-custom_exit_code>=0.3.0
 pytest-dependency>=0.6.0
 pytest-mock>=3.14.0
+pytest-timeout>=2.3.1
 aioresponses>=0.7.6