From 33480fa1083a57b1d56d3591b4df89fed8a5089a Mon Sep 17 00:00:00 2001
From: Bruno Domingues <bruno.domingues@aignosi.com.br>
Date: Tue, 25 Nov 2025 16:44:51 -0300
Subject: [PATCH] SIENTIAPDE-1350: Add unit tests for the CleanupFiles
 workflow, covering both input bucket and default bucket scenarios.

---
 tests/workflows/test_cleanup_files.py | 76 +++++++++++++++++++++++++++
 1 file changed, 76 insertions(+)
 create mode 100644 tests/workflows/test_cleanup_files.py

diff --git a/tests/workflows/test_cleanup_files.py b/tests/workflows/test_cleanup_files.py
new file mode 100644
index 0000000..28d02e5
--- /dev/null
+++ b/tests/workflows/test_cleanup_files.py
@@ -0,0 +1,76 @@
+"""Unit tests for the CleanupFiles workflow."""
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+
+@pytest.mark.asyncio
+@patch('model_manager.workflows.cleanup_files.workflow')
+@patch('model_manager.workflows.cleanup_files.POD_ID', 'temporal-pod')
+async def test_cleanup_files_workflow_with_input_bucket(mock_workflow_module):
+    """Test the CleanupFiles workflow when bucket_name is provided in the input."""
+    from model_manager.workflows.cleanup_files import CleanupFiles
+
+    # Mock execute_activity_method
+    mock_workflow_module.execute_activity_method = AsyncMock()
+
+    # Instantiate and run the workflow
+    workflow_instance = CleanupFiles()
+    await workflow_instance.run({'bucket_name': 'input-bucket'})
+
+    # Verify that the activities were called with the correct parameters
+    calls = mock_workflow_module.execute_activity_method.call_args_list
+    assert len(calls) == 2
+
+    # Check cleanup_minio_files call
+    minio_call_args = calls[0][0][1]
+    assert minio_call_args['bucket_name'] == 'input-bucket'
+    assert minio_call_args['metadata'] == {
+        'pod_id': 'temporal-pod',
+        'workflow_name': 'cleanup_files',
+    }
+
+    # Check cleanup_temp_directories call
+    local_call_args = calls[1][0][1]
+    assert local_call_args['temp_path'] == 'model_manager/reports/temp'
+    assert local_call_args['metadata'] == {
+        'pod_id': 'temporal-pod',
+        'workflow_name': 'cleanup_files',
+    }
+
+
+@pytest.mark.asyncio
+@patch('model_manager.workflows.cleanup_files.workflow')
+@patch('model_manager.workflows.cleanup_files.POD_ID', 'temporal-pod')
+@patch('model_manager.workflows.cleanup_files.DEFAULT_CLEANUP_BUCKET', 'env-var-bucket')
+async def test_cleanup_files_workflow_with_default_bucket(mock_workflow_module):
+    """Test the CleanupFiles workflow when using the default bucket from environment variables."""
+    from model_manager.workflows.cleanup_files import CleanupFiles
+
+    # Mock execute_activity_method
+    mock_workflow_module.execute_activity_method = AsyncMock()
+
+    # Instantiate and run the workflow
+    workflow_instance = CleanupFiles()
+    await workflow_instance.run({})  # Empty input
+
+    # Verify that the activities were called
+    calls = mock_workflow_module.execute_activity_method.call_args_list
+    assert len(calls) == 2
+
+    # Check cleanup_minio_files call
+    minio_call_args = calls[0][0][1]
+    assert minio_call_args['bucket_name'] == 'env-var-bucket'
+    assert minio_call_args['metadata'] == {
+        'pod_id': 'temporal-pod',
+        'workflow_name': 'cleanup_files',
+    }
+
+    # Check cleanup_temp_directories call
+    local_call_args = calls[1][0][1]
+    assert local_call_args['temp_path'] == 'model_manager/reports/temp'
+    assert local_call_args['metadata'] == {
+        'pod_id': 'temporal-pod',
+        'workflow_name': 'cleanup_files',
+    }