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
|
// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.service.health;
import com.yahoo.config.model.api.ApplicationInfo;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.HostName;
import com.yahoo.vespa.applicationmodel.ClusterId;
import com.yahoo.vespa.applicationmodel.ConfigId;
import com.yahoo.vespa.applicationmodel.ServiceStatus;
import com.yahoo.vespa.applicationmodel.ServiceType;
import com.yahoo.vespa.service.duper.ProxyHostApplication;
import com.yahoo.vespa.service.duper.TestZoneApplication;
import com.yahoo.vespa.service.duper.ZoneApplication;
import com.yahoo.vespa.service.executor.Cancellable;
import com.yahoo.vespa.service.executor.RunletExecutor;
import com.yahoo.vespa.service.model.ServiceId;
import org.junit.Test;
import java.time.Duration;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
/**
* @author hakonhall
*/
public class StateV1HealthModelTest {
private RunletExecutor executor = mock(RunletExecutor.class);
private Duration healthStaleness = Duration.ofSeconds(1);
private Duration requestTimeout = Duration.ofSeconds(2);
private Duration keepAlive = Duration.ofSeconds(3);
private final ProxyHostApplication proxyHostApplication = new ProxyHostApplication();
private final List<HostName> hostnames = Stream.of("host1", "host2").map(HostName::from).collect(Collectors.toList());
private final ApplicationInfo proxyHostApplicationInfo = proxyHostApplication.makeApplicationInfo(hostnames);
private StateV1HealthModel model;
private Map<ServiceId, HealthEndpoint> endpoints;
@Test
public void test() {
model = new StateV1HealthModel(healthStaleness, requestTimeout, keepAlive, executor, false);
endpoints = model.extractHealthEndpoints(proxyHostApplicationInfo);
assertEquals(2, endpoints.size());
ApplicationId applicationId = ApplicationId.from("hosted-vespa", "proxy-host", "default");
ClusterId clusterId = new ClusterId("proxy-host");
ServiceId hostAdmin1 = new ServiceId(applicationId, clusterId, ServiceType.HOST_ADMIN, new ConfigId("proxy-host/host1"));
ServiceId hostAdmin2 = new ServiceId(applicationId, clusterId, ServiceType.HOST_ADMIN, new ConfigId("proxy-host/host2"));
HealthEndpoint endpoint1 = endpoints.get(hostAdmin1);
assertNotNull(endpoint1);
assertEquals("http://host1:8080/state/v1/health", endpoint1.description());
HealthEndpoint endpoint2 = endpoints.get(hostAdmin2);
assertNotNull(endpoint2);
assertEquals("http://host2:8080/state/v1/health", endpoint2.description());
Cancellable cancellable = mock(Cancellable.class);
when(executor.scheduleWithFixedDelay(any(), any())).thenReturn(cancellable);
try (HealthMonitor healthMonitor = endpoint1.startMonitoring()) {
assertEquals(ServiceStatus.DOWN, healthMonitor.getStatus());
}
}
@Test
public void testMonitoringTenantHostHealth() {
model = new StateV1HealthModel(healthStaleness, requestTimeout, keepAlive, executor, true);
ApplicationInfo zoneApplicationInfo = new TestZoneApplication.Builder()
.addNodeAdminCluster("h1")
.addRoutingCluster("r1")
.build()
.makeApplicationInfo();
endpoints = model.extractHealthEndpoints(zoneApplicationInfo);
assertEquals(1, endpoints.size());
HealthEndpoint endpoint = endpoints.values().iterator().next();
assertEquals("http://h1:8080/state/v1/health", endpoint.description());
ServiceId serviceId = endpoint.getServiceId();
assertEquals(ZoneApplication.getApplicationId(), serviceId.getApplicationId());
assertEquals(ZoneApplication.getNodeAdminClusterId(), serviceId.getClusterId());
assertEquals(ZoneApplication.getNodeAdminServiceType(), serviceId.getServiceType());
}
}
|
