def test_cant_connect(init_config, instance, aggregator):
ambari = AmbariCheck(init_config=init_config, instances=[instance])
ambari._make_request = MagicMock(return_value=None)
try:
ambari.get_clusters('localhost')
except CheckException:
pass
aggregator.assert_service_check(name="ambari.can_connect",
status=AgentCheck.CRITICAL)
def test_cant_connect(init_config, instance, aggregator):
instance['tags'] = ['foo:bar']
ambari = AmbariCheck('Ambari', init_config, [instance])
ambari._make_request = MagicMock(return_value=None)
try:
ambari.get_clusters()
except CheckException:
pass
aggregator.assert_service_check(
name="ambari.can_connect", status=AgentCheck.CRITICAL, tags=['foo:bar', 'url:localhost']
)
def test_get_clusters(init_config, instance, aggregator):
ambari = AmbariCheck(init_config=init_config, instances=[instance])
ambari._make_request = MagicMock(
return_value={
'href': 'localhost/api/v1/clusters',
'items': [{'href': 'localhost/api/v1/clusters/LabCluster', 'Clusters': {'cluster_name': 'LabCluster'}}],
}
)
clusters = ambari.get_clusters('localhost', ['foo:bar'])
ambari._make_request.assert_called_with('localhost/api/v1/clusters')
aggregator.assert_service_check(name="ambari.can_connect", status=AgentCheck.OK, tags=['foo:bar', 'url:localhost'])
assert clusters == ['LabCluster']
def test_get_clusters(init_config, instance):
ambari = AmbariCheck(init_config=init_config, instances=[instance])
ambari._make_request = MagicMock(
return_value={
'href':
'localhost/api/v1/clusters',
'items': [{
'href': 'localhost/api/v1/clusters/LabCluster',
'Clusters': {
'cluster_name': 'LabCluster'
}
}],
})
clusters = ambari.get_clusters('localhost')
ambari._make_request.assert_called_with('localhost/api/v1/clusters')
assert clusters == ['LabCluster']