def test_get_host_metrics(instance):
ambari = AmbariCheck(instances=[instance])
ambari._get_hosts_info = MagicMock(return_value=responses.HOSTS_INFO)
ambari._submit_gauge = MagicMock()
ambari.set_external_tags = MagicMock()
cluster_tag = ['ambari_cluster:cluster1']
ambari.get_host_metrics('localhost', ['cluster1'])
ambari.set_external_tags.assert_called_with([('my_host_1', {
'ambari': cluster_tag
}), ('my_host_2', {
'ambari': cluster_tag
})])
assert ambari._submit_gauge.call_count == 30
ambari._submit_gauge.assert_has_calls(
[
call('boottime', 1555934503.0, cluster_tag, 'my_host_2'),
call('cpu.cpu_idle', 62.8, cluster_tag, 'my_host_2'),
call('cpu.cpu_nice', 0.0, cluster_tag, 'my_host_2'),
call('cpu.cpu_num', 4.0, cluster_tag, 'my_host_2'),
call('cpu.cpu_system', 5.1, cluster_tag, 'my_host_2'),
call('cpu.cpu_user', 32.0, cluster_tag, 'my_host_2'),
call('cpu.cpu_wio', 0.0, cluster_tag, 'my_host_2'),
call('disk.disk_free', 124.35, cluster_tag, 'my_host_2'),
call('disk.disk_total', 148.29, cluster_tag, 'my_host_2'),
call('disk.read_bytes', 1594053632.0, cluster_tag, 'my_host_2'),
call('disk.read_count', 42717.0, cluster_tag, 'my_host_2'),
call('disk.read_time', 240986.0, cluster_tag, 'my_host_2'),
call('disk.write_bytes', 117000843264.0, cluster_tag, 'my_host_2'),
call('disk.write_count', 499318.0, cluster_tag, 'my_host_2'),
call('disk.write_time', 5946304.0, cluster_tag, 'my_host_2'),
call('load.load_fifteen', 0.99, cluster_tag, 'my_host_2'),
call('load.load_five', 1.35, cluster_tag, 'my_host_2'),
call('load.load_one', 0.57, cluster_tag, 'my_host_2'),
call('memory.mem_cached', 3554248.0, cluster_tag, 'my_host_2'),
call('memory.mem_free', 11327848.0, cluster_tag, 'my_host_2'),
call('memory.mem_shared', 0.0, cluster_tag, 'my_host_2'),
call('memory.mem_total', 15399208.0, cluster_tag, 'my_host_2'),
call('memory.swap_free', 0.0, cluster_tag, 'my_host_2'),
call('memory.swap_total', 0.0, cluster_tag, 'my_host_2'),
call('network.bytes_in', 683.2346950556641, cluster_tag,
'my_host_2'),
call('network.bytes_out', 12517.203580542699, cluster_tag,
'my_host_2'),
call('network.pkts_in', 8.499187630576825, cluster_tag,
'my_host_2'),
call('network.pkts_out', 10.498996484830196, cluster_tag,
'my_host_2'),
call('process.proc_run', 0.0, cluster_tag, 'my_host_2'),
call('process.proc_total', 128.0, cluster_tag, 'my_host_2'),
],
any_order=True,
)
def test_get_host_metrics(instance):
ambari = AmbariCheck(instances=[instance])
ambari._get_hosts_info = MagicMock(return_value=responses.HOSTS_INFO)
ambari._submit_gauge = MagicMock()
ambari.set_external_tags = MagicMock()
ambari.get_host_metrics('localhost', ['cluster1'], [])
ambari.set_external_tags.assert_called_with([('my_host_1', {
'ambari': ['ambari_cluster:cluster1']
}), ('my_host_2', {
'ambari': ['ambari_cluster:cluster1']
})])
assert ambari._submit_gauge.call_count == 30
def test_get_component_metrics(init_config, instance):
ambari = AmbariCheck(init_config=init_config, instances=[instance])
ambari._make_request = MagicMock(return_value=responses.COMPONENT_METRICS)
ambari._submit_gauge = MagicMock()
namenode_tags = [
'ambari_cluster:LabCluster', 'ambari_service:hdfs',
'ambari_component:namenode'
]
ambari.get_component_metrics(
'localhost',
'LabCluster',
'HDFS',
base_tags=['ambari_cluster:LabCluster', 'ambari_service:hdfs'],
component_whitelist=['NAMENODE'],
metric_whitelist=['cpu'],
)
ambari._make_request.assert_called_with(
'localhost/api/v1/clusters/LabCluster/services/HDFS/components?fields=metrics'
)
ambari._submit_gauge.assert_has_calls(
[
call('cpu.cpu_idle', 90.3, namenode_tags),
call('cpu.cpu_idle._avg', 90.3, namenode_tags),
call('cpu.cpu_idle._max', 90.3, namenode_tags),
call('cpu.cpu_idle._min', 90.3, namenode_tags),
call('cpu.cpu_idle._sum', 90.3, namenode_tags),
call('cpu.cpu_nice', 0.0, namenode_tags),
call('cpu.cpu_nice._avg', 0.0, namenode_tags),
call('cpu.cpu_nice._max', 0.0, namenode_tags),
call('cpu.cpu_nice._min', 0.0, namenode_tags),
call('cpu.cpu_nice._sum', 0.0, namenode_tags),
call('cpu.cpu_system', 1.6333333333333335, namenode_tags),
call('cpu.cpu_system._avg', 1.6333333333333335, namenode_tags),
call('cpu.cpu_system._max', 1.6333333333333335, namenode_tags),
call('cpu.cpu_system._min', 1.6333333333333335, namenode_tags),
call('cpu.cpu_system._sum', 1.6333333333333335, namenode_tags),
call('cpu.cpu_user', 8.033333333333333, namenode_tags),
call('cpu.cpu_user._avg', 8.033333333333333, namenode_tags),
call('cpu.cpu_user._max', 8.033333333333333, namenode_tags),
call('cpu.cpu_user._min', 8.033333333333333, namenode_tags),
call('cpu.cpu_user._sum', 8.033333333333333, namenode_tags),
call('cpu.cpu_wio', 0.0, namenode_tags),
call('cpu.cpu_wio._avg', 0.0, namenode_tags),
call('cpu.cpu_wio._max', 0.0, namenode_tags),
call('cpu.cpu_wio._min', 0.0, namenode_tags),
call('cpu.cpu_wio._sum', 0.0, namenode_tags),
],
any_order=True,
)