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)