From 7cd84967dd90a122aaeede30b3b640f3591ebdee Mon Sep 17 00:00:00 2001
From: Mark Payne <markap14@hotmail.com>
Date: Mon, 10 Aug 2015 17:32:38 -0400
Subject: [PATCH] NIFI-835: Ensure that on session commit the provenance events
 that were emitted are returned by TestRunner.getProvenanceEvents()

---
 .../src/main/java/org/apache/nifi/util/MockProcessSession.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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 13198dc0bb..a5f9f3c7bf 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();
     }