NIFI-835: Ensure that on session commit the provenance events that were emitted are returned by TestRunner.getProvenanceEvents()
Project: http://git-wip-us.apache.org/repos/asf/nifi/repo Commit: http://git-wip-us.apache.org/repos/asf/nifi/commit/7cd84967 Tree: http://git-wip-us.apache.org/repos/asf/nifi/tree/7cd84967 Diff: http://git-wip-us.apache.org/repos/asf/nifi/diff/7cd84967 Branch: refs/heads/develop Commit: 7cd84967dd90a122aaeede30b3b640f3591ebdee Parents: 33279fd Author: Mark Payne <[email protected]> Authored: Mon Aug 10 17:32:38 2015 -0400 Committer: Mark Payne <[email protected]> Committed: Mon Aug 10 17:32:38 2015 -0400 ---------------------------------------------------------------------- .../src/main/java/org/apache/nifi/util/MockProcessSession.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/nifi/blob/7cd84967/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 13198dc..a5f9f3c 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 @@ -66,7 +66,7 @@ public class MockProcessSession implements ProcessSession { private final Map<Long, MockFlowFile> originalVersions = new HashMap<>(); private final SharedSessionState sharedState; private final Map<String, Long> counterMap = new HashMap<>(); - private final ProvenanceReporter provenanceReporter; + private final MockProvenanceReporter provenanceReporter; private boolean committed = false; private boolean rolledback = false; @@ -135,6 +135,7 @@ public class MockProcessSession implements ProcessSession { sharedState.adjustCounter(entry.getKey(), entry.getValue()); } + sharedState.addProvenanceEvents(provenanceReporter.getEvents()); counterMap.clear(); }
