def test_v1_sampled_lists(simple_metric):
simple_metric.max_list_size = 10
t = metrics.Tags()
for i in range(1, 20):
simple_metric.add_network_stats(1, 1, 1, 1)
simple_metric.add_network_connection("a", i, "a", i)
simple_metric.add_network_connection("a", i, "a", i)
simple_metric.add_listening_ports("TCP", [{'port': i}])
simple_metric.add_listening_ports("UDP", [{'port': i}])
report = simple_metric._v1_metrics()
metric_block = report[t.metrics]
assert len(metric_block[t.tcp_conn][t.established_connections][
t.connections]) == 10
assert metric_block[t.tcp_conn][t.established_connections][t.total] > len(
metric_block[t.tcp_conn][t.established_connections])
assert len(metric_block[t.listening_tcp_ports][t.ports]) == 10
assert metric_block[t.listening_tcp_ports][t.total] > len(
metric_block[t.listening_tcp_ports][t.ports])
assert len(metric_block[t.listening_udp_ports][t.ports]) == 10
assert metric_block[t.listening_udp_ports][t.total] > len(
metric_block[t.listening_udp_ports][t.ports])
def test_v1_metrics_optional_elements(simple_metric):
t = metrics.Tags(short_names=False)
report = simple_metric._v1_metrics()
metric_block = report[t.metrics]
# top-level element in metrics block
assert t.interface_stats in metric_block
assert t.listening_tcp_ports in metric_block
assert t.listening_udp_ports in metric_block
assert t.tcp_conn in metric_block
assert t.listening_udp_ports in report[t.metrics]
simple_metric._interface_stats = {}
simple_metric.listening_tcp_ports = []
simple_metric.listening_udp_ports = []
simple_metric._net_connections = []
report = simple_metric._v1_metrics()
metric_block = report[t.metrics]
assert t.interface_stats not in metric_block
assert t.listening_tcp_ports not in metric_block
assert t.listening_udp_ports not in metric_block
assert t.tcp_conn not in metric_block
def test_sampled_lists(simple_metric):
t = metrics.Tags()
m = metrics.Metrics(client_id="FAKE_CLIENT_ID")
m.max_list_size = 10
for i in range(1, 20):
m.add_network_connection("10.10.10.10", i)
m.listening_udp_ports.append(i)
m.listening_tcp_ports.append(i)
report = m._v1_metrics()
metric_block = report[t.metrics]
assert len(metric_block[t.listening_tcp_ports][t.ports]) == 10
assert len(metric_block[t.listening_udp_ports][t.ports]) == 10
assert len(metric_block[t.tcp_conn][t.connections]) == 10
def test_sampled_lists(simple_metric):
t = metrics.Tags()
m = metrics.Metrics()
m.max_list_size = 10
for i in range(1, 20):
m.add_network_connection("10.10.10." + str(i), i, "eth0", 99999)
m.listening_udp_ports.append(i)
m.listening_tcp_ports.append(i)
report = m._v1_metrics()
metric_block = report[t.metrics]
assert len(metric_block[t.listening_tcp_ports][t.ports]) == 10
assert len(metric_block[t.listening_udp_ports][t.ports]) == 10
assert len(metric_block[t.tcp_conn][t.established_connections][t.connections]) == 10
def test_v1_metrics_basic_structure_long_names(simple_metric):
t = metrics.Tags()
report = simple_metric._v1_metrics()
# overall structure
assert t.header in report
assert t.metrics in report
# header elements
header_block = report[t.header]
assert t.report_id in header_block
assert t.version in header_block
metric_block = report[t.metrics]
# top-level element in metrics block
assert t.interface_stats in metric_block
assert t.listening_tcp_ports in metric_block
assert t.listening_udp_ports in metric_block
assert t.tcp_conn in metric_block
def test_v1_sampled_lists(simple_metric):
simple_metric.max_list_size = 10
t = metrics.Tags()
for i in range(1, 20):
simple_metric.add_network_stats(1, 1, 1, 1)
simple_metric.add_network_connection("a", "1")
simple_metric.add_network_connection("a", "1")
simple_metric.listening_tcp_ports.append(i)
simple_metric.listening_udp_ports.append(i)
report = simple_metric._v1_metrics()
metric_block = report[t.metrics]
assert len(metric_block[t.tcp_conn][t.connections]) == 10
assert metric_block[t.tcp_conn][t.total] > len(
metric_block[t.tcp_conn][t.connections])
assert len(metric_block[t.listening_tcp_ports][t.ports]) == 10
assert metric_block[t.listening_tcp_ports][t.total] > len(
metric_block[t.listening_tcp_ports][t.ports])
assert len(metric_block[t.listening_udp_ports][t.ports]) == 10
assert metric_block[t.listening_udp_ports][t.total] > len(
metric_block[t.listening_udp_ports][t.ports])
def test_tags():
t_l = metrics.Tags(False)
t_s = metrics.Tags(True)
