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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server.application;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.vespa.curator.mock.MockCurator;
import com.yahoo.vespa.curator.transaction.CuratorTransaction;
import org.junit.Test;
import java.time.Instant;
import java.util.Optional;
import java.util.OptionalLong;
import static org.junit.Assert.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;
/**
* @author jonmv
*/
public class ApplicationCuratorDatabaseTest {
private final MockCurator curator = new MockCurator();
@Test
public void testReindexingStatusSerialization() {
ApplicationId id = ApplicationId.defaultId();
ApplicationCuratorDatabase db = new ApplicationCuratorDatabase(id.tenant(), curator);
assertEquals(Optional.empty(), db.readReindexingStatus(id));
ApplicationReindexing reindexing = ApplicationReindexing.empty()
.withPending("one", "a", 10)
.withReady("two", "b", Instant.ofEpochMilli(2), 0.2, "test reindexing")
.withPending("two", "b", 20)
.withReady("one", "a", Instant.ofEpochMilli(1), 0.2, "test reindexing")
.withReady("two", "c", Instant.ofEpochMilli(3), 0.2, "test reindexing")
.enabled(false);
db.writeReindexingStatus(id, reindexing);
assertEquals(reindexing, db.readReindexingStatus(id).orElseThrow());
}
@Test
public void testReadingAndWritingApplicationData() {
ApplicationId id = ApplicationId.defaultId();
ApplicationCuratorDatabase db = new ApplicationCuratorDatabase(id.tenant(), curator);
assertEquals(Optional.empty(), db.applicationData(id));
db.createApplication(id, false);
assertEquals(Optional.empty(), db.applicationData(id)); // still empty, as no data has been written to node
deleteApplication(db, id);
db.createApplication(id, true);
// Can be read as json, but no active session or last deployed session
Optional<ApplicationData> applicationData = db.applicationData(id);
assertTrue(applicationData.isPresent());
assertEquals(id, applicationData.get().applicationId());
assertFalse(applicationData.get().activeSession().isPresent());
assertFalse(applicationData.get().lastDeployedSession().isPresent());
// Prepare session 2, no active session
prepareSession(db, id, 2, OptionalLong.empty(), false);
// Activate session 2, last deployed session not present (not writing json)
activateSession(db, id, 2, false);
// Can be read as session id only
applicationData = db.applicationData(id);
assertTrue(applicationData.isPresent());
assertEquals(id, applicationData.get().applicationId());
assertTrue(applicationData.get().activeSession().isPresent());
assertEquals(2, applicationData.get().activeSession().get().longValue());
assertFalse(applicationData.get().lastDeployedSession().isPresent());
// Can be read as session data as well
applicationData = db.applicationData(id);
assertTrue(applicationData.isPresent());
assertEquals(id, applicationData.get().applicationId());
assertTrue(applicationData.get().activeSession().isPresent());
assertEquals(2, applicationData.get().activeSession().get().longValue());
assertFalse(applicationData.get().lastDeployedSession().isPresent());
// Prepare session 3, last deployed session is still 2
prepareSession(db, id, 3, OptionalLong.of(2), true);
// Can be read as json, active session is still 2 and last deployed session is 3
applicationData = db.applicationData(id);
assertTrue(applicationData.isPresent());
assertEquals(id, applicationData.get().applicationId());
assertTrue(applicationData.get().activeSession().isPresent(), applicationData.get().toString());
assertEquals(2L, applicationData.get().activeSession().get().longValue());
assertTrue(applicationData.get().lastDeployedSession().isPresent());
assertEquals(3, applicationData.get().lastDeployedSession().get().longValue());
activateSession(db, id, 3, true);
// Can be read as json, active session and last deployed session present
applicationData = db.applicationData(id);
assertTrue(applicationData.isPresent());
assertEquals(id, applicationData.get().applicationId());
assertTrue(applicationData.get().activeSession().isPresent());
assertEquals(3, applicationData.get().activeSession().get().longValue());
assertTrue(applicationData.get().lastDeployedSession().isPresent());
assertEquals(3, applicationData.get().lastDeployedSession().get().longValue());
// createApplication should not overwrite the node if it already exists
db.createApplication(id, true);
// Can be read as json, active session and last deployed session present
applicationData = db.applicationData(id);
assertTrue(applicationData.isPresent());
assertEquals(id, applicationData.get().applicationId());
assertTrue(applicationData.get().activeSession().isPresent());
assertEquals(3, applicationData.get().activeSession().get().longValue());
assertTrue(applicationData.get().lastDeployedSession().isPresent());
assertEquals(3, applicationData.get().lastDeployedSession().get().longValue());
}
private void deleteApplication(ApplicationCuratorDatabase db, ApplicationId applicationId) {
try (var t = db.createDeleteTransaction(applicationId)) {
t.commit();
}
}
private void prepareSession(ApplicationCuratorDatabase db, ApplicationId applicationId, long sessionId, OptionalLong activesSessionId, boolean writeAsJson) {
try (var t = db.createWritePrepareTransaction(new CuratorTransaction(curator),
applicationId,
sessionId,
activesSessionId,
writeAsJson)) {
t.commit();
}
}
private void activateSession(ApplicationCuratorDatabase db, ApplicationId applicationId, long sessionId, boolean writeAsJson) {
try (var t = db.createWriteActiveTransaction(new CuratorTransaction(curator), applicationId, sessionId, writeAsJson)) {
t.commit();
}
}
}
|
