Code import - branch release/SIENTIAPDE-1645

tests/workflows/test_cleanup_files.py

@@ -0,0 +1,30 @@
+"""Unit tests for the CleanupFiles workflow."""
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+
+@pytest.mark.asyncio
+@patch('model_manager.workflows.cleanup_files.workflow')
+async def test_cleanup_files_workflow(mock_workflow_module):
+    """Test the CleanupFiles workflow."""
+    from model_manager.runtime_paths import REPORTS_TEMP_DIR
+    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({})
+
+    # Verify that the activities were called with the correct parameters
+    calls = mock_workflow_module.execute_activity_method.call_args_list
+    assert len(calls) == 1
+
+    # Check cleanup_temp_directories call
+    local_call_args = calls[0][0][1]
+    assert local_call_args['temp_path'] == REPORTS_TEMP_DIR
+    assert local_call_args['metadata']['workflow_name'] == 'cleanup_files'
+    assert 'pod_id' in local_call_args['metadata']