Examples with ChannelResponder org.apache.nifi.processor.util.listen.response.ChannelResponder used on opensource projects

Example 6 with ChannelResponder

use of org.apache.nifi.processor.util.listen.response.ChannelResponder in project nifi by apache.

the class TestListenUDP method testBatchingWithDifferentSenders.

@Test
public void testBatchingWithDifferentSenders() throws IOException, InterruptedException {
    final String sender1 = "sender1";
    final String sender2 = "sender2";
    final ChannelResponder responder = Mockito.mock(ChannelResponder.class);
    final byte[] message = "test message".getBytes(StandardCharsets.UTF_8);
    final List<StandardEvent> mockEvents = new ArrayList<>();
    mockEvents.add(new StandardEvent(sender1, message, responder));
    mockEvents.add(new StandardEvent(sender1, message, responder));
    mockEvents.add(new StandardEvent(sender2, message, responder));
    mockEvents.add(new StandardEvent(sender2, message, responder));
    MockListenUDP mockListenUDP = new MockListenUDP(mockEvents);
    runner = TestRunners.newTestRunner(mockListenUDP);
    runner.setProperty(ListenRELP.PORT, "1");
    runner.setProperty(ListenRELP.MAX_BATCH_SIZE, "10");
    // sending 4 messages with a batch size of 10, but should get 2 FlowFiles because of different senders
    runner.run();
    runner.assertAllFlowFilesTransferred(ListenRELP.REL_SUCCESS, 2);
    verifyProvenance(2);
}

Example 7 with ChannelResponder

use of org.apache.nifi.processor.util.listen.response.ChannelResponder in project nifi by apache.

the class TestBeatsEventFactory method testCreateLumberJackEvent.

@Test
public void testCreateLumberJackEvent() {
    final String sender = "testsender1";
    final byte[] data = "this is a test line".getBytes();
    final int seqNumber = 1;
    final String fields = "{\"file\":\"test\"}";
    final Map<String, String> metadata = new HashMap<>();
    metadata.put(EventFactory.SENDER_KEY, sender);
    metadata.put(BeatsMetadata.SEQNUMBER_KEY, String.valueOf(seqNumber));
    final ChannelResponder responder = new SocketChannelResponder(null);
    final EventFactory<BeatsEvent> factory = new BeatsEventFactory();
    final BeatsEvent event = factory.create(data, metadata, responder);
    Assert.assertEquals(sender, event.getSender());
    Assert.assertEquals(seqNumber, event.getSeqNumber());
    Assert.assertEquals(data, event.getData());
}