SIENTIAPDE-994

Add shutdown tests for Activities and OPC classes; improve test assertions

tests/laborious/activities/test_activities.py

@@ -147,3 +147,47 @@ async def test_prepare_activity(_mock_opc_init,
         'model_name']
     assert activities.notification_handler.base_notification.model_id == input_data[
         'model_id']
+
+
+@patch('laborious.activities.activities.Postgres', return_value=MagicMock())
+@patch('laborious.activities.activities.MLFlow', return_value=MagicMock())
+@patch('laborious.activities.activities.OPC', return_value=MagicMock())
+def test_shutdown(mock_opc_init,
+                  _mock_mlflow_init, mock_postgres_init):
+    postgres_config = {
+        'host': 'localhost',
+        'port': 5432,
+        'user': 'postgres',
+        'password': 'postgres',
+        'dbname': 'postgres',
+        'min_connections': 1,
+        'max_connections': 10
+    }
+
+    mlflow_config = {
+        'host': 'localhost',
+        'port': 5000,
+        'username': 'mlflow',
+        'password': 'mlflow'
+    }
+
+    opc_config = {
+        'bootstrap_servers': 'localhost:9092',
+        'polling_time': 1000,
+        'group_id': 'test-group'
+    }
+
+    logger = MagicMock()
+    notification_handler = MagicMock()
+
+    activities = Activities(
+        postgres_config=postgres_config,
+        mlflow_config=mlflow_config,
+        opc_config=opc_config,
+        logger=logger,
+        notification_handler=notification_handler
+    )
+
+    activities.shutdown()
+    mock_opc_init.shutdown.assert_called_once()
+    mock_postgres_init.close.assert_called_once()

tests/laborious/activities/test_opc.py

@@ -189,3 +189,8 @@ async def test_write_opc_data_empty_config(opc):
 
     # Assert
     opc.opc_repository['server1'].write_data.assert_not_called()
+
+
+def test_shutdown(opc):
+    opc.shutdown()
+    opc.opc_repository['server1'].disconnect.assert_called_once()