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
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.document.restapi;
import com.yahoo.vespaclient.ClusterDef;
import org.junit.Test;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.*;
public class OperationHandlerImplTest {
@Test(expected = IllegalArgumentException.class)
public void missingClusterDef() throws RestApiException {
List<ClusterDef> clusterDef = new ArrayList<>();
OperationHandlerImpl.resolveClusterRoute(Optional.empty(), clusterDef);
}
@Test(expected = IllegalArgumentException.class)
public void missingClusterDefSpecifiedCluster() throws RestApiException {
List<ClusterDef> clusterDef = new ArrayList<>();
OperationHandlerImpl.resolveClusterRoute(Optional.of("cluster"), clusterDef);
}
@Test(expected = RestApiException.class)
public void oneClusterPresentNotMatching() throws RestApiException {
List<ClusterDef> clusterDef = new ArrayList<>();
clusterDef.add(new ClusterDef("foo", "configId"));
OperationHandlerImpl.resolveClusterRoute(Optional.of("cluster"), clusterDef);
}
@Test()
public void oneClusterMatching() throws RestApiException {
List<ClusterDef> clusterDef = new ArrayList<>();
clusterDef.add(new ClusterDef("foo", "configId"));
assertThat(OperationHandlerImpl.resolveClusterRoute(Optional.of("foo"), clusterDef),
is("[Storage:cluster=foo;clusterconfigid=configId]"));
}
@Test()
public void oneClusterMatchingManyAvailable() throws RestApiException {
List<ClusterDef> clusterDef = new ArrayList<>();
clusterDef.add(new ClusterDef("foo2", "configId2"));
clusterDef.add(new ClusterDef("foo", "configId"));
clusterDef.add(new ClusterDef("foo3", "configId2"));
assertThat(OperationHandlerImpl.resolveClusterRoute(Optional.of("foo"), clusterDef),
is("[Storage:cluster=foo;clusterconfigid=configId]"));
}
@Test()
public void checkErrorMessage() throws RestApiException, IOException {
List<ClusterDef> clusterDef = new ArrayList<>();
clusterDef.add(new ClusterDef("foo2", "configId2"));
clusterDef.add(new ClusterDef("foo", "configId"));
clusterDef.add(new ClusterDef("foo3", "configId2"));
try {
OperationHandlerImpl.resolveClusterRoute(Optional.of("wrong"), clusterDef);
} catch(RestApiException e) {
ByteArrayOutputStream stream = new ByteArrayOutputStream();
e.getResponse().render(stream);
String errorMsg = new String( stream.toByteArray());
assertThat(errorMsg, is("{\"errors\":[\"Your vespa cluster contains the content clusters foo2 " +
"(configId2), foo (configId), foo3 (configId2), not wrong. Please select a valid vespa cluster.\"]}"));
return;
}
fail("Expected exception");
}
}
|
