From 35efb67c89381d93d107a577f8f0b6bff3b397b9 Mon Sep 17 00:00:00 2001 From: Bruno Domingues Date: Tue, 4 Aug 2026 14:43:15 -0300 Subject: [PATCH] fix(test_activities): Use aclose for async mock shutdown --- tests/laborious/activities/test_activities.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/laborious/activities/test_activities.py b/tests/laborious/activities/test_activities.py index 8e6cd78..54fdd89 100644 --- a/tests/laborious/activities/test_activities.py +++ b/tests/laborious/activities/test_activities.py @@ -173,7 +173,7 @@ async def test_shutdown( mock_mlflow_init, mock_storage_init, ): - mock_opc_init.close = AsyncMock() + mock_opc_init.aclose = AsyncMock() postgres_config = { 'host': 'localhost', 'port': 5432, @@ -221,7 +221,7 @@ async def test_shutdown( ) await activities.shutdown() - mock_opc_init.close.assert_called_once() + mock_opc_init.aclose.assert_called_once() mock_storage_init.close.assert_called_once() mock_mlflow_init.close.assert_called_once() mock_gates_init.close.assert_called_once()