2 files changed, 14 insertions, 8 deletions

diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java
index fcdfb47..28b47a9 100644
--- a/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java
+++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java
@@ -38,6 +38,8 @@ 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;
 
@@ -76,14 +78,14 @@ public class AgentIntegrationTest {
     @Test
     public void testRootGetZoneAttributeValue() throws Exception {
         AttributesMap rootAttributes = api.getZoneAttributeValues("/");
-        assertEquals(new ValueInt(0l), rootAttributes.get("level"));
+        // 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 ValueInt(1l), attributes.get("level"));
         assertEquals(new ValueString("uw"), attributes.get("name"));
     }
 
@@ -118,9 +120,10 @@ public class AgentIntegrationTest {
         String name = "&query";
         String queryCode = "SELECT 1 AS one";
         api.installQuery(name, queryCode);
-        // TODO: test this eventually runs
-        // AttributesMap attributes = api.getZoneAttributeValues("/pjwstk");
-        // assertEquals(new ValueInt(1l), attributes.get("one"));
+
+        Thread.sleep(queriesInterval * 2);
+        AttributesMap attributes = api.getZoneAttributeValues("/pjwstk");
+        assertEquals(new ValueInt(1l), attributes.getOrNull("one"));
     }
 
     @Test
diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/QurnikTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/QurnikTest.java
index b6e4ea8..1e4fbda 100644
--- a/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/QurnikTest.java
+++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/QurnikTest.java
@@ -76,7 +76,8 @@ public class QurnikTest {
         UpdateAttributesMessage updateAttributesMessage = (UpdateAttributesMessage) stanikMessage;
         assertEquals("/", updateAttributesMessage.getPathName());
         AttributesMap updatedAttributes = updateAttributesMessage.getAttributes();
-        assertEquals(2, TestUtil.iterableSize(updatedAttributes));
+        assertEquals(3, TestUtil.iterableSize(updatedAttributes));
+        assertEquals(new ValueString(null), updatedAttributes.getOrNull("name"));
         assertEquals(new ValueInt(1l), updatedAttributes.getOrNull("one"));
         long timestamp = ((ValueTime) updatedAttributes.getOrNull("timestamp")).getValue();
         assertTrue(timeBefore <= timestamp);
@@ -191,7 +192,8 @@ public class QurnikTest {
         assertEquals("/", message2.getPathName());
         AttributesMap attributes2 = message2.getAttributes();
         System.out.println("got attributes " + attributes2.toString());
-        assertEquals(6, TestUtil.iterableSize(attributes2));
+        assertEquals(7, TestUtil.iterableSize(attributes2));
+        assertEquals(new ValueString(null), attributes2.getOrNull("name"));
         assertEquals(new ValueInt(67l), attributes2.getOrNull("x"));
         assertEquals(new ValueInt(100l), attributes2.getOrNull("y"));
         assertEquals(new ValueInt(5l), attributes2.getOrNull("z"));
@@ -241,7 +243,8 @@ public class QurnikTest {
         assertEquals("/", message2.getPathName());
         AttributesMap attributes2 = message2.getAttributes();
         System.out.println("got attributes " + attributes2.toString());
-        assertEquals(3, TestUtil.iterableSize(attributes2));
+        assertEquals(4, TestUtil.iterableSize(attributes2));
+        assertEquals(new ValueString(null), attributes2.getOrNull("name"));
         assertEquals(new ValueInt(1l), attributes2.getOrNull("one"));
         assertEquals(new ValueInt(2l), attributes2.getOrNull("two"));
         long timestamp2 = ((ValueTime) attributes2.getOrNull("timestamp")).getValue();