def test_main_success(self):
fake_client = mock.MagicMock()
with contextlib.nested(
mock.patch(
'setup_marathon_job.parse_args',
return_value=self.fake_args,
autospec=True,
),
mock.patch(
'setup_marathon_job.get_main_marathon_config',
return_value=self.fake_marathon_config,
autospec=True,
),
mock.patch(
'paasta_tools.marathon_tools.get_marathon_client',
return_value=fake_client,
autospec=True,
),
mock.patch(
'paasta_tools.marathon_tools.load_marathon_service_config',
return_value=self.fake_marathon_service_config,
autospec=True,
),
mock.patch(
'setup_marathon_job.setup_service',
return_value=(0, 'it_is_finished'),
autospec=True,
),
mock.patch('setup_marathon_job.load_system_paasta_config', autospec=True),
mock.patch('setup_marathon_job.send_event', autospec=True),
mock.patch('sys.exit', autospec=True),
) as (
parse_args_patch,
get_main_conf_patch,
get_client_patch,
read_service_conf_patch,
setup_service_patch,
load_system_paasta_config_patch,
sensu_patch,
sys_exit_patch,
):
load_system_paasta_config_patch.return_value.get_cluster = mock.Mock(return_value=self.fake_cluster)
setup_marathon_job.main()
parse_args_patch.assert_called_once_with()
get_main_conf_patch.assert_called_once_with()
get_client_patch.assert_called_once_with(
self.fake_marathon_config.get_url(),
self.fake_marathon_config.get_username(),
self.fake_marathon_config.get_password(),
)
read_service_conf_patch.assert_called_once_with(
decompose_job_id(self.fake_args.service_instance)[0],
decompose_job_id(self.fake_args.service_instance)[1],
self.fake_cluster,
soa_dir=self.fake_args.soa_dir,
)
setup_service_patch.assert_called_once_with(
decompose_job_id(self.fake_args.service_instance)[0],
decompose_job_id(self.fake_args.service_instance)[1],
fake_client,
self.fake_marathon_config,
self.fake_marathon_service_config,
'no_more',
)
sys_exit_patch.assert_called_once_with(0)
def test_main_sends_event_if_no_deployments(self):
fake_client = mock.MagicMock()
with contextlib.nested(
mock.patch(
'setup_marathon_job.parse_args',
return_value=self.fake_args,
autospec=True,
),
mock.patch(
'setup_marathon_job.get_main_marathon_config',
return_value=self.fake_marathon_config,
autospec=True,
),
mock.patch(
'paasta_tools.marathon_tools.get_marathon_client',
return_value=fake_client,
autospec=True,
),
mock.patch(
'paasta_tools.marathon_tools.load_marathon_service_config',
side_effect=NoDeploymentsAvailable(),
autospec=True,
),
mock.patch(
'setup_marathon_job.setup_service',
return_value=(1, 'NEVER'),
autospec=True,
),
mock.patch('setup_marathon_job.load_system_paasta_config', autospec=True),
mock.patch('setup_marathon_job.send_event', autospec=True),
) as (
parse_args_patch,
get_main_conf_patch,
get_client_patch,
read_service_conf_patch,
setup_service_patch,
load_system_paasta_config_patch,
sensu_patch,
):
load_system_paasta_config_patch.return_value.get_cluster = mock.Mock(return_value=self.fake_cluster)
with raises(SystemExit) as exc_info:
setup_marathon_job.main()
parse_args_patch.assert_called_once_with()
get_main_conf_patch.assert_called_once_with()
get_client_patch.assert_called_once_with(
self.fake_marathon_config.get_url(),
self.fake_marathon_config.get_username(),
self.fake_marathon_config.get_password())
read_service_conf_patch.assert_called_once_with(
decompose_job_id(self.fake_args.service_instance)[0],
decompose_job_id(self.fake_args.service_instance)[1],
self.fake_cluster,
soa_dir=self.fake_args.soa_dir)
expected_string = 'No deployments found for %s in cluster %s' % (
self.fake_args.service_instance, self.fake_cluster)
sensu_patch.assert_called_once_with(
decompose_job_id(self.fake_args.service_instance)[0],
decompose_job_id(self.fake_args.service_instance)[1],
self.fake_args.soa_dir,
Status.CRITICAL,
expected_string
)
assert exc_info.value.code == 0