Java Code Examples for org.apache.nifi.util.TestRunner#assertQueueNotEmpty()

The following examples show how to use org.apache.nifi.util.TestRunner#assertQueueNotEmpty() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: TestUpdateAttribute.java From localization_nifi with Apache License 2.0

5 votes

@Test
public void testStateFailures() throws Exception {
    final TestRunner runner = TestRunners.newTestRunner(new UpdateAttribute());
    final UpdateAttribute processor = (UpdateAttribute) runner.getProcessor();
    final ProcessSessionFactory processSessionFactory = runner.getProcessSessionFactory();
    MockStateManager mockStateManager = runner.getStateManager();

    runner.setProperty(UpdateAttribute.STORE_STATE, STORE_STATE_LOCALLY);
    runner.setProperty("count", "${getStateValue('count'):plus(1)}");
    runner.setProperty("sum", "${getStateValue('sum'):plus(${pencils})}");
    runner.setProperty(UpdateAttribute.STATEFUL_VARIABLES_INIT_VALUE, "0");

    processor.onScheduled(runner.getProcessContext());

    final Map<String, String> attributes2 = new HashMap<>();
    attributes2.put("pencils", "2");

    mockStateManager.setFailOnStateGet(Scope.LOCAL, true);

    runner.enqueue(new byte[0],attributes2);
    processor.onTrigger(runner.getProcessContext(), processSessionFactory.createSession());

    runner.assertQueueNotEmpty();

    mockStateManager.setFailOnStateGet(Scope.LOCAL, false);
    mockStateManager.setFailOnStateSet(Scope.LOCAL, true);

    processor.onTrigger(runner.getProcessContext(), processSessionFactory.createSession());

    runner.assertQueueEmpty();

    runner.assertAllFlowFilesTransferred(UpdateAttribute.REL_FAILED_SET_STATE, 1);
    runner.getFlowFilesForRelationship(UpdateAttribute.REL_FAILED_SET_STATE).get(0).assertAttributeEquals("count", "1");
    runner.getFlowFilesForRelationship(UpdateAttribute.REL_FAILED_SET_STATE).get(0).assertAttributeEquals("sum", "2");
}

Example 2

Source File: TestMergeContent.java From nifi with Apache License 2.0

5 votes

/**
 * This test will verify that if we have a FlowFile larger than the Max Size for a Bin, it will go into its
 * own bin and immediately be processed as its own bin.
 */
@Test
public void testFlowFileLargerThanBin() {
    final TestRunner runner = TestRunners.newTestRunner(new MergeContent());
    runner.setProperty(MergeContent.MERGE_STRATEGY, MergeContent.MERGE_STRATEGY_BIN_PACK);
    runner.setProperty(MergeContent.MIN_ENTRIES, "2");
    runner.setProperty(MergeContent.MAX_ENTRIES, "2");
    runner.setProperty(MergeContent.MIN_SIZE, "1 KB");
    runner.setProperty(MergeContent.MAX_SIZE, "5 KB");

    runner.enqueue(new byte[1026]); // add flowfile that fits within the bin limits
    runner.enqueue(new byte[1024 * 6]); // add flowfile that is larger than the bin limit
    runner.run(2); // run twice so that we have a chance to create two bins (though we shouldn't create 2, because only 1 bin will be full)

    runner.assertTransferCount(MergeContent.REL_ORIGINAL, 1);
    runner.assertTransferCount(MergeContent.REL_MERGED, 1);
    runner.assertTransferCount(MergeContent.REL_FAILURE, 0);
    assertEquals(runner.getFlowFilesForRelationship(MergeContent.REL_MERGED).get(0).getAttribute(CoreAttributes.UUID.key()),
            runner.getFlowFilesForRelationship(MergeContent.REL_ORIGINAL).get(0).getAttribute(MergeContent.MERGE_UUID_ATTRIBUTE));

    final MockFlowFile bundle = runner.getFlowFilesForRelationship(MergeContent.REL_MERGED).get(0);
    assertEquals(1024 * 6, bundle.getSize());

    // Queue should not be empty because the first FlowFile will be transferred back to the input queue
    // when we run out @OnStopped logic, since it won't be transferred to any bin.
    runner.assertQueueNotEmpty();
}

Example 3