Github user MikeThomsen commented on a diff in the pull request:
https://github.com/apache/nifi/pull/2617#discussion_r180204173
--- 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());
+ ff1 = session.unpenalize(ff1);
+ assertEquals(false, ff1.isPenalized());
--- End diff --
Please change to `assertFalse`
---
