From b3dd7a505549662d558eec550f210f0d6f0ec547 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Magdalena=20Grodzi=C5=84ska?= Date: Tue, 14 Jan 2020 21:53:53 +0100 Subject: Fix tests after value query change --- .../mimuw/cloudatlas/agent/modules/StanikTest.java | 57 +++++++++++----------- 1 file changed, 29 insertions(+), 28 deletions(-) (limited to 'src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/StanikTest.java') diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/StanikTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/StanikTest.java index 92ba051..3ad176a 100644 --- a/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/StanikTest.java +++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/StanikTest.java @@ -60,7 +60,7 @@ public class StanikTest { assertEquals(3, TestUtil.iterableSize(zmi.getAttributes())); assertEquals(new ValueInt(0l), zmi.getAttributes().getOrNull("level")); assertEquals(new ValueString("/new"), zmi.getAttributes().getOrNull("owner")); - Map> queries = stateMessage.getQueries(); + Map queries = stateMessage.getQueries(); assertEquals(2, TestUtil.iterableSize(queries.keySet())); } @@ -218,57 +218,58 @@ public class StanikTest { @Test public void addQuery() throws Exception { - HashMap> queries = new HashMap>(); - queries.put(new Attribute("&query"), new SimpleImmutableEntry(new ValueQuery("SELECT 1 AS one"), new ValueTime(42l))); + HashMap queries = new HashMap(); + queries.put(new Attribute("&query"), new ValueQuery("SELECT 1 AS one", 42l)); UpdateQueriesMessage message = new UpdateQueriesMessage("test_msg", 0, queries); + System.out.println(message); stanik.handleTyped(message); - HashMap> actualQueries = stanik.getQueries(); + HashMap actualQueries = stanik.getQueries(); assertEquals(3, TestUtil.iterableSize(actualQueries.keySet())); assertTrue(actualQueries.containsKey(new Attribute("&query"))); - Entry timestampedQuery = actualQueries.get(new Attribute("&query")); - assertEquals(new ValueTime(42l), timestampedQuery.getValue()); - assertEquals(new ValueQuery("SELECT 1 AS one"), timestampedQuery.getKey()); + ValueQuery timestampedQuery = actualQueries.get(new Attribute("&query")); + assertEquals(42l, timestampedQuery.getTimestamp()); + assertEquals("SELECT 1 AS one", timestampedQuery.getCode()); } @Test public void updateQueries() throws Exception { - HashMap> queries = new HashMap>(); - queries.put(new Attribute("&query1"), new SimpleImmutableEntry(new ValueQuery("SELECT 1 AS one"), new ValueTime(42l))); - queries.put(new Attribute("&query3"), new SimpleImmutableEntry(new ValueQuery("SELECT 23 AS x"), new ValueTime(43l))); - queries.put(new Attribute("&query4"), new SimpleImmutableEntry(new ValueQuery("SELECT 1000 AS foo"), new ValueTime(43l))); + HashMap queries = new HashMap(); + queries.put(new Attribute("&query1"), new ValueQuery("SELECT 1 AS one", 42l)); + queries.put(new Attribute("&query3"), new ValueQuery("SELECT 23 AS x", 43l)); + queries.put(new Attribute("&query4"), new ValueQuery("SELECT 1000 AS foo", 43l)); UpdateQueriesMessage message = new UpdateQueriesMessage("test_msg", 0, queries); stanik.handleTyped(message); - HashMap> otherQueries = new HashMap>(); - otherQueries.put(new Attribute("&query1"), new SimpleImmutableEntry(new ValueQuery("SELECT 2 AS one"), new ValueTime(41l))); - otherQueries.put(new Attribute("&query2"), new SimpleImmutableEntry(new ValueQuery("SELECT 42 AS answer"), new ValueTime(39l))); - otherQueries.put(new Attribute("&query3"), new SimpleImmutableEntry(new ValueQuery("SELECT 17 AS y"), new ValueTime(44l))); + HashMap otherQueries = new HashMap(); + otherQueries.put(new Attribute("&query1"), new ValueQuery("SELECT 2 AS one", 41l)); + otherQueries.put(new Attribute("&query2"), new ValueQuery("SELECT 42 AS answer", 39l)); + otherQueries.put(new Attribute("&query3"), new ValueQuery("SELECT 17 AS y", 44l)); UpdateQueriesMessage otherMessage = new UpdateQueriesMessage("test_msg", 0, otherQueries); stanik.handleTyped(otherMessage); - HashMap> actualQueries = stanik.getQueries(); + HashMap actualQueries = stanik.getQueries(); assertEquals(6, TestUtil.iterableSize(actualQueries.keySet())); assertTrue(actualQueries.containsKey(new Attribute("&query1"))); assertTrue(actualQueries.containsKey(new Attribute("&query2"))); assertTrue(actualQueries.containsKey(new Attribute("&query3"))); assertTrue(actualQueries.containsKey(new Attribute("&query4"))); - Entry timestampedQuery1 = actualQueries.get(new Attribute("&query1")); - assertEquals(new ValueTime(42l), timestampedQuery1.getValue()); - assertEquals(new ValueQuery("SELECT 1 AS one"), timestampedQuery1.getKey()); + ValueQuery timestampedQuery1 = actualQueries.get(new Attribute("&query1")); + assertEquals(42l, timestampedQuery1.getTimestamp()); + assertEquals("SELECT 1 AS one", timestampedQuery1.getCode()); - Entry timestampedQuery2 = actualQueries.get(new Attribute("&query2")); - assertEquals(new ValueTime(39l), timestampedQuery2.getValue()); - assertEquals(new ValueQuery("SELECT 42 AS answer"), timestampedQuery2.getKey()); + ValueQuery timestampedQuery2 = actualQueries.get(new Attribute("&query2")); + assertEquals(39l, timestampedQuery2.getTimestamp()); + assertEquals("SELECT 42 AS answer", timestampedQuery2.getCode()); - Entry timestampedQuery3 = actualQueries.get(new Attribute("&query3")); - assertEquals(new ValueTime(44l), timestampedQuery3.getValue()); - assertEquals(new ValueQuery("SELECT 17 AS y"), timestampedQuery3.getKey()); + ValueQuery timestampedQuery3 = actualQueries.get(new Attribute("&query3")); + assertEquals(44l, timestampedQuery3.getTimestamp()); + assertEquals("SELECT 17 AS y", timestampedQuery3.getCode()); - Entry timestampedQuery4 = actualQueries.get(new Attribute("&query4")); - assertEquals(new ValueTime(43l), timestampedQuery4.getValue()); - assertEquals(new ValueQuery("SELECT 1000 AS foo"), timestampedQuery4.getKey()); + ValueQuery timestampedQuery4 = actualQueries.get(new Attribute("&query4")); + assertEquals(43l, timestampedQuery4.getTimestamp()); + assertEquals("SELECT 1000 AS foo", timestampedQuery4.getCode()); } @Test -- cgit v1.2.3