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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.config.model.provision;
import com.yahoo.config.application.api.ApplicationPackage;
import com.yahoo.config.model.api.HostProvisioner;
import com.yahoo.config.model.test.MockApplicationPackage;
import com.yahoo.config.provision.HostSpec;
import com.yahoo.vespa.model.VespaModel;
import org.junit.jupiter.api.Test;
import org.xml.sax.SAXException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.jupiter.api.Assertions.assertTrue;
/**
* @author hmusum
*/
public class SingleNodeProvisionerTest {
@Test
void require_basic_works() {
SingleNodeProvisioner hostProvisioner = new SingleNodeProvisioner();
// 4 services, 2 host aliases, mapping to 2 host.
List<String> aliases = createAliases();
Map<String, HostSpec> map = allocate(hostProvisioner, aliases);
assertCorrectNumberOfHost(map, 1);
assertThat(map.size(), is(2));
assertTrue(map.keySet().containsAll(aliases));
// 5 services, 3 host aliases, mapping to 2 host.
aliases = createAliases(Collections.singletonList("node3"));
map = allocate(hostProvisioner, aliases);
assertCorrectNumberOfHost(map, 1);
assertThat(map.size(), is(3));
assertTrue(map.keySet().containsAll(aliases));
// 5 services, 3 host aliases, mapping to 3 host.
aliases = createAliases(Collections.singletonList("node4"));
map = allocate(hostProvisioner, aliases);
assertThat(map.size(), is(3));
assertCorrectNumberOfHost(map, 1);
assertTrue(map.keySet().containsAll(aliases));
}
@Test
void require_allocate_clustermembership_works() throws IOException, SAXException {
String servicesXml = "<services version='1.0'>"
+ " <admin version='3.0'>"
+ " <nodes count='1' />"
+ " </admin>"
+ " <container version='1.0'>"
+ " <search />"
+ " <nodes count='1' />"
+ " </container>"
+ "</services>";
ApplicationPackage app = new MockApplicationPackage.Builder().withServices(servicesXml).build();
VespaModel model = new VespaModel(app);
assertThat(model.getHosts().size(), is(1));
}
private Map<String, HostSpec> allocate(HostProvisioner provisioner, List<String> aliases) {
Map<String, HostSpec> map = new LinkedHashMap<>();
for (String alias : aliases) {
map.put(alias, provisioner.allocateHost(alias));
}
return map;
}
private void assertCorrectNumberOfHost(Map<String, HostSpec> hostToServiceMap, int expectedHostCount) {
Set<String> hostSet = new HashSet<>();
for (HostSpec host : hostToServiceMap.values()) {
hostSet.add(host.hostname());
}
assertThat(hostSet.size(), is(expectedHostCount));
}
private List<String> createAliases() {
return createAliases(new ArrayList<String>());
}
// Admin services on node1, container on node2 + additional specs
private List<String> createAliases(Collection<String> additionalAliases) {
List<String> aliases = new ArrayList<>();
aliases.add("node1");
aliases.add("node1");
aliases.add("node1");
aliases.add("node2");
aliases.addAll(additionalAliases);
return aliases;
}
}
|
