1 files changed, 53 insertions, 2 deletions
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/admin/otel/OpenTelemetryConfigGeneratorTest.java b/config-model/src/test/java/com/yahoo/vespa/model/admin/otel/OpenTelemetryConfigGeneratorTest.java
index 7c4968aac84..c24fcb27dc9 100644
--- a/config-model/src/test/java/com/yahoo/vespa/model/admin/otel/OpenTelemetryConfigGeneratorTest.java
+++ b/config-model/src/test/java/com/yahoo/vespa/model/admin/otel/OpenTelemetryConfigGeneratorTest.java
@@ -2,8 +2,24 @@
 package com.yahoo.vespa.model.admin.otel;
 
 import com.yahoo.config.model.ApplicationConfigProducerRoot.StatePortInfo;
+import com.yahoo.config.model.producer.TreeConfigProducer;
+import com.yahoo.config.model.test.MockRoot;
+import com.yahoo.config.provision.ApplicationId;
+import com.yahoo.config.provision.ClusterMembership;
+import com.yahoo.config.provision.Environment;
+import com.yahoo.config.provision.HostSpec;
+import com.yahoo.config.provision.NodeResources;
+import com.yahoo.config.provision.RegionName;
+import com.yahoo.config.provision.SystemName;
+import com.yahoo.config.provision.Zone;
+import com.yahoo.vespa.model.AbstractService;
+import com.yahoo.vespa.model.Host;
+import com.yahoo.vespa.model.HostResource;
+import com.yahoo.vespa.model.AbstractService;
+import com.yahoo.vespa.model.PortAllocBridge;
 import org.junit.jupiter.api.Test;
 import java.util.List;
+import java.util.Optional;
 import static org.junit.jupiter.api.Assertions.*;
 
 /**
@@ -13,10 +29,45 @@ public class OpenTelemetryConfigGeneratorTest {
 
     @Test
     void testBuildsYaml() {
-        var generator = new OpenTelemetryConfigGenerator(null);
-        generator.addStatePorts(List.of(new StatePortInfo("localhost", 19098, "config-sentinel", "sentinel")));
+        var mockZone = new Zone(SystemName.PublicCd, Environment.prod, RegionName.from("mock"));
+        var app = ApplicationId.from("mytenant", "myapp", "myinstance");
+        var generator = new OpenTelemetryConfigGenerator(mockZone, app);
+        var root = new MockRoot();
+
+        var mockHost = new Host(root, "localhost2.local");
+        var mockVersion = new com.yahoo.component.Version(8);
+        var mockCluster = ClusterMembership.from("container/feeding/2/3", mockVersion, Optional.empty());
+        var noResource = NodeResources.unspecified();
+        var mockHostSpec = new HostSpec("localhost1.local",
+                                        noResource, noResource, noResource,
+                                        mockCluster,
+                                        Optional.empty(), Optional.empty(), Optional.empty());
+        var mockHostResource = new HostResource(mockHost, mockHostSpec);
+        var mockSvc1 = new MockService(root, "sentinel");
+        mockSvc1.setHostResource(mockHostResource);
+        var mockPort1 = new StatePortInfo("localhost", 19098, mockSvc1);
+
+        var mockSvc2 = new MockService(root, "searchnode");
+        mockSvc2.setProp("clustername", "mycluster");
+        mockSvc2.setProp("clustertype", "mockup");
+        var mockPort2 = new StatePortInfo("other.host.local", 19102, mockSvc2);
+
+        generator.addStatePorts(List.of(mockPort1, mockPort2));
         String yaml = generator.generate();
+        // System.err.println(">>>\n" + yaml + "\n<<<");
         assertTrue(yaml.contains("sentinel"));
     }
 
+    static class MockService extends AbstractService {
+        private final String name;
+        public MockService(TreeConfigProducer<?> parent, String name) {
+            super(parent, name);
+            this.name = name;
+        }
+        public String getServiceName() { return name; }
+        public String getServiceType() { return "dummy"; }
+        @Override public int getPortCount() { return 0; }
+        @Override public void allocatePorts(int start, PortAllocBridge from) { }
+    }
+
 }