1 files changed, 140 insertions, 0 deletions
diff --git a/service-monitor/src/test/java/com/yahoo/vespa/service/monitor/internal/ModelGeneratorTest.java b/service-monitor/src/test/java/com/yahoo/vespa/service/monitor/internal/ModelGeneratorTest.java
new file mode 100644
index 00000000000..cf07c39950c
--- /dev/null
+++ b/service-monitor/src/test/java/com/yahoo/vespa/service/monitor/internal/ModelGeneratorTest.java
@@ -0,0 +1,140 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.service.monitor.internal;
+
+import com.yahoo.config.model.api.SuperModel;
+import com.yahoo.config.provision.Environment;
+import com.yahoo.config.provision.RegionName;
+import com.yahoo.config.provision.Zone;
+import com.yahoo.vespa.applicationmodel.ApplicationInstance;
+import com.yahoo.vespa.applicationmodel.ApplicationInstanceReference;
+import com.yahoo.vespa.applicationmodel.ServiceCluster;
+import com.yahoo.vespa.applicationmodel.ServiceInstance;
+import com.yahoo.vespa.applicationmodel.ServiceStatus;
+import com.yahoo.vespa.service.monitor.ServiceModel;
+import org.junit.Test;
+
+import java.util.Collections;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public class ModelGeneratorTest {
+    private final String ENVIRONMENT = "prod";
+    private final String REGION = "us-west-1";
+    private final String HOSTNAME = "hostname";
+    private final int PORT = 2;
+
+    @Test
+    public void toApplicationModelWithConfigServerApplication() throws Exception {
+        SuperModel superModel =
+                ExampleModel.createExampleSuperModelWithOneRpcPort(HOSTNAME, PORT);
+        ModelGenerator modelGenerator = new ModelGenerator();
+
+        Zone zone = new Zone(Environment.from(ENVIRONMENT), RegionName.from(REGION));
+
+        List<String> configServerHosts = Stream.of("cfg1", "cfg2", "cfg3")
+                .collect(Collectors.toList());
+
+        SlobrokMonitorManagerImpl slobrokMonitorManager = mock(SlobrokMonitorManagerImpl.class);
+        when(slobrokMonitorManager.getStatus(any(), any(), any()))
+                .thenReturn(ServiceStatus.UP);
+
+        ServiceModel serviceModel =
+                modelGenerator.toServiceModel(
+                        superModel,
+                        zone,
+                        configServerHosts,
+                        slobrokMonitorManager);
+
+        Map<ApplicationInstanceReference,
+                ApplicationInstance> applicationInstances =
+                serviceModel.getAllApplicationInstances();
+
+        assertEquals(2, applicationInstances.size());
+
+        Iterator<Map.Entry<ApplicationInstanceReference,
+                ApplicationInstance>> iterator =
+                applicationInstances.entrySet().iterator();
+
+        ApplicationInstance applicationInstance1 = iterator.next().getValue();
+        ApplicationInstance applicationInstance2 = iterator.next().getValue();
+
+        if (applicationInstance1.applicationInstanceId().equals(
+                ConfigServerApplication.APPLICATION_INSTANCE_ID)) {
+            verifyConfigServerApplication(applicationInstance1);
+            verifyOtherApplication(applicationInstance2);
+        } else {
+            verifyConfigServerApplication(applicationInstance2);
+            verifyOtherApplication(applicationInstance1);
+        }
+    }
+
+    @Test
+    public void toApplicationModel() throws Exception {
+        SuperModel superModel =
+                ExampleModel.createExampleSuperModelWithOneRpcPort(HOSTNAME, PORT);
+        ModelGenerator modelGenerator = new ModelGenerator();
+
+        Zone zone = new Zone(Environment.from(ENVIRONMENT), RegionName.from(REGION));
+
+        List<String> configServerHosts = Collections.emptyList();
+
+        SlobrokMonitorManagerImpl slobrokMonitorManager = mock(SlobrokMonitorManagerImpl.class);
+        when(slobrokMonitorManager.getStatus(any(), any(), any()))
+                .thenReturn(ServiceStatus.UP);
+
+        ServiceModel serviceModel =
+                modelGenerator.toServiceModel(
+                        superModel,
+                        zone,
+                        configServerHosts,
+                        slobrokMonitorManager);
+
+        Map<ApplicationInstanceReference,
+                ApplicationInstance> applicationInstances =
+                serviceModel.getAllApplicationInstances();
+
+        assertEquals(1, applicationInstances.size());
+        verifyOtherApplication(applicationInstances.values().iterator().next());
+    }
+
+    private void verifyOtherApplication(ApplicationInstance applicationInstance) {
+        assertEquals(String.format("%s:%s:%s:%s:%s",
+                ExampleModel.TENANT,
+                ExampleModel.APPLICATION_NAME,
+                ENVIRONMENT,
+                REGION,
+                ExampleModel.INSTANCE_NAME),
+                applicationInstance.reference().toString());
+
+        assertEquals(ExampleModel.TENANT, applicationInstance.tenantId().toString());
+        Set<ServiceCluster> serviceClusters =
+                applicationInstance.serviceClusters();
+        assertEquals(1, serviceClusters.size());
+        ServiceCluster serviceCluster = serviceClusters.iterator().next();
+        assertEquals(ExampleModel.CLUSTER_ID, serviceCluster.clusterId().toString());
+        assertEquals(ExampleModel.SERVICE_TYPE, serviceCluster.serviceType().toString());
+        Set<ServiceInstance> serviceInstances =
+                serviceCluster.serviceInstances();
+        assertEquals(1, serviceClusters.size());
+        ServiceInstance serviceInstance = serviceInstances.iterator().next();
+        assertEquals(HOSTNAME, serviceInstance.hostName().toString());
+        assertEquals(ExampleModel.CONFIG_ID, serviceInstance.configId().toString());
+        assertEquals(ServiceStatus.UP, serviceInstance.serviceStatus());
+    }
+
+    private void verifyConfigServerApplication(
+            ApplicationInstance applicationInstance) {
+        assertEquals(ConfigServerApplication.APPLICATION_INSTANCE_ID,
+                applicationInstance.applicationInstanceId());
+    }
+
+}