m-chrzan.xyz
aboutsummaryrefslogtreecommitdiff
path: root/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java
blob: 33fb56f77c9f32e4c6c99474e359679f53a9d620 (plain)
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
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 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 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"));
        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);
        AttributesMap attributes = api.getZoneAttributeValues("/pjwstk");
        assertEquals(new ValueQuery(queryCode), attributes.get(name));
    }

    @Test
    public void testInstallQueryRuns() throws Exception {
        String name = "&query";
        String queryCode = "SELECT 1 AS one";
        api.installQuery(name, queryCode);
        AttributesMap attributes = api.getZoneAttributeValues("/pjwstk");
        assertEquals(new ValueInt(1l), attributes.get("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));
    }

    @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"));
    }
}