gitea_admin/sientia-dataops-orchestrator_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1646

Browse Source 
SIENTIAPDE-1646: Remove validate.sh script, refactor TemporalManager for improved readability, and update temporal schedule tests to explicitly define workflow types and align task queue naming conventions.
This commit is contained in:
vitor-aignosi
2026-07-21 10:17:30 -03:00
parent 7b845a7de4
commit a4b6f6aa88
3 changed files with 38 additions and 139 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
tests/orchestrator/activities/test_temporal_manager.py
View File

@@ -206,7 +206,7 @@ async def test_create_schedule(
'test-workflow',
input_data['schedules']['scouter']['test-schedule'],
id='test-schedule',
task_queue='test-workflow-legacy-queue',
task_queue='test-workflow-queue',
execution_timeout=timedelta(seconds=100),
run_timeout=timedelta(seconds=100),
task_timeout=timedelta(seconds=100),
@@ -216,7 +216,7 @@ async def test_create_schedule(
'test-workflow',
input_data['schedules']['scouter']['test-schedule-invalid-frequency'],
id='test-schedule-invalid-frequency',
task_queue='test-workflow-legacy-queue',
task_queue='test-workflow-queue',
execution_timeout=timedelta(seconds=400),
run_timeout=timedelta(seconds=400),
task_timeout=timedelta(seconds=400),
@@ -226,7 +226,7 @@ async def test_create_schedule(
'test-workflow',
input_data['schedules']['laborious']['test-schedule-laborious'],
id='test-schedule-laborious',
task_queue='test-workflow-legacy-queue',
task_queue='test-workflow-queue',
execution_timeout=timedelta(seconds=500),
run_timeout=timedelta(seconds=500),
task_timeout=timedelta(seconds=500),
@@ -350,10 +350,24 @@ async def test_update_schedules(
input_data = {
'schedules': {
'scouter': {
'test-schedule': {'frequency': '1m', 'data': {'test': 'test'}},
'test-schedule_no_handler': {'frequency': '1m', 'data': {'test': 'test'}},
'test-schedule': {
'workflow_type': 'scouter',
'frequency': '1m',
'data': {'test': 'test'},
},
'test-schedule_no_handler': {
'workflow_type': 'scouter',
'frequency': '1m',
'data': {'test': 'test'},
},
},
'laborious': {
'test-schedule-laborious': {
'workflow_type': 'laborious',
'frequency': '2m',
'data': {'test': 'test'},
}
},
'laborious': {'test-schedule-laborious': {'frequency': '2m', 'data': {'test': 'test'}}},
}
}
@@ -514,7 +528,7 @@ async def test_create_schedules_default_runtime_legacy_queue(
'test-schedule': {
'model_id': 1,
'model_name': 'test-model-name',
'workflow_type': 'scouter',
'workflow_type': 'drift',
'frequency': '1m',
'data': {'test': 'test'},
}
@@ -527,10 +541,10 @@ async def test_create_schedules_default_runtime_legacy_queue(
await temporal_manager.create_schedules(input_data)
mock_schedule_action_start_workflow.assert_called_once_with(
'scouter',
'drift',
input_data['schedules']['scouter']['test-schedule'],
id='test-schedule',
task_queue='scouter-legacy-queue',
task_queue='drift-legacy-queue',
execution_timeout=timedelta(seconds=300),
run_timeout=timedelta(seconds=300),
task_timeout=timedelta(seconds=300),
@@ -562,7 +576,7 @@ async def test_create_schedules_tenant_runtime_queue(
'test-schedule': {
'model_id': 1,
'model_name': 'test-model-name',
'workflow_type': 'scouter',
'workflow_type': 'drift',
'frequency': '1m',
'runtime': 'tenant-x',
'data': {'test': 'test'},
@@ -576,10 +590,10 @@ async def test_create_schedules_tenant_runtime_queue(
await temporal_manager.create_schedules(input_data)
mock_schedule_action_start_workflow.assert_called_once_with(
'scouter',
'drift',
input_data['schedules']['scouter']['test-schedule'],
id='test-schedule',
task_queue='scouter-tenant-x-queue',
task_queue='drift-tenant-x-queue',
execution_timeout=timedelta(seconds=300),
run_timeout=timedelta(seconds=300),
task_timeout=timedelta(seconds=300),