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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
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.zookeeper;
import com.yahoo.cloud.config.ZookeeperServerConfig;
import com.yahoo.io.IOUtils;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import java.io.File;
import java.io.IOException;
import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;
import static com.yahoo.vespa.defaults.Defaults.getDefaults;
/**
* Tests the zookeeper server.
*/
public class VespaZooKeeperServerImplTest {
@Rule
public TemporaryFolder folder = new TemporaryFolder();
@Test
public void config_is_written_correctly_when_one_server() throws IOException {
File cfgFile = folder.newFile();
File idFile = folder.newFile();
ZookeeperServerConfig.Builder builder = new ZookeeperServerConfig.Builder();
builder.zooKeeperConfigFile(cfgFile.getAbsolutePath());
builder.myidFile(idFile.getAbsolutePath());
builder.server(newServer(0, "foo", 123, 321));
builder.myid(0);
createServer(builder);
validateConfigFileSingleHost(cfgFile);
validateIdFile(idFile, "");
}
@Test
public void config_is_written_correctly_when_multiple_servers() throws IOException {
File cfgFile = folder.newFile();
File idFile = folder.newFile();
ZookeeperServerConfig.Builder builder = new ZookeeperServerConfig.Builder();
builder.zooKeeperConfigFile(cfgFile.getAbsolutePath());
builder.server(newServer(0, "foo", 123, 321));
builder.server(newServer(1, "bar", 234, 432));
builder.server(newServer(2, "baz", 345, 543));
builder.myidFile(idFile.getAbsolutePath());
builder.myid(1);
createServer(builder);
validateConfigFileMultipleHosts(cfgFile);
validateIdFile(idFile, "1\n");
}
private void createServer(ZookeeperServerConfig.Builder builder) {
new VespaZooKeeperServerImpl(new ZookeeperServerConfig(builder), false);
}
@Test(expected = RuntimeException.class)
public void require_that_this_id_must_be_present_amongst_servers() {
ZookeeperServerConfig.Builder builder = new ZookeeperServerConfig.Builder();
builder.server(newServer(1, "bar", 234, 432));
builder.server(newServer(2, "baz", 345, 543));
builder.myid(0);
createServer(builder);
}
@Test
public void juteMaxBufferCanBeSet() throws IOException {
ZookeeperServerConfig.Builder builder = new ZookeeperServerConfig.Builder();
builder.myid(0);
File idFile = folder.newFile();
File cfgFile = folder.newFile();
builder.server(new ZookeeperServerConfig.Server.Builder().id(0).hostname("testhost"));
builder.zooKeeperConfigFile(cfgFile.getAbsolutePath());
builder.myidFile(idFile.getAbsolutePath());
createServer(builder);
assertThat(System.getProperty(VespaZooKeeperServerImpl.ZOOKEEPER_JUTE_MAX_BUFFER), is("" + new ZookeeperServerConfig(builder).juteMaxBuffer()));
final int max_buffer = 1;
builder.juteMaxBuffer(max_buffer);
createServer(builder);
assertThat(System.getProperty(VespaZooKeeperServerImpl.ZOOKEEPER_JUTE_MAX_BUFFER), is("" + max_buffer));
}
private ZookeeperServerConfig.Server.Builder newServer(int id, String hostName, int electionPort, int quorumPort) {
ZookeeperServerConfig.Server.Builder builder = new ZookeeperServerConfig.Server.Builder();
builder.id(id);
builder.hostname(hostName);
builder.electionPort(electionPort);
builder.quorumPort(quorumPort);
return builder;
}
private void validateIdFile(File idFile, String expected) throws IOException {
String actual = IOUtils.readFile(idFile);
assertThat(actual, is(expected));
}
private void validateConfigFileSingleHost(File cfgFile) throws IOException {
String expected =
"tickTime=2000\n" +
"initLimit=20\n" +
"syncLimit=15\n" +
"maxClientCnxns=0\n" +
"snapCount=50000\n" +
"dataDir=" + getDefaults().underVespaHome("var/zookeeper") + "\n" +
"clientPort=2181\n" +
"autopurge.purgeInterval=1\n" +
"autopurge.snapRetainCount=15\n" +
"4lw.commands.whitelist=conf,cons,crst,dump,envi,mntr,ruok,srst,srvr,stat,wchs\n" +
"server.0=foo:321:123\n";
validateConfigFile(cfgFile, expected);
}
private void validateConfigFileMultipleHosts(File cfgFile) throws IOException {
String expected =
"tickTime=2000\n" +
"initLimit=20\n" +
"syncLimit=15\n" +
"maxClientCnxns=0\n" +
"snapCount=50000\n" +
"dataDir=" + getDefaults().underVespaHome("var/zookeeper") + "\n" +
"clientPort=2181\n" +
"autopurge.purgeInterval=1\n" +
"autopurge.snapRetainCount=15\n" +
"4lw.commands.whitelist=conf,cons,crst,dump,envi,mntr,ruok,srst,srvr,stat,wchs\n" +
"server.0=foo:321:123\n" +
"server.1=bar:432:234\n" +
"server.2=baz:543:345\n";
validateConfigFile(cfgFile, expected);
}
private void validateConfigFile(File cfgFile, String expected) throws IOException {
String actual = IOUtils.readFile(cfgFile);
assertThat(actual, is(expected));
}
}
|
