def test_gauge(self, telemetry: TelemetryFixture):
telemetry.gauge(
"category1", "gauge1",
lambda observer: observer.observe(10, {'label1': 'label1'}))
telemetry.gauge(
"category1", "gauge2",
lambda observer: observer.observe(1.2, {'label1': 'label1'}))
telemetry.collect()
assert telemetry.get_gauge('category1.gauge1', {
'label1': 'label1'
}).count == 1
assert telemetry.get_gauge('category1.gauge1', {
'label1': 'label1'
}).min == 10
assert telemetry.get_gauge('category1.gauge1', {
'label1': 'label1'
}).max == 10
assert telemetry.get_gauge('category1.gauge1', {
'label1': 'label1'
}).last == 10
assert telemetry.get_gauge('category1.gauge1', {
'label1': 'label1'
}).sum == 10
assert telemetry.get_gauge('category1.gauge2', {
'label1': 'label1'
}).count == 1
assert telemetry.get_gauge('category1.gauge2', {
'label1': 'label1'
}).min == 1.2
assert telemetry.get_gauge('category1.gauge2', {
'label1': 'label1'
}).max == 1.2
assert telemetry.get_gauge('category1.gauge2', {
'label1': 'label1'
}).last == 1.2
assert telemetry.get_gauge('category1.gauge2', {
'label1': 'label1'
}).sum == 1.2
def test_http_server(self, monkeypatch, telemetry: TelemetryFixture):
address = 'localhost:19102'
monkeypatch.setenv('METRICS_EXPORTERS', 'prometheus')
monkeypatch.setenv('METRICS_PROMETHEUS_PREFIX', 'test_prefix')
monkeypatch.setenv('METRICS_INTERVAL', '5')
monkeypatch.setenv('METRICS_PROMETHEUS_BIND_ADDRESS', address)
telemetry.initialize()
http = urllib3.PoolManager()
with telemetry.span("category1",
"span1",
attributes={
TestAttributes.ATTRIB1: "attrib1",
TestAttributes.LABEL1: 'label1'
}) as span:
time.sleep(.5)
telemetry.counter("category1", "counter1", 2.0)
def gauge(obv: Observer):
obv.observe(1, {Attributes.ENV: 'test'})
telemetry.gauge("category1", "gauge", gauge)
# wait for Prometheus collection interval to pass (METRICS_INTERVAL)
time.sleep(5)
telemetry.collect()
response = http.request('GET', 'http://localhost:19102/metrics')
def fetch_metric(name: str, labels: dict = {}):
response = http.request('GET', 'http://localhost:19102/metrics')
lines = response.data.decode('utf8').split('\n')
matches = list(
filter(lambda line: not line.startswith("#") and name in line,
lines))
if len(matches) == 0:
pytest.fail(f"Metric not found: {name}")
elif len(matches) == 1:
return float(matches[0].split(' ')[1])
else:
pytest.fail(f"More than one match for metric: {name}")
assert fetch_metric('test_prefix_trace_duration_count') == 1.0
assert fetch_metric('test_prefix_trace_duration_sum') >= 500
assert fetch_metric('test_prefix_category1_gauge') == 1.0
# double-check that metrics continue to be returned on duplicate fetches
assert fetch_metric('test_prefix_trace_duration_count') == 1.0
assert fetch_metric('test_prefix_trace_duration_sum') >= 500
with telemetry.span("category1",
"span1",
attributes={
TestAttributes.ATTRIB1: "attrib1",
TestAttributes.LABEL1: 'label1'
}) as span:
time.sleep(.5)
# wait for Prometheus collection interval to pass (METRICS_INTERVAL)
time.sleep(2)
telemetry.collect()
assert fetch_metric('test_prefix_trace_duration_count') == 2.0
assert fetch_metric('test_prefix_trace_duration_sum') >= 1000
telemetry.shutdown()