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

SIENTIAPDE-1712

Browse Source 
Add 'on_conflict' parameter to multiple test cases in test_orchestrator_functions.py for improved error handling consistency.
This commit is contained in:
vitor-aignosi
2026-04-01 13:53:53 -03:00
parent c4f221438b
commit 4d5231ca16
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/orchestrator/utils/test_orchestrator_functions.py
View File

@@ -33,6 +33,7 @@ def test_common_config():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'execution_timeout_seconds': 300, 'execution_timeout_seconds': 300,
'task_timeout_seconds': 300, 'task_timeout_seconds': 300,
} }
@@ -58,6 +59,7 @@ def test_drift():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'schema': 'sientia_data', 'schema': 'sientia_data',
'source_table_name': 'laborious_data', 'source_table_name': 'laborious_data',
'target_table_name': 'drift_metrics', 'target_table_name': 'drift_metrics',
@@ -88,6 +90,7 @@ def test_simple_metrics():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'schema': 'sientia_data', 'schema': 'sientia_data',
'predictions_table_name': 'predictions', 'predictions_table_name': 'predictions',
'data_table_name': 'laborious_data', 'data_table_name': 'laborious_data',
@@ -119,6 +122,7 @@ def test_minimal_retrain():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'query': 'select * from sientia_data.laborious_data order by "timestamp" desc limit 30;', 'query': 'select * from sientia_data.laborious_data order by "timestamp" desc limit 30;',
'schema': 'sientia_data', 'schema': 'sientia_data',
'table_name': 'log_retrain', 'table_name': 'log_retrain',
@@ -153,6 +157,7 @@ def test_scouter():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'topic': 'raw_test_schedule', 'topic': 'raw_test_schedule',
'trigger_laborious': False, 'trigger_laborious': False,
'filters': {'test_filter_name': {'policy': 'test_policy'}}, 'filters': {'test_filter_name': {'policy': 'test_policy'}},
@@ -254,6 +259,7 @@ def test_predictions_batch(mock_process_path_priority, mock_overlap_filter_confi
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'query': 'test_query', 'query': 'test_query',
'schema': 'sientia_data', 'schema': 'sientia_data',
'table_name': 'predictions', 'table_name': 'predictions',
@@ -556,6 +562,7 @@ def test_base_scouter():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'trigger_laborious': False, 'trigger_laborious': False,
'filters': { 'filters': {
'test_filter_name': {'policy': 'test_policy'}, 'test_filter_name': {'policy': 'test_policy'},
@@ -605,6 +612,7 @@ def test_pi_web_api_scouter():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'trigger_laborious': False, 'trigger_laborious': False,
'filters': {'test_filter_name': {'policy': 'test_policy'}}, 'filters': {'test_filter_name': {'policy': 'test_policy'}},
'schema': 'sientia_data', 'schema': 'sientia_data',
@@ -660,6 +668,7 @@ def test_pi_web_api_scouter_with_timeout_greater_than_frequency():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'trigger_laborious': False, 'trigger_laborious': False,
'filters': {}, 'filters': {},
'schema': 'sientia_data', 'schema': 'sientia_data',
@@ -710,6 +719,7 @@ def test_pi_web_api_scouter_with_no_timeout():
'model_id': 'test_model_id', 'model_id': 'test_model_id',
'model_name': 'test_model_name', 'model_name': 'test_model_name',
'model_config': {'test_config': 'test_config'}, 'model_config': {'test_config': 'test_config'},
'on_conflict': 'error',
'trigger_laborious': False, 'trigger_laborious': False,
'filters': {}, 'filters': {},
'schema': 'sientia_data', 'schema': 'sientia_data',