c0bef2d688
- Changed project name in values.yaml from "sientia-dataops-model-manager" to "sientia-model-manager". - Added new environment variables for GitHub repository and branch configuration. - Refactored cleanup paths to use a centralized REPORTS_TEMP_DIR constant for consistency. - Updated runtime configurations and adjusted volume mounts for better resource management. - Enabled SSH access for the model manager and disabled Grafana dashboard creation. - Updated tests to reflect changes in directory paths and environment variable usage.
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""Unit tests for the CleanupFiles workflow."""
|
|
|
|
import os
|
|
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()
|
|
with patch.dict(os.environ, {'POD_ID': 'temporal-pod'}):
|
|
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'] == {
|
|
'pod_id': 'temporal-pod',
|
|
'workflow_name': 'cleanup_files',
|
|
}
|