1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
// 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.cloud.config.ConfigserverConfig;
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 com.yahoo.vespa.service.monitor.application.ConfigServerApplication;
import com.yahoo.vespa.service.monitor.internal.slobrok.SlobrokMonitorManagerImpl;
import org.junit.Test;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
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 toApplicationModel() throws Exception {
SuperModel superModel = ExampleModel.createExampleSuperModelWithOneRpcPort(HOSTNAME, PORT);
ConfigserverConfig config = ConfigserverUtil.create(
true, "cfg1", "cfg2", "cfg3");
DuperModel duperModel = new DuperModel(config);
ModelGenerator modelGenerator = new ModelGenerator();
Zone zone = new Zone(Environment.from(ENVIRONMENT), RegionName.from(REGION));
SlobrokMonitorManagerImpl slobrokMonitorManager = mock(SlobrokMonitorManagerImpl.class);
when(slobrokMonitorManager.getStatus(any(), any(), any(), any()))
.thenReturn(ServiceStatus.UP);
ServiceModel serviceModel =
modelGenerator.toServiceModel(
duperModel.getApplicationInfos(superModel),
zone,
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);
}
}
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());
}
}
|