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"));
+    }
+
 }