def test_check_service_replication_for_namespace_with_no_deployments():
service = 'test_service'
instance = 'worker'
cluster = 'fake_cluster'
with contextlib.nested(
mock.patch(
'paasta_tools.marathon_tools.get_proxy_port_for_instance',
autospec=True,
return_value=None),
mock.patch(
'paasta_tools.marathon_tools.get_expected_instance_count_for_namespace',
autospec=True),
) as (
mock_get_proxy_port_for_instance,
mock_get_expected_count,
):
mock_client = mock.Mock()
mock_get_expected_count.side_effect = check_marathon_services_replication.NoDeploymentsAvailable
check_marathon_services_replication.check_service_replication(
client=mock_client,
service=service,
instance=instance,
cluster=cluster,
crit_threshold=None,
soa_dir=None)
assert mock_get_proxy_port_for_instance.call_count == 0
def test_check_service_replication_for_non_smartstack():
service = 'test_service'
instance = 'worker'
cluster = 'fake_cluster'
with contextlib.nested(
mock.patch('paasta_tools.marathon_tools.get_proxy_port_for_instance', autospec=True, return_value=None),
mock.patch('paasta_tools.marathon_tools.get_expected_instance_count_for_namespace',
autospec=True, return_value=100),
mock.patch('check_marathon_services_replication.check_healthy_marathon_tasks_for_service_instance',
autospec=True),
) as (
mock_get_proxy_port_for_instance,
mock_get_expected_count,
mock_check_healthy_marathon_tasks,
):
mock_client = mock.Mock()
check_marathon_services_replication.check_service_replication(
client=mock_client, service=service, instance=instance, cluster=cluster, crit_threshold=None, soa_dir=None)
mock_check_healthy_marathon_tasks.assert_called_once_with(
client=mock_client,
service=service,
instance=instance,
cluster=cluster,
soa_dir=None,
crit_threshold=None,
expected_count=100)
def test_check_service_replication_for_normal_smartstack():
service = 'test_service'
instance = 'test_instance'
cluster = 'fake_cluster'
with contextlib.nested(
mock.patch(
'paasta_tools.marathon_tools.get_proxy_port_for_instance',
autospec=True,
return_value=666),
mock.patch(
'paasta_tools.marathon_tools.get_expected_instance_count_for_namespace',
autospec=True,
return_value=100),
mock.patch(
'check_marathon_services_replication.check_smartstack_replication_for_instance',
autospec=True),
) as (mock_get_proxy_port_for_instance, mock_get_expected_count,
mock_check_smartstack_replication_for_service):
mock_client = mock.Mock()
check_marathon_services_replication.check_service_replication(
client=mock_client,
service=service,
instance=instance,
cluster=cluster,
crit_threshold=None,
soa_dir=None)
mock_check_smartstack_replication_for_service.assert_called_once_with(
service=service,
instance=instance,
cluster=cluster,
soa_dir=None,
crit_threshold=None,
expected_count=100,
)
def test_check_service_replication_for_namespace_with_no_deployments():
service = 'test_service'
instance = 'worker'
cluster = 'fake_cluster'
with contextlib.nested(
mock.patch('paasta_tools.marathon_tools.get_proxy_port_for_instance', autospec=True, return_value=None),
mock.patch('paasta_tools.marathon_tools.get_expected_instance_count_for_namespace',
autospec=True),
) as (
mock_get_proxy_port_for_instance,
mock_get_expected_count,
):
mock_client = mock.Mock()
mock_get_expected_count.side_effect = check_marathon_services_replication.NoDeploymentsAvailable
check_marathon_services_replication.check_service_replication(
client=mock_client, service=service, instance=instance, cluster=cluster, crit_threshold=None, soa_dir=None)
assert mock_get_proxy_port_for_instance.call_count == 0
