def test_get_service_health_no_response(init_config, instance, aggregator):
ambari = AmbariCheck('Ambari', init_config, [instance])
ambari._make_request = MagicMock(return_value=None)
ambari.get_service_checks('LabCluster', 'HDFS', service_tags=['ambari_cluster:LabCluster', 'ambari_service:hdfs'])
ambari._make_request.assert_called_with('localhost/api/v1/clusters/LabCluster/services/HDFS?fields=ServiceInfo')
aggregator.assert_service_check(
name="ambari.state", status=AgentCheck.CRITICAL, tags=['ambari_cluster:LabCluster', 'ambari_service:hdfs']
)
def test_get_service_health(init_config, instance, aggregator):
ambari = AmbariCheck('Ambari', init_config, [instance])
ambari._make_request = MagicMock(return_value=responses.SERVICE_HEALTH_METRICS)
ambari.get_service_checks('LabCluster', 'HDFS', service_tags=['ambari_cluster:LabCluster', 'ambari_service:hdfs'])
ambari._make_request.assert_called_with('localhost/api/v1/clusters/LabCluster/services/HDFS?fields=ServiceInfo')
aggregator.assert_service_check(
name="ambari.state",
status=AgentCheck.OK,
tags=['ambari_cluster:LabCluster', 'ambari_service:hdfs', 'state:INSTALLED'],
)
def test_should_collect_service_status(instance):
ambari = AmbariCheck('Ambari', {'collect_service_metrics': False, 'collect_service_status': True}, [instance])
_mock_clusters(ambari)
ambari.get_host_metrics = MagicMock()
ambari.get_component_metrics = MagicMock()
ambari.get_service_checks = MagicMock()
ambari.check(instance)
assert ambari.get_host_metrics.called
assert not ambari.get_component_metrics.called
assert ambari.get_service_checks.called