def test_send_event_if_under_replication_critical():
service = 'test_service'
instance = 'worker'
cluster = 'fake_cluster'
crit = 90
expected_count = 100
available = 89
soa_dir = '/dne'
with contextlib.nested(
mock.patch('check_marathon_services_replication.send_event', autospec=True),
mock.patch('check_marathon_services_replication.get_context', autospec=True),
) as (
mock_send_event,
mock_get_context,
):
check_marathon_services_replication.send_event_if_under_replication(
service=service,
instance=instance,
cluster=cluster,
crit_threshold=crit,
expected_count=expected_count,
num_available=available,
soa_dir=soa_dir)
mock_send_event.assert_called_once_with(
service=service,
namespace=instance,
cluster=cluster,
soa_dir=soa_dir,
status=2,
output=mock.ANY)
def test_send_event_if_under_replication_critical():
service = 'test_service'
instance = 'worker'
cluster = 'fake_cluster'
crit = 90
expected_count = 100
available = 89
soa_dir = '/dne'
with contextlib.nested(
mock.patch('check_marathon_services_replication.send_event',
autospec=True), ) as (mock_send_event, ):
check_marathon_services_replication.send_event_if_under_replication(
service=service,
instance=instance,
cluster=cluster,
crit_threshold=crit,
expected_count=expected_count,
num_available=available,
soa_dir=soa_dir)
mock_send_event.assert_called_once_with(service=service,
namespace=instance,
cluster=cluster,
soa_dir=soa_dir,
status=2,
output=mock.ANY)
def test_send_event_if_under_replication_handles_0_expected():
service = 'test_service'
instance = 'worker'
cluster = 'fake_cluster'
crit = 90
expected_count = 0
available = 0
soa_dir = '/dne'
with contextlib.nested(
mock.patch('check_marathon_services_replication.send_event', autospec=True),
) as (
mock_send_event,
):
check_marathon_services_replication.send_event_if_under_replication(
service, instance, cluster, crit, expected_count, available, soa_dir)
mock_send_event.assert_called_once_with(
service=service,
namespace=instance,
cluster=cluster,
soa_dir=soa_dir,
status=0,
output=mock.ANY)