Add node's role to json output.

2 files changed, 4 insertions, 0 deletions

diff --git a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/application/ApplicationMetricsHandlerTest.java b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/application/ApplicationMetricsHandlerTest.java
index 959b73d5682..074b7877430 100644
--- a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/application/ApplicationMetricsHandlerTest.java
+++ b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/application/ApplicationMetricsHandlerTest.java
@@ -119,6 +119,7 @@ public class ApplicationMetricsHandlerTest {
         assertEquals(1, jsonModel.nodes.size());
         GenericJsonModel nodeModel = jsonModel.nodes.get(0);
         assertEquals(HOST, nodeModel.hostname);
+        assertEquals(MOCK_METRICS_PATH, nodeModel.role);
         assertEquals(2, nodeModel.node.metrics.size());
         assertEquals(16.222, nodeModel.node.metrics.get(0).values.get(CPU_METRIC), 0.0001d);
     }
diff --git a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/metric/model/json/GenericApplicationModelTest.java b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/metric/model/json/GenericApplicationModelTest.java
index 73b6c014b92..5a2d374af2e 100644
--- a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/metric/model/json/GenericApplicationModelTest.java
+++ b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/metric/model/json/GenericApplicationModelTest.java
@@ -34,6 +34,7 @@ public class GenericApplicationModelTest {
         assertEquals(2, model.nodes.size());
         GenericJsonModel node0Model = model.nodes.get(0);
         assertEquals("node0", node0Model.hostname);
+        assertEquals("role0", node0Model.role);
         assertEquals(1, node0Model.services.size());
         GenericService service = node0Model.services.get(0);
         assertEquals(1, service.metrics.size());
@@ -42,6 +43,7 @@ public class GenericApplicationModelTest {
         GenericJsonModel node1Model = model.nodes.get(1);
         GenericNode node1 = node1Model.node;
         assertEquals("node1", node1Model.hostname);
+        assertEquals("role1", node1Model.role);
         assertEquals(32.444, node1.metrics.get(0).values.get("cpu.util"), 0.001d);
 
         assertThatSerializedModelEqualsTestFile(model);
@@ -71,6 +73,7 @@ public class GenericApplicationModelTest {
         GenericJsonModel nodeModel = model.nodes.get(0);
         assertNotNull(nodeModel.node);
         assertEquals("hostname", nodeModel.hostname);
+        assertEquals("my-role", nodeModel.role);
         assertEquals(1, nodeModel.node.metrics.size());
         GenericMetrics nodeMetrics = nodeModel.node.metrics.get(0);
         assertEquals(1.234, nodeMetrics.values.get("node-metric"), 0.001d);