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