Apache
/
nifi
mirror of https://github.com/apache/nifi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

NIFI-835: Ensure that on session commit the provenance events that were emitted are returned by TestRunner.getProvenanceEvents()

This commit is contained in:
Mark Payne 2015-08-10 17:32:38 -04:00
parent 33279fd9d7
commit 7cd84967dd
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java
View File

@ -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();
}