SIENTIAPDE-1250: Refactor: Remove 'laborious' directory from test structure and update README.md accordingly.

2025-10-07 10:02:38 -03:00
parent 851f34c80e
commit 2628dc92d2
23 changed files with 11 additions and 13 deletions
--- a/tests/activities/test_activities.py
+++ b/tests/activities/test_activities.py
@@ -0,0 +1,139 @@
+from unittest.mock import ANY, MagicMock, patch
+
+from pytest import mark
+from sientia_do.temporal.activities.postgres import Postgres
+
+from model_manager.activities.activities import Activities
+from model_manager.activities.gates import Gates
+from model_manager.activities.mlflow import MLFlow
+
+
+@patch('model_manager.activities.activities.Postgres.__init__')
+@patch('model_manager.activities.activities.MLFlow.__init__')
+@patch('model_manager.activities.activities.MinIO.__init__')
+@patch('model_manager.activities.activities.Gates.__init__')
+def test___init__(mock_gates_init, mock_minio_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'}
+
+    minio_config = {
+        'endpoint_url': 'http://localhost:9000',
+        'access_key': 'minioadmin',
+        'secret_key': 'minioadmin',
+        'region': 'us-east-1',
+        'use_ssl': False,
+        'max_retry_attempts': 3,
+        'retry_mode': 'adaptive',
+        'connect_timeout': 10,
+        'read_timeout': 60,
+    }
+
+    logger = MagicMock()
+    notification_handler = MagicMock()
+
+    activities = Activities(
+        postgres_config=postgres_config,
+        mlflow_config=mlflow_config,
+        minio_config=minio_config,
+        logger=logger,
+        notification_handler=notification_handler,
+    )
+
+    assert isinstance(activities, Activities)
+    assert isinstance(activities, Postgres)
+    assert isinstance(activities, MLFlow)
+    assert isinstance(activities, Gates)
+
+    mock_postgres_init.assert_called_once_with(
+        ANY,
+        host=postgres_config['host'],
+        port=postgres_config['port'],
+        user=postgres_config['user'],
+        password=postgres_config['password'],
+        dbname=postgres_config['dbname'],
+        min_connections=postgres_config['min_connections'],
+        max_connections=postgres_config['max_connections'],
+        logger=logger,
+        notification_handler=notification_handler,
+    )
+
+    mock_mlflow_init.assert_called_once_with(
+        ANY,
+        mlflow_host=mlflow_config['host'],
+        mlflow_port=mlflow_config['port'],
+        mlflow_username=mlflow_config['username'],
+        mlflow_password=mlflow_config['password'],
+        logger=logger,
+        notification_handler=notification_handler,
+    )
+
+    mock_minio_init.assert_called_once_with(
+        ANY,
+        endpoint_url=minio_config['endpoint_url'],
+        access_key=minio_config['access_key'],
+        secret_key=minio_config['secret_key'],
+        region=minio_config['region'],
+        use_ssl=minio_config['use_ssl'],
+        max_retry_attempts=minio_config['max_retry_attempts'],
+        retry_mode=minio_config['retry_mode'],
+        connect_timeout=minio_config['connect_timeout'],
+        read_timeout=minio_config['read_timeout'],
+        logger=logger,
+        notification_handler=notification_handler,
+    )
+
+    mock_gates_init.assert_called_once_with(
+        ANY, logger=logger, notification_handler=notification_handler
+    )
+
+
+@mark.asyncio
+@patch('model_manager.activities.activities.Postgres', return_value=MagicMock())
+@patch('model_manager.activities.activities.MLFlow', return_value=MagicMock())
+async def test_shutdown(_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'}
+
+    minio_config = {
+        'endpoint_url': 'http://localhost:9000',
+        'access_key': 'minioadmin',
+        'secret_key': 'minioadmin',
+        'region': 'us-east-1',
+        'use_ssl': False,
+        'max_retry_attempts': 3,
+        'retry_mode': 'adaptive',
+        'connect_timeout': 10,
+        'read_timeout': 60,
+    }
+
+    logger = MagicMock()
+    notification_handler = MagicMock()
+
+    activities = Activities(
+        postgres_config=postgres_config,
+        mlflow_config=mlflow_config,
+        minio_config=minio_config,
+        logger=logger,
+        notification_handler=notification_handler,
+    )
+
+    await activities.shutdown()
+    mock_postgres_init.close.assert_called_once()