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
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server.session;
import com.yahoo.config.FileReference;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.path.Path;
import com.yahoo.text.Utf8;
import com.yahoo.vespa.config.server.zookeeper.ConfigCurator;
import com.yahoo.vespa.config.util.ConfigUtils;
import com.yahoo.vespa.curator.Curator;
import com.yahoo.vespa.curator.mock.MockCurator;
import org.junit.Before;
import org.junit.Test;
import java.time.Instant;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;
/**
* @author Ulf Lilleengen
*/
public class SessionZooKeeperClientTest {
private Curator curator;
private ConfigCurator configCurator;
@Before
public void setup() {
curator = new MockCurator();
configCurator = ConfigCurator.create(curator);
}
@Test
public void require_that_status_can_be_updated() {
SessionZooKeeperClient zkc = createSessionZKClient("1");
zkc.writeStatus(Session.Status.NEW);
assertThat(zkc.readStatus(), is(Session.Status.NEW));
zkc.writeStatus(Session.Status.PREPARE);
assertThat(zkc.readStatus(), is(Session.Status.PREPARE));
zkc.writeStatus(Session.Status.ACTIVATE);
assertThat(zkc.readStatus(), is(Session.Status.ACTIVATE));
zkc.writeStatus(Session.Status.DEACTIVATE);
assertThat(zkc.readStatus(), is(Session.Status.DEACTIVATE));
}
@Test
public void require_that_status_is_written_to_zk() {
SessionZooKeeperClient zkc = createSessionZKClient("2");
zkc.writeStatus(Session.Status.NEW);
String path = "/2" + ConfigCurator.SESSIONSTATE_ZK_SUBPATH;
assertTrue(configCurator.exists(path));
assertThat(configCurator.getData(path), is("NEW"));
}
@Test
public void require_that_status_is_read_from_zk() {
SessionZooKeeperClient zkc = createSessionZKClient("3");
curator.set(Path.fromString("3").append(ConfigCurator.SESSIONSTATE_ZK_SUBPATH), Utf8.toBytes("PREPARE"));
assertThat(zkc.readStatus(), is(Session.Status.PREPARE));
}
@Test
public void require_that_application_id_is_written_to_zk() {
ApplicationId id = new ApplicationId.Builder()
.tenant("tenant")
.applicationName("foo").instanceName("bim").build();
SessionZooKeeperClient zkc = createSessionZKClient("3");
zkc.writeApplicationId(id);
String path = "/3/" + SessionZooKeeperClient.APPLICATION_ID_PATH;
assertTrue(configCurator.exists(path));
assertThat(configCurator.getData(path), is("tenant:foo:bim"));
}
@Test
public void require_that_application_id_is_read_from_zk() {
ApplicationId id = new ApplicationId.Builder()
.tenant("tenant")
.applicationName("bar").instanceName("quux").build();
String idNoVersion = id.serializedForm();
assertApplicationIdParse("3", idNoVersion, idNoVersion);
}
@Test
public void require_that_create_time_can_be_written_and_read() {
SessionZooKeeperClient zkc = createSessionZKClient("3");
curator.delete(Path.fromString("3"));
assertThat(zkc.readCreateTime(), is(Instant.EPOCH));
Instant now = Instant.now();
zkc.createNewSession(now);
// resolution is in seconds, so need to go back use that when comparing
assertThat(zkc.readCreateTime(), is(Instant.ofEpochSecond(now.getEpochSecond())));
}
@Test
public void require_that_application_package_file_reference_can_be_written_and_read() {
final FileReference testRef = new FileReference("test-ref");
SessionZooKeeperClient zkc = createSessionZKClient("3");
zkc.writeApplicationPackageReference(testRef);
assertThat(zkc.readApplicationPackageReference(), is(testRef));
}
private void assertApplicationIdParse(String sessionId, String idString, String expectedIdString) {
SessionZooKeeperClient zkc = createSessionZKClient(sessionId);
String path = "/" + sessionId + "/" + SessionZooKeeperClient.APPLICATION_ID_PATH;
configCurator.putData(path, idString);
ApplicationId zkId = zkc.readApplicationId();
assertThat(zkId.serializedForm(), is(expectedIdString));
}
private SessionZooKeeperClient createSessionZKClient(String sessionId) {
SessionZooKeeperClient zkc = new SessionZooKeeperClient(curator,
ConfigCurator.create(curator),
Path.fromString(sessionId),
ConfigUtils.getCanonicalHostName());
zkc.createNewSession(Instant.now());
return zkc;
}
}
|
