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')