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
|
package pl.edu.mimuw.cloudatlas.agent;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.*;
import static org.junit.Assert.assertThat;
import static org.hamcrest.CoreMatchers.hasItems;
import java.lang.Runtime;
import java.lang.Process;
import java.lang.Thread;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import java.rmi.registry.LocateRegistry;
import java.rmi.RemoteException;
import java.rmi.registry.Registry;
import pl.edu.mimuw.cloudatlas.api.Api;
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;
public class AgentIntegrationTest {
private static Process registryProcess;
private static Process agentProcess;
private static final long queriesInterval = 100;
private static Registry registry;
private static Api api;
@BeforeClass
public static void bindApi() throws Exception {
registryProcess = Runtime.getRuntime().exec("./scripts/registry");
Thread.sleep(10000);
agentProcess = Runtime.getRuntime().exec("./gradlew runAgent -Dhostname=localhost");
Thread.sleep(10000);
registry = LocateRegistry.getRegistry("localhost");
api = (Api) registry.lookup("Api");
}
@AfterClass
public static void killProcesses() throws Exception {
try {
registryProcess.destroy();
agentProcess.destroy();
} catch (Exception e) {
System.out.println("Caught exception: " + e);
}
}
@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 ValueString(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"));
assertEquals(new ValueTime("2012/11/09 21:12:00.000"), attributes.get("timestamp"));
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, queryCode);
// TODO: test something here
}
@Test
public void testInstallQueryRuns() throws Exception {
String name = "&query";
String queryCode = "SELECT 1 AS one";
api.installQuery(name, queryCode);
Thread.sleep(queriesInterval * 2);
AttributesMap attributes = api.getZoneAttributeValues("/pjwstk");
assertEquals(new ValueInt(1l), attributes.getOrNull("one"));
}
@Test
public void testUninstallQuery() throws Exception {
String name = "&query";
String queryCode = "SELECT 1 AS one";
api.installQuery(name, queryCode);
api.uninstallQuery(name);
AttributesMap attributes = api.getZoneAttributeValues("/pjwstk");
assertNull(attributes.getOrNull(name));
// TODO: test this correctly
}
@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"));
}
}
|