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()