Search in sources :

Example 6 with BasicFilterOperation

use of com.nextdoor.bender.operation.filter.BasicFilterOperation in project bender by Nextdoor.

the class ConditionalOperationTest method testTwoConditions.

@Test
public void testTwoConditions() {
    List<Pair<FilterOperation, List<OperationProcessor>>> conditions = new ArrayList<Pair<FilterOperation, List<OperationProcessor>>>();
    /*
     * Case 1
     */
    List<OperationProcessor> case1Ops = new ArrayList<OperationProcessor>();
    DummyAppendOperationFactory pos = new DummyAppendOperationFactory();
    DummyAppendOperationConfig posConf = new DummyAppendOperationConfig();
    posConf.setAppendStr("+");
    pos.setConf(posConf);
    case1Ops.add(new OperationProcessor(pos));
    FilterOperation case1Filter = new BasicFilterOperation(false);
    conditions.add(new ImmutablePair<FilterOperation, List<OperationProcessor>>(case1Filter, case1Ops));
    /*
     * Case 2
     */
    List<OperationProcessor> case2Ops = new ArrayList<OperationProcessor>();
    DummyAppendOperationFactory neg = new DummyAppendOperationFactory();
    DummyAppendOperationConfig negConf = new DummyAppendOperationConfig();
    negConf.setAppendStr("-");
    neg.setConf(negConf);
    case2Ops.add(new OperationProcessor(neg));
    FilterOperation case2Filter = new BasicFilterOperation(true);
    conditions.add(new ImmutablePair<FilterOperation, List<OperationProcessor>>(case2Filter, case2Ops));
    ConditionalOperation op = new ConditionalOperation(conditions, false);
    /*
     * Create thread that supplies input events
     */
    Queue<InternalEvent> inputQueue = new Queue<InternalEvent>();
    supply(2, inputQueue);
    /*
     * Process
     */
    Stream<InternalEvent> input = inputQueue.stream();
    Stream<InternalEvent> output = op.getOutputStream(input);
    List<String> actual = output.map(m -> {
        return m.getEventObj().getPayload().toString();
    }).collect(Collectors.toList());
    List<String> expected = Arrays.asList("0-", "1-");
    assertEquals(2, actual.size());
    assertTrue(expected.containsAll(actual));
}
Also used : Arrays(java.util.Arrays) Monitor(com.nextdoor.bender.monitoring.Monitor) FilterOperation(com.nextdoor.bender.operation.FilterOperation) Assert.assertTrue(org.junit.Assert.assertTrue) Test(org.junit.Test) DummyStringEvent(com.nextdoor.bender.testutils.DummyDeserializerHelper.DummyStringEvent) InternalEvent(com.nextdoor.bender.InternalEvent) DummyAppendOperationConfig(com.nextdoor.bender.testutils.DummyAppendOperationHelper.DummyAppendOperationConfig) Collectors(java.util.stream.Collectors) OperationProcessor(com.nextdoor.bender.operation.OperationProcessor) ImmutablePair(org.apache.commons.lang3.tuple.ImmutablePair) ArrayList(java.util.ArrayList) List(java.util.List) Stream(java.util.stream.Stream) ConditionalOperation(com.nextdoor.bender.operation.conditional.ConditionalOperation) Queue(com.oath.cyclops.async.adapters.Queue) Pair(org.apache.commons.lang3.tuple.Pair) BasicFilterOperation(com.nextdoor.bender.operation.filter.BasicFilterOperation) DummyAppendOperationFactory(com.nextdoor.bender.testutils.DummyAppendOperationHelper.DummyAppendOperationFactory) BasicFilterOperationConfig(com.nextdoor.bender.operation.filter.BasicFilterOperationConfig) Assert.assertEquals(org.junit.Assert.assertEquals) Before(org.junit.Before) BasicFilterOperationFactory(com.nextdoor.bender.operation.filter.BasicFilterOperationFactory) ArrayList(java.util.ArrayList) OperationProcessor(com.nextdoor.bender.operation.OperationProcessor) InternalEvent(com.nextdoor.bender.InternalEvent) DummyAppendOperationFactory(com.nextdoor.bender.testutils.DummyAppendOperationHelper.DummyAppendOperationFactory) ConditionalOperation(com.nextdoor.bender.operation.conditional.ConditionalOperation) FilterOperation(com.nextdoor.bender.operation.FilterOperation) BasicFilterOperation(com.nextdoor.bender.operation.filter.BasicFilterOperation) DummyAppendOperationConfig(com.nextdoor.bender.testutils.DummyAppendOperationHelper.DummyAppendOperationConfig) ArrayList(java.util.ArrayList) List(java.util.List) Queue(com.oath.cyclops.async.adapters.Queue) ImmutablePair(org.apache.commons.lang3.tuple.ImmutablePair) Pair(org.apache.commons.lang3.tuple.Pair) BasicFilterOperation(com.nextdoor.bender.operation.filter.BasicFilterOperation) Test(org.junit.Test)

Aggregations

InternalEvent (com.nextdoor.bender.InternalEvent)6 Monitor (com.nextdoor.bender.monitoring.Monitor)6 FilterOperation (com.nextdoor.bender.operation.FilterOperation)6 OperationProcessor (com.nextdoor.bender.operation.OperationProcessor)6 ConditionalOperation (com.nextdoor.bender.operation.conditional.ConditionalOperation)6 BasicFilterOperation (com.nextdoor.bender.operation.filter.BasicFilterOperation)6 BasicFilterOperationConfig (com.nextdoor.bender.operation.filter.BasicFilterOperationConfig)6 BasicFilterOperationFactory (com.nextdoor.bender.operation.filter.BasicFilterOperationFactory)6 DummyAppendOperationConfig (com.nextdoor.bender.testutils.DummyAppendOperationHelper.DummyAppendOperationConfig)6 DummyAppendOperationFactory (com.nextdoor.bender.testutils.DummyAppendOperationHelper.DummyAppendOperationFactory)6 DummyStringEvent (com.nextdoor.bender.testutils.DummyDeserializerHelper.DummyStringEvent)6 Queue (com.oath.cyclops.async.adapters.Queue)6 ArrayList (java.util.ArrayList)6 Arrays (java.util.Arrays)6 List (java.util.List)6 Collectors (java.util.stream.Collectors)6 Stream (java.util.stream.Stream)6 ImmutablePair (org.apache.commons.lang3.tuple.ImmutablePair)6 Pair (org.apache.commons.lang3.tuple.Pair)6 Assert.assertEquals (org.junit.Assert.assertEquals)6