Repository: nifi Updated Branches: refs/heads/develop 596240437 -> 1eeed7301
NIFI-852: Ensure that calling MockProcessSession.transfer(FlowFile, Relationship.SELF) will re-queue the FlowFile Project: http://git-wip-us.apache.org/repos/asf/nifi/repo Commit: http://git-wip-us.apache.org/repos/asf/nifi/commit/1eeed730 Tree: http://git-wip-us.apache.org/repos/asf/nifi/tree/1eeed730 Diff: http://git-wip-us.apache.org/repos/asf/nifi/diff/1eeed730 Branch: refs/heads/develop Commit: 1eeed730103e5c231b27365eda9aec25ec2c8e12 Parents: 5962404 Author: Mark Payne <[email protected]> Authored: Fri Aug 14 09:00:49 2015 -0400 Committer: Mark Payne <[email protected]> Committed: Fri Aug 14 09:35:06 2015 -0400 ---------------------------------------------------------------------- .../java/org/apache/nifi/util/MockProcessSession.java | 10 ++++++++++ 1 file changed, 10 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/nifi/blob/1eeed730/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java ---------------------------------------------------------------------- diff --git a/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java b/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java index a5f9f3c..4ad3138 100644 --- a/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java +++ b/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java @@ -552,6 +552,11 @@ public class MockProcessSession implements ProcessSession { @Override public void transfer(final FlowFile flowFile, final Relationship relationship) { + if (relationship == Relationship.SELF) { + transfer(flowFile); + return; + } + validateState(flowFile); List<MockFlowFile> list = transferMap.get(relationship); if (list == null) { @@ -565,6 +570,11 @@ public class MockProcessSession implements ProcessSession { @Override public void transfer(final Collection<FlowFile> flowFiles, final Relationship relationship) { + if (relationship == Relationship.SELF) { + transfer(flowFiles); + return; + } + for (final FlowFile flowFile : flowFiles) { validateState(flowFile); }
