diff options
author | Ola Aunronning <olaa@yahooinc.com> | 2023-10-27 15:57:13 +0200 |
---|---|---|
committer | Ola Aunronning <olaa@yahooinc.com> | 2023-10-27 15:57:13 +0200 |
commit | c02490fd4e205f43bfdc6cdbf72f9ab9853b6c76 (patch) | |
tree | 6a643f58e7adc7230ef1746d2160ab75680eb85a /container-core/src/test/java/com/yahoo | |
parent | 026305ca0de9f2b446e62d85c153df6324e0a0ff (diff) |
Allow filtering prometheus metrics
Diffstat (limited to 'container-core/src/test/java/com/yahoo')
-rw-r--r-- | container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java | 25 |
1 files changed, 16 insertions, 9 deletions
diff --git a/container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java b/container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java index 807e58918da..160d6452473 100644 --- a/container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java +++ b/container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java @@ -7,6 +7,8 @@ import com.yahoo.container.jdisc.RequestHandlerTestDriver; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; +import java.nio.file.Files; +import java.nio.file.Path; import java.util.ArrayList; import java.util.List; import java.util.Map; @@ -146,21 +148,22 @@ public class MetricsPacketsHandlerTest extends StateHandlerTestBase { } @Test - public void prometheus_metrics() { + public void prometheus_metrics() throws Exception { var context = StateMetricContext.newInstance(Map.of("dim-1", "value1")); var snapshot = new MetricSnapshot(); snapshot.set(context, "gauge.metric", 0.2); snapshot.add(context, "counter.metric", 5); + snapshot.add(context, "configserver.requests", 120); + // Infrastructure set only contains max and average + snapshot.set(context, "lockAttempt.lockedLoad", 500); snapshotProvider.setSnapshot(snapshot); + var response = requestAsString("http://localhost/metrics-packets?format=prometheus"); - var expectedResponse = """ - # HELP gauge_metric_last\s - # TYPE gauge_metric_last untyped - gauge_metric_last{dim_1="value1",vespa_service="state-handler-test-base",} 0.2 0 - # HELP counter_metric_count\s - # TYPE counter_metric_count untyped - counter_metric_count{dim_1="value1",vespa_service="state-handler-test-base",} 5 0 - """; + var expectedResponse = readFile("prometheus-unfiltered"); + assertEquals(expectedResponse, response); + + response = requestAsString("http://localhost/metrics-packets?format=prometheus&metric-set=infrastructure"); + expectedResponse = readFile("prometheus-filtered"); assertEquals(expectedResponse, response); } @@ -262,4 +265,8 @@ public class MetricsPacketsHandlerTest extends StateHandlerTestBase { snapshotProvider.setSnapshot(snapshot); } + private String readFile(String fileName) throws Exception { + return Files.readString(Path.of("src/test/resources/metrics-packets-handler-responses/" + fileName + ".txt")); + } + } |