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
141
142
143
144
145
146
|
// Copyright Vespa.ai. 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.createApplicationInOldFormat(id);
assertEquals(Optional.empty(), db.applicationData(id)); // still empty, as no data has been written to node
deleteApplication(db, id);
db.createApplication(id);
// 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
prepareSessionOldFormat(db, id, 2, OptionalLong.empty());
// Activate session 2, last deployed session not present (not writing json)
activateSessionOldFormat(db, id, 2);
// 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));
// 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);
// 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);
// 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) {
try (var t = db.createWritePrepareTransaction(new CuratorTransaction(curator),
applicationId,
sessionId,
activesSessionId)) {
t.commit();
}
}
private void prepareSessionOldFormat(ApplicationCuratorDatabase db, ApplicationId applicationId, long sessionId, OptionalLong activesSessionId) {
return; // Nothing to do, just return
}
private void activateSession(ApplicationCuratorDatabase db, ApplicationId applicationId, long sessionId) {
try (var t = db.createWriteActiveTransaction(new CuratorTransaction(curator), applicationId, sessionId)) {
t.commit();
}
}
private void activateSessionOldFormat(ApplicationCuratorDatabase db, ApplicationId applicationId, long sessionId) {
try (var t = db.createWriteActiveTransactionInOldFormat(new CuratorTransaction(curator), applicationId, sessionId)) {
t.commit();
}
}
}
|
