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
156
157
158
159
160
161
162
|
package pl.edu.mimuw.cloudatlas.agent;
import org.junit.Before;
import org.junit.Test;
import static org.junit.Assert.*;
import static org.hamcrest.CoreMatchers.hasItems;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import pl.edu.mimuw.cloudatlas.api.Api;
import pl.edu.mimuw.cloudatlas.interpreter.Main;
import pl.edu.mimuw.cloudatlas.model.AttributesMap;
import pl.edu.mimuw.cloudatlas.model.TypePrimitive;
import pl.edu.mimuw.cloudatlas.model.Value;
import pl.edu.mimuw.cloudatlas.model.ValueDouble;
import pl.edu.mimuw.cloudatlas.model.ValueInt;
import pl.edu.mimuw.cloudatlas.model.ValueList;
import pl.edu.mimuw.cloudatlas.model.ValueNull;
import pl.edu.mimuw.cloudatlas.model.ValueQuery;
import pl.edu.mimuw.cloudatlas.model.ValueString;
import pl.edu.mimuw.cloudatlas.model.ValueTime;
import pl.edu.mimuw.cloudatlas.model.ZMI;
import pl.edu.mimuw.cloudatlas.querysigner.QueryData;
public class ApiImplementationTests {
private ApiImplementation api;
@Before
public void initializeApi() throws Exception {
ZMI root = Main.createTestHierarchy2();
api = new ApiImplementation(root);
}
@Test
public void testGetZoneSet() throws Exception {
Set<String> set = api.getZoneSet();
assertEquals(8, set.size());
assertThat(set, hasItems("/"));
assertThat(set, hasItems("/uw"));
assertThat(set, hasItems("/uw/violet07", "/uw/khaki31", "/uw/khaki13"));
assertThat(set, hasItems("/pjwstk"));
assertThat(set, hasItems("/pjwstk/whatever01", "/pjwstk/whatever02"));
}
@Test
public void testRootGetZoneAttributeValue() throws Exception {
AttributesMap rootAttributes = api.getZoneAttributeValues("/");
assertEquals(new ValueInt(0l), rootAttributes.get("level"));
assertEquals(ValueNull.getInstance(), rootAttributes.get("name"));
}
@Test
public void testIntermediateGetZoneAttributeValue() throws Exception {
AttributesMap attributes = api.getZoneAttributeValues("/uw");
assertEquals(new ValueInt(1l), attributes.get("level"));
assertEquals(new ValueString("uw"), attributes.get("name"));
}
@Test
public void testLeafGetZoneAttributeValue() throws Exception {
AttributesMap attributes = api.getZoneAttributeValues("/pjwstk/whatever01");
assertEquals(new ValueInt(2l), attributes.get("level"));
assertEquals(new ValueString("whatever01"), attributes.get("name"));
assertEquals(new ValueString("/pjwstk/whatever01"), attributes.get("owner"));
long timestamp = ((ValueTime) attributes.get("timestamp")).getValue();
assertTrue(timestamp <= System.currentTimeMillis());
assertEquals(new ValueInt(1l), attributes.get("cardinality"));
assertEquals(new ValueTime("2012/10/18 07:03:00.000"), attributes.get("creation"));
assertEquals(new ValueDouble(0.1), attributes.get("cpu_usage"));
assertEquals(new ValueInt(7l), attributes.get("num_cores"));
assertEquals(new ValueInt(215l), attributes.get("num_processes"));
List<Value> phpModules = new ArrayList<Value>();
phpModules.add(new ValueString("rewrite"));
assertEquals(new ValueList(phpModules, TypePrimitive.STRING), attributes.get("php_modules"));
}
@Test
public void testInstallQuery() throws Exception {
String name = "&query";
String queryCode = "SELECT 1 AS one";
api.installQuery(name, new QueryData(queryCode, new byte[0]));
assertAttributeInZmiEquals(name, new ValueQuery(queryCode), "/");
assertAttributeInZmiEquals(name, new ValueQuery(queryCode), "/uw");
assertAttributeInZmiEquals(name, new ValueQuery(queryCode), "/pjwstk");
}
@Test
public void testInstallQueryRuns() throws Exception {
String name = "&query";
String queryCode = "SELECT 1 AS one";
api.installQuery(name, new QueryData(queryCode, new byte[0]));
assertAttributeInZmiEquals("one", new ValueInt(1l), "/");
assertAttributeInZmiEquals("one", new ValueInt(1l), "/uw");
assertAttributeInZmiEquals("one", new ValueInt(1l), "/pjwstk");
}
@Test
public void testInstallQueryRuns2() throws Exception {
String name = "&query";
String queryCode = "SELECT sum(num_processes) AS num_processes";
api.installQuery(name, new QueryData(queryCode, new byte[0]));
assertAttributeInZmiEquals("num_processes", new ValueInt(362l), "/uw");
assertAttributeInZmiEquals("num_processes", new ValueInt(437l), "/pjwstk");
assertAttributeInZmiEquals("num_processes", new ValueInt(799l), "/");
}
@Test
public void testInstallQueryWithInvalidNameFails() throws Exception {
String name = "query";
String queryCode = "SELECT 1 AS one";
try {
api.installQuery(name, new QueryData(queryCode, new byte[0]));
assertTrue("should have thrown", false);
} catch (Exception e) {
assertEquals("Invalid query identifier", e.getMessage());
}
}
public void assertAttributeInZmiEquals(String attribute, Value expected, String zmiPath) throws Exception {
AttributesMap attributes = api.getZoneAttributeValues(zmiPath);
assertEquals(expected, attributes.get(attribute));
}
@Test
public void testUninstallQuery() throws Exception {
String name = "&query";
String queryCode = "SELECT 1 AS one";
api.installQuery(name, new QueryData(queryCode, new byte[0]));
api.uninstallQuery(name, new QueryData(queryCode, new byte[0]));
AttributesMap attributes = api.getZoneAttributeValues("/pjwstk");
assertNull(attributes.getOrNull(name));
}
@Test
public void testSetAttributeValueChange() throws Exception {
Value numProcesses = new ValueInt(42l);
api.setAttributeValue("/uw/khaki13", "num_processes", numProcesses);
AttributesMap attributes = api.getZoneAttributeValues("/uw/khaki13");
assertEquals(numProcesses, attributes.get("num_processes"));
}
@Test
public void testSetAttributeValueAdd() throws Exception {
Value numProcesses = new ValueInt(42l);
api.setAttributeValue("/uw/khaki13", "an_attribute", numProcesses);
AttributesMap attributes = api.getZoneAttributeValues("/uw/khaki13");
assertEquals(numProcesses, attributes.get("an_attribute"));
}
@Test
public void testSetAttributeValueRunsQueries() throws Exception {
api.installQuery("&query", new QueryData("SELECT sum(num_processes) AS num_processes", new byte[0]));
Value numProcesses = new ValueInt(42l);
api.setAttributeValue("/uw/khaki13", "num_processes", numProcesses);
assertAttributeInZmiEquals("num_processes", new ValueInt(297l), "/uw");
assertAttributeInZmiEquals("num_processes", new ValueInt(437l), "/pjwstk");
assertAttributeInZmiEquals("num_processes", new ValueInt(734l), "/");
}
}
|