from unittest.mock import call, patch from tadas.platform import jenkins as jenkins_utils @patch.object(jenkins_utils, 'trigger_jenkins_job') @patch.object(jenkins_utils, 'get_server') def test_trigger_dbt_metrics(get_server_mock, trigger_jenkins_job_mock, monkeypatch): monkeypatch.setenv('SNOWFLAKE_WAREHOUSE', 'TEST_WAREHOUSE') jenkins_utils.trigger_dbt_metrics() assert trigger_jenkins_job_mock.call_args_list == [ call( server=get_server_mock.return_value, job_name='dbt-scheduler-analytics-parametrized-pipeline', params={ 'ENVIRONMENT': 'dev', 'MODELS': 'METRICS_TADAS_DBT', 'WAREHOUSE': 'TEST_WAREHOUSE', 'CLEAR_CACHE': False, } ) ]