def test_allowed_jenkins_states_failure(self, state, allowed_jenkins_states):
jenkins_mock = Mock(spec=jenkins.Jenkins, auth='secret')
jenkins_mock.get_job_info.return_value = {'nextBuildNumber': '1'}
jenkins_mock.get_build_info.return_value = {
'result': state,
'url': 'http://aaa.fake-url.com/congratulation/its-a-job',
}
jenkins_mock.build_job_url.return_value = 'http://www.jenkins.url/somewhere/in/the/universe'
hook_mock = Mock(spec=JenkinsHook)
hook_mock.get_jenkins_server.return_value = jenkins_mock
with patch.object(JenkinsJobTriggerOperator, "get_hook") as get_hook_mocked, patch(
'airflow.providers.jenkins.operators.jenkins_job_trigger.jenkins_request_with_headers'
) as mock_make_request:
mock_make_request.side_effect = [
{'body': '', 'headers': {'Location': 'http://what-a-strange.url/18'}},
{'body': '{"executable":{"number":"1"}}', 'headers': {}},
]
get_hook_mocked.return_value = hook_mock
operator = JenkinsJobTriggerOperator(
dag=None,
task_id="operator_test",
job_name="a_job_on_jenkins",
jenkins_connection_id="fake_jenkins_connection",
allowed_jenkins_states=allowed_jenkins_states,
# The hook is mocked, this connection won't be used
sleep_time=1,
)
with pytest.raises(AirflowException):
operator.execute(None)
def test_execute(self, _, parameters):
jenkins_mock = Mock(spec=jenkins.Jenkins, auth='secret')
jenkins_mock.get_build_info.return_value = {
'result': 'SUCCESS',
'url': 'http://aaa.fake-url.com/congratulation/its-a-job',
}
jenkins_mock.build_job_url.return_value = 'http://www.jenkins.url/somewhere/in/the/universe'
hook_mock = Mock(spec=JenkinsHook)
hook_mock.get_jenkins_server.return_value = jenkins_mock
with patch.object(JenkinsJobTriggerOperator, "get_hook") as get_hook_mocked, patch(
'airflow.providers.jenkins.operators.jenkins_job_trigger.jenkins_request_with_headers'
) as mock_make_request:
mock_make_request.side_effect = [
{'body': '', 'headers': {'Location': 'http://what-a-strange.url/18'}},
{'body': '{"executable":{"number":"1"}}', 'headers': {}},
]
get_hook_mocked.return_value = hook_mock
operator = JenkinsJobTriggerOperator(
dag=None,
jenkins_connection_id="fake_jenkins_connection",
# The hook is mocked, this connection won't be used
task_id="operator_test",
job_name="a_job_on_jenkins",
parameters=parameters,
sleep_time=1,
)
operator.execute(None)
self.assertEqual(jenkins_mock.get_build_info.call_count, 1)
jenkins_mock.get_build_info.assert_called_once_with(name='a_job_on_jenkins', number='1')
def test_build_job_request_settings(self):
jenkins_mock = Mock(spec=jenkins.Jenkins, auth='secret', timeout=2)
jenkins_mock.build_job_url.return_value = 'http://apache.org'
with patch(
'airflow.providers.jenkins.operators.jenkins_job_trigger.jenkins_request_with_headers'
) as mock_make_request:
operator = JenkinsJobTriggerOperator(
dag=None,
task_id="build_job_test",
job_name="a_job_on_jenkins",
jenkins_connection_id="fake_jenkins_connection")
operator.build_job(jenkins_mock)
mock_request = mock_make_request.call_args_list[0][0][1]
self.assertEqual(mock_request.method, 'POST')
self.assertEqual(mock_request.url, 'http://apache.org')