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
147
148
149
150
151
152
153
154
155
|
// Copyright 2016 Yahoo Inc. 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.config.provision.TenantName;
import com.yahoo.path.Path;
import com.yahoo.text.Utf8;
import com.yahoo.vespa.config.server.MockReloadHandler;
import com.yahoo.vespa.config.server.TestWithCurator;
import org.junit.Test;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import static org.hamcrest.Matchers.is;
import static org.junit.Assert.*;
/**
* @author lulf
* @since 5.1
*/
public class TenantApplicationsTest extends TestWithCurator {
@Test
public void require_that_applications_are_read_from_zookeeper() throws Exception {
curatorFramework.create().creatingParentsIfNeeded().forPath("/foo:dev:baz", Utf8.toAsciiBytes(3));
curatorFramework.create().creatingParentsIfNeeded().forPath("/bar:test:bim", Utf8.toAsciiBytes(4));
TenantApplications repo = createZKAppRepo();
List<ApplicationId> applications = repo.listApplications();
assertThat(applications.size(), is(2));
assertThat(applications.get(0).application().value(), is("dev"));
assertThat(applications.get(1).application().value(), is("test"));
assertThat(repo.getSessionIdForApplication(applications.get(0)), is(3L));
assertThat(repo.getSessionIdForApplication(applications.get(1)), is(4L));
}
@Test
public void require_that_invalid_entries_are_skipped() throws Exception {
curatorFramework.create().creatingParentsIfNeeded().forPath("/foo:dev:baz");
curatorFramework.create().creatingParentsIfNeeded().forPath("/invalid");
TenantApplications repo = createZKAppRepo();
List<ApplicationId> applications = repo.listApplications();
assertThat(applications.size(), is(1));
assertThat(applications.get(0).application().value(), is("dev"));
}
@Test(expected = IllegalArgumentException.class)
public void require_that_requesting_session_for_unknown_application_throws_exception() throws Exception {
curatorFramework.create().creatingParentsIfNeeded().forPath("/foo:dev:baz:bim");
TenantApplications repo = createZKAppRepo();
repo.getSessionIdForApplication(new ApplicationId.Builder()
.tenant("exist")
.applicationName("tenant").instanceName("here").build());
}
@Test(expected = IllegalArgumentException.class)
public void require_that_requesting_session_for_empty_application_throws_exception() throws Exception {
curatorFramework.create().creatingParentsIfNeeded().forPath("/foo:dev:baz:bim");
TenantApplications repo = createZKAppRepo();
repo.getSessionIdForApplication(new ApplicationId.Builder()
.tenant("tenant")
.applicationName("foo").instanceName("bim").build());
}
@Test
public void require_that_application_ids_can_be_written() throws Exception {
TenantApplications repo = createZKAppRepo();
repo.createPutApplicationTransaction(createAppplicationId("myapp"), 3l).commit();
String path = "/mytenant:myapp:myinst";
assertTrue(curatorFramework.checkExists().forPath(path) != null);
assertThat(Utf8.toString(curatorFramework.getData().forPath(path)), is("3"));
repo.createPutApplicationTransaction(createAppplicationId("myapp"), 5l).commit();
assertTrue(curatorFramework.checkExists().forPath(path) != null);
assertThat(Utf8.toString(curatorFramework.getData().forPath(path)), is("5"));
}
@Test
public void require_that_application_ids_can_be_deleted() throws Exception {
TenantApplications repo = createZKAppRepo();
ApplicationId id1 = createAppplicationId("myapp");
ApplicationId id2 = createAppplicationId("myapp2");
repo.createPutApplicationTransaction(id1, 1).commit();
repo.createPutApplicationTransaction(id2, 1).commit();
assertThat(repo.listApplications().size(), is(2));
repo.deleteApplication(id1).commit();
assertThat(repo.listApplications().size(), is(1));
repo.deleteApplication(id2).commit();
assertThat(repo.listApplications().size(), is(0));
}
@Test
public void require_that_repos_behave_similarly() throws Exception {
TenantApplications zkRepo = createZKAppRepo();
TenantApplications memRepo = new MemoryTenantApplications();
for (TenantApplications repo : Arrays.asList(zkRepo, memRepo)) {
ApplicationId id1 = createAppplicationId("myapp");
ApplicationId id2 = createAppplicationId("myapp2");
repo.createPutApplicationTransaction(id1, 4).commit();
repo.createPutApplicationTransaction(id2, 5).commit();
List<ApplicationId> lst = repo.listApplications();
Collections.sort(lst);
assertThat(lst.size(), is(2));
assertThat(lst.get(0).application(), is(id1.application()));
assertThat(lst.get(1).application(), is(id2.application()));
assertThat(repo.getSessionIdForApplication(id1), is(4l));
assertThat(repo.getSessionIdForApplication(id2), is(5l));
repo.createPutApplicationTransaction(id1, 6).commit();
lst = repo.listApplications();
Collections.sort(lst);
assertThat(lst.size(), is(2));
assertThat(lst.get(0).application(), is(id1.application()));
assertThat(lst.get(1).application(), is(id2.application()));
assertThat(repo.getSessionIdForApplication(id1), is(6l));
assertThat(repo.getSessionIdForApplication(id2), is(5l));
repo.deleteApplication(id1).commit();
assertThat(repo.listApplications().size(), is(1));
repo.deleteApplication(id2).commit();
}
}
@Test
public void require_that_reload_handler_is_called_when_apps_are_removed() throws Exception {
curatorFramework.create().creatingParentsIfNeeded().forPath("/foo:test:baz", Utf8.toAsciiBytes(3));
curatorFramework.create().creatingParentsIfNeeded().forPath("/bar:dev:bim", Utf8.toAsciiBytes(4));
MockReloadHandler reloadHandler = new MockReloadHandler();
TenantApplications repo = createZKAppRepo(reloadHandler);
assertNull(reloadHandler.lastRemoved);
repo.deleteApplication(new ApplicationId.Builder()
.tenant("foo")
.applicationName("test").instanceName("baz").build())
.commit();
long endTime = System.currentTimeMillis() + 60_000;
while (System.currentTimeMillis() < endTime && reloadHandler.lastRemoved == null) {
Thread.sleep(100);
}
assertNotNull(reloadHandler.lastRemoved);
assertThat(reloadHandler.lastRemoved.serializedForm(), is("foo:test:baz"));
}
private TenantApplications createZKAppRepo() {
return createZKAppRepo(new MockReloadHandler());
}
private TenantApplications createZKAppRepo(MockReloadHandler reloadHandler) {
return ZKTenantApplications.create(curator, Path.createRoot(), reloadHandler, TenantName.from("mytenant"));
}
private static ApplicationId createAppplicationId(String name) {
return new ApplicationId.Builder()
.tenant("mytenant")
.applicationName(name).instanceName("myinst").build();
}
}
|
