gitea_admin/sientia-dataops-laborious_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(test_activities): Use aclose for async mock shutdown

Browse Source
This commit is contained in:
Bruno Domingues
2026-08-04 14:43:15 -03:00
parent acd925cf2a
commit 35efb67c89
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/laborious/activities/test_activities.py
View File

@@ -173,7 +173,7 @@ async def test_shutdown(
mock_mlflow_init, mock_mlflow_init,
mock_storage_init, mock_storage_init,
): ):
mock_opc_init.close = AsyncMock() mock_opc_init.aclose = AsyncMock()
postgres_config = { postgres_config = {
'host': 'localhost', 'host': 'localhost',
'port': 5432, 'port': 5432,
@@ -221,7 +221,7 @@ async def test_shutdown(
) )
await activities.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_storage_init.close.assert_called_once()
mock_mlflow_init.close.assert_called_once() mock_mlflow_init.close.assert_called_once()
mock_gates_init.close.assert_called_once() mock_gates_init.close.assert_called_once()