m-chrzan.xyz
aboutsummaryrefslogtreecommitdiff
path: root/src/test/java/pl/edu/mimuw/cloudatlas/agent/ApiImplementationTests.java
blob: 002c43cf3d3be852fb6ba4f849ee9aa35b448c4d (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
152
153
154
155
156
157
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;

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, queryCode);
        assertAttributeInZmiEquals(name, new ValueQuery(queryCode), "/");
        assertAttributeInZmiEquals(name, new ValueQuery(queryCode), "/uw");
        assertAttributeInZmiEquals(name, new ValueQuery(queryCode), "/pjwstk");
    }

    @Test
    public void testInstallQueryRuns() throws Exception {
        api.installQuery("&query", "SELECT 1 AS one");
        assertAttributeInZmiEquals("one", new ValueInt(1l), "/");
        assertAttributeInZmiEquals("one", new ValueInt(1l), "/uw");
        assertAttributeInZmiEquals("one", new ValueInt(1l), "/pjwstk");
    }

    @Test
    public void testInstallQueryRuns2() throws Exception {
        api.installQuery("&query", "SELECT sum(num_processes) AS num_processes");
        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, queryCode);
            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, 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"));
    }

    @Test
    public void testSetAttributeValueRunsQueries() throws Exception {
        api.installQuery("&query", "SELECT sum(num_processes) AS num_processes");
        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), "/");
    }
}