Github user MikeThomsen commented on a diff in the pull request:
https://github.com/apache/nifi/pull/2617#discussion_r180204129
--- Diff:
nifi-mock/src/test/java/org/apache/nifi/util/TestMockProcessSession.java ---
@@ -101,6 +101,17 @@ public void
testKeepPenalizedStatusAfterPuttingAttribute(){
assertEquals(true, ff1.isPenalized());
}
+ @Test
+ public void testUnpenalizeFlowFile() {
+ final Processor processor = new PoorlyBehavedProcessor();
+ final MockProcessSession session = new MockProcessSession(new
SharedSessionState(processor, new AtomicLong(0L)), processor);
+ FlowFile ff1 = session.createFlowFile("hello, world".getBytes());
+ ff1 = session.penalize(ff1);
+ assertEquals(true, ff1.isPenalized());
--- End diff --
A little better if it's `assertTrue`
---
