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

Co-authored-by: Joongi Kim <[email protected]>

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(
@@ -79,11 +80,13 @@ async def _mock_task(reporter):
         assert done_handler_ctx["event_name"] == "bgtask_done"
         assert done_handler_ctx["message"] == "hooray"
     finally:
-        await redis_helper.execute(producer.redis_client, lambda r: r.flushdb())
+        await root_ctx.background_task_manager.shutdown()
         await producer.close()
         await dispatcher.close()
+        await redis_helper.execute(producer.redis_client, lambda r: r.flushdb())
 
 
+@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(
@@ -119,6 +122,7 @@ async def _mock_task(reporter):
         assert fail_handler_ctx["message"] is not None
         assert "ZeroDivisionError" in fail_handler_ctx["message"]
     finally:
-        await redis_helper.execute(producer.redis_client, lambda r: r.flushdb())
+        await root_ctx.background_task_manager.shutdown()
         await producer.close()
         await dispatcher.close()
+        await redis_helper.execute(producer.redis_client, lambda r: r.flushdb())

tools/pytest-requirements.txt

@@ -3,6 +3,7 @@ pytest-asyncio>=0.23.2
 pytest-aiohttp>=1.0.5
 pytest-cov>=4.1
 pytest-custom_exit_code>=0.3.0
-pytest-dependency>=0.5.1
-pytest-mock>=3.12.0
+pytest-dependency>=0.6.0
+pytest-mock>=3.14.0
+pytest-timeout>=2.3.1
 aioresponses>=0.7.6