def test_generate_pipeline_success_with_opts(
mock_generate_pipeline,
mock_validate_service_name):
# paasta generate succeeds when service name provided as arg
mock_validate_service_name.return_value = None
args = MagicMock()
args.service = 'fake_service'
assert paasta_generate_pipeline(args) is None
mock_generate_pipeline.assert_called_once_with(service='fake_service')
def test_paasta_generate_pipeline_service_not_found(
mock_stdout, mock_guess_service_name, mock_validate_service_name):
# paasta generate cannot guess service name and none is provided
mock_guess_service_name.return_value = 'not_a_service'
mock_validate_service_name.side_effect = NoSuchService(None)
args = MagicMock()
args.service = None
expected_output = "%s\n" % NoSuchService.GUESS_ERROR_MSG
# Fail if exit(1) does not get called
with raises(SystemExit) as sys_exit:
paasta_generate_pipeline(args)
output = mock_stdout.getvalue()
assert sys_exit.value.code == 1
assert output == expected_output
def test_paasta_generate_pipeline_success_no_opts(
mock_generate_pipeline,
mock_guess_service_name,
mock_validate_service_name):
# paasta generate succeeds when service name must be guessed
mock_guess_service_name.return_value = 'fake_service'
mock_validate_service_name.return_value = None
args = MagicMock()
args.service = None
assert paasta_generate_pipeline(args) is None
mock_generate_pipeline.assert_called_once_with(service='fake_service')
