def test_register_gauge(self, registry: CollectorRegistry):
gauge = Gauge(name="temperature_celsius",
description="Temperature in celsius")
registry.register(gauge)
assert registry.metrics[gauge.name] == gauge
assert registry.metrics[gauge.name].registry == registry
def test_register_counter(self, registry: CollectorRegistry):
counter = Counter(name="http_requests_total",
description="Total HTTP Requests")
registry.register(counter)
assert registry.metrics[counter.name] == counter
assert registry.metrics[counter.name].registry == registry
def gauge(registry: instrumentor.CollectorRegistry):
gauge = instrumentor.Gauge(
name="temperature_celsius",
description="Temperature Celsius",
allowed_labels=["location", "id"],
)
registry.register(gauge)
return gauge
def test_update_buffer(self, registry: CollectorRegistry):
test_key = "test"
test_value = 1
test_action = UpdateAction(key=test_key, value=test_value, set=False)
to_update = [test_action]
registry.update_buffer(to_update)
assert registry.buffer[test_key] == test_action
def histogram(registry: instrumentor.CollectorRegistry):
histogram = instrumentor.Histogram(
name="http_response_time_seconds",
description="HTTP Response Time in seconds.",
buckets=[0.1, 0.2, 0.4, 0.8, 1.6],
allowed_labels=["code"],
)
registry.register(histogram)
return histogram
def test_eager(self, registry: CollectorRegistry, redis: StrictRedis):
counter = Counter(name="http_requests_total",
description="Total HTTP Requests")
registry.eager = True
registry.register(counter)
assert registry.eager
counter.inc()
assert redis.hgetall("testing") != {}
def test_unregister_gauge(self, registry: CollectorRegistry):
gauge = Gauge(name="temperature_celsius",
description="Temperature Celsius")
registry.register(gauge)
assert registry.metrics[gauge.name] == gauge
assert registry.metrics[gauge.name].registry == registry
registry.unregister(gauge)
with pytest.raises(KeyError):
x = registry.metrics[gauge.name]
assert gauge.registry is None
def test_unregister_counter(self, registry: CollectorRegistry):
counter = Counter(name="http_requests_total",
description="Total HTTP Requests")
registry.register(counter)
assert registry.metrics[counter.name] == counter
assert registry.metrics[counter.name].registry == registry
registry.unregister(counter)
with pytest.raises(KeyError):
x = registry.metrics[counter.name]
assert counter.registry is None
def counter(registry: instrumentor.CollectorRegistry):
counter = instrumentor.Counter(name="http_total_requests",
description="Test",
allowed_labels=["code", "path"])
registry.register(counter)
return counter