Search in sources :

Example 16 with Query

use of org.apache.apex.malhar.lib.appdata.schemas.Query in project apex-malhar by apache.

the class WEQueryQueueManagerTest method testSimpleRemoveEmpty.

@Test
public void testSimpleRemoveEmpty() {
    WindowEndQueueManager<Query, Void> wqqm = new WindowEndQueueManager<>();
    wqqm.setup(null);
    wqqm.beginWindow(0);
    QueryBundle<Query, Void, MutableLong> qb = wqqm.dequeue();
    Query queryD = qb == null ? null : qb.getQuery();
    Assert.assertEquals("The queries must match.", null, queryD);
    qb = wqqm.dequeue();
    queryD = qb == null ? null : qb.getQuery();
    Assert.assertEquals("The queries must match.", null, queryD);
    wqqm.endWindow();
    wqqm.teardown();
}
Also used : MutableLong(org.apache.commons.lang3.mutable.MutableLong) Query(org.apache.apex.malhar.lib.appdata.schemas.Query) Test(org.junit.Test)

Example 17 with Query

use of org.apache.apex.malhar.lib.appdata.schemas.Query in project apex-malhar by apache.

the class WEQueryQueueManagerTest method testSimpleAddAfterStarted.

@Test
public void testSimpleAddAfterStarted() {
    WindowEndQueueManager<Query, Void> wqqm = new WindowEndQueueManager<>();
    wqqm.setup(null);
    wqqm.beginWindow(0);
    Query query = new MockQuery("0");
    wqqm.enqueue(query, null, new MutableLong(1L));
    Query query1 = new MockQuery("1");
    wqqm.enqueue(query1, null, new MutableLong(1L));
    Query queryD = wqqm.dequeue().getQuery();
    Query query2 = new MockQuery("2");
    wqqm.enqueue(query2, null, new MutableLong(1L));
    Query query1D = wqqm.dequeue().getQuery();
    Query query2D = wqqm.dequeue().getQuery();
    QueryBundle<Query, Void, MutableLong> qb = wqqm.dequeue();
    Query query3D = qb == null ? null : qb.getQuery();
    wqqm.endWindow();
    wqqm.teardown();
    Assert.assertEquals("The queries must match.", query, queryD);
    Assert.assertEquals("The queries must match.", query1, query1D);
    Assert.assertEquals("The queries must match.", query2, query2D);
    Assert.assertEquals("The queries must match.", null, query3D);
}
Also used : MutableLong(org.apache.commons.lang3.mutable.MutableLong) Query(org.apache.apex.malhar.lib.appdata.schemas.Query) Test(org.junit.Test)

Example 18 with Query

use of org.apache.apex.malhar.lib.appdata.schemas.Query in project apex-malhar by apache.

the class WEQueryQueueManagerTest method testSimpleAddOneRemove.

@Test
public void testSimpleAddOneRemove() {
    WindowEndQueueManager<Query, Void> wqqm = new WindowEndQueueManager<>();
    wqqm.setup(null);
    wqqm.beginWindow(0);
    Query query = new MockQuery("1");
    wqqm.enqueue(query, null, new MutableLong(1L));
    Query queryD = wqqm.dequeue().getQuery();
    QueryBundle<Query, Void, MutableLong> qb = wqqm.dequeue();
    Query queryD1 = qb == null ? null : qb.getQuery();
    wqqm.endWindow();
    wqqm.teardown();
    Assert.assertEquals("The queries must match.", query, queryD);
    Assert.assertEquals("The queries must match.", null, queryD1);
}
Also used : MutableLong(org.apache.commons.lang3.mutable.MutableLong) Query(org.apache.apex.malhar.lib.appdata.schemas.Query) Test(org.junit.Test)

Example 19 with Query

use of org.apache.apex.malhar.lib.appdata.schemas.Query in project apex-malhar by apache.

the class SimpleQueryQueueManagerTest method simpleTest.

@Test
public void simpleTest() {
    SimpleQueueManager<Query, Void, Void> sqqm = new SimpleQueueManager<Query, Void, Void>();
    sqqm.setup(null);
    sqqm.beginWindow(0);
    Query query = new MockQuery("1");
    sqqm.enqueue(query, null, null);
    Query queryD = sqqm.dequeue().getQuery();
    QueryBundle<Query, Void, Void> qb = sqqm.dequeue();
    Query queryD1 = qb == null ? null : qb.getQuery();
    sqqm.teardown();
    sqqm.teardown();
    Assert.assertEquals("The query object must equal", query, queryD);
    Assert.assertEquals("The query object must equal", null, queryD1);
}
Also used : Query(org.apache.apex.malhar.lib.appdata.schemas.Query) Test(org.junit.Test)

Aggregations

Query (org.apache.apex.malhar.lib.appdata.schemas.Query)19 Test (org.junit.Test)19 MutableBoolean (org.apache.commons.lang3.mutable.MutableBoolean)11 MutableLong (org.apache.commons.lang3.mutable.MutableLong)6 ExceptionSaverExceptionHandler (org.apache.apex.malhar.lib.appdata.ThreadUtils.ExceptionSaverExceptionHandler)1 Result (org.apache.apex.malhar.lib.appdata.schemas.Result)1