From 85cb5dd1e7dd324268f60d10ae922d66f11e1369 Mon Sep 17 00:00:00 2001
From: Mike Moser <moser.mw@gmail.com>
Date: Thu, 27 Aug 2015 23:15:43 -0400
Subject: [PATCH] NIFI-904 mock controller service fix

Signed-off-by: joewitt <joewitt@apache.org>
---
 .../java/org/apache/nifi/util/StandardProcessorTestRunner.java   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nifi-mock/src/main/java/org/apache/nifi/util/StandardProcessorTestRunner.java b/nifi-mock/src/main/java/org/apache/nifi/util/StandardProcessorTestRunner.java
index f21b981445..80265a74e4 100644
--- a/nifi-mock/src/main/java/org/apache/nifi/util/StandardProcessorTestRunner.java
+++ b/nifi-mock/src/main/java/org/apache/nifi/util/StandardProcessorTestRunner.java
@@ -534,6 +534,7 @@ public class StandardProcessorTestRunner implements TestRunner {
 
         final ComponentLog logger = new MockProcessorLog(identifier, service);
         final MockControllerServiceInitializationContext initContext = new MockControllerServiceInitializationContext(requireNonNull(service), requireNonNull(identifier), logger);
+        initContext.addControllerServices(context);
         service.initialize(initContext);
 
         final Map<PropertyDescriptor, String> resolvedProps = new HashMap<>();