def test_should_not_collect_if_disabled(instance):
ambari = AmbariCheck('Ambari', {'collect_service_metrics': False, 'collect_service_status': False}, [instance])
_mock_clusters(ambari)
ambari.get_host_metrics = MagicMock()
ambari.get_service_status_and_metrics = MagicMock()
ambari.check(instance)
assert not ambari.get_service_status_and_metrics.called
def test_should_collect_host_metrics(instance):
ambari = AmbariCheck(
init_config={'collect_service_metrics': False, 'collect_service_status': False}, instances=[instance]
)
_mock_clusters(ambari)
ambari.get_host_metrics = MagicMock()
ambari.get_service_status_and_metrics = MagicMock()
ambari.check(instance)
assert ambari.get_host_metrics.called
assert not ambari.get_service_status_and_metrics.called
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
def test_check(aggregator):
ambari_ip = "localhost"
init_config = {"collect_host_metrics": True, "collect_service_metrics": True, "collect_service_status": True}
instances = [
{
"url": "https://{}:8443/ambari-lab-2/dp-proxy/ambari".format(ambari_ip),
"username": "******",
"password": "******",
"tags": ["test:manual"],
"services": {
"HDFS": {"NAMENODE": [], "DATANODE": []},
"YARN": {"NODEMANANGER": ["cpu", "disk", "load", "memory", "network", "process"], "YARNCLIENT": []},
"MAPREDUCE2": {"HISTORYSERVER": ["BufferPool", "Memory", "jvm"]},
},
"timeout": 30,
"tls_verify": False,
}
]
check = AmbariCheck(init_config=init_config, instances=instances)
check.check(instances[0])
aggregator.assert_all_metrics_covered()
def test_check(aggregator):
init_config = {
"collect_host_metrics": True,
"collect_service_metrics": True,
"collect_service_status": True
}
instances = [{
"url": "https://${ambari_ip}:8443/ambari-lab-2/dp-proxy/ambari",
"username": "******",
"password": "******",
"tags": ["test:manual"],
"services": {
"HDFS": ["NAMENODE", "DATANODE"],
"YARN": ["NODEMANANGER", "YARNCLIENT"],
"SPARK": []
},
"metric_headers": ["cpu", "jvm"],
"timeout": 30,
"tls_verify": False,
}]
check = AmbariCheck(init_config=init_config, instances=instances)
check.check(instances[0])
# import pdb; pdb.set_trace()
aggregator.assert_all_metrics_covered()