From 650c6aa820fca0eb8d35c506c7a2e71ca892370d Mon Sep 17 00:00:00 2001
From: samhjelmfelt <samhjelmfelt@yahoo.com>
Date: Mon, 20 May 2019 16:07:22 -0700
Subject: [PATCH] Small fix after rebase

---
 .../stateless/core/StatelessProcessContext.java   | 15 +++++++++++++++
 .../stateless/core/StatelessProcessorWrapper.java |  2 +-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/nifi-stateless/nifi-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessContext.java b/nifi-stateless/nifi-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessContext.java
index 3766b0741b..d0b6db7d79 100644
--- a/nifi-stateless/nifi-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessContext.java
+++ b/nifi-stateless/nifi-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessContext.java
@@ -27,6 +27,7 @@ import org.apache.nifi.components.state.StateManager;
 import org.apache.nifi.controller.ControllerService;
 import org.apache.nifi.controller.ControllerServiceInitializationContext;
 import org.apache.nifi.controller.ControllerServiceLookup;
+import org.apache.nifi.controller.NodeTypeProvider;
 import org.apache.nifi.logging.ComponentLog;
 import org.apache.nifi.processor.Processor;
 import org.apache.nifi.processor.Relationship;
@@ -392,6 +393,20 @@ public class StatelessProcessContext implements SchedulingContext, ControllerSer
         return stateManager;
     }
 
+    @Override
+    public NodeTypeProvider getNodeTypeProvider() {
+        return new NodeTypeProvider() {
+            @Override
+            public boolean isClustered() {
+                return false;
+            }
+            @Override
+            public boolean isPrimary() {
+                return true;
+            }
+        };
+    }
+
     @Override
     public String getName() {
         return componentName;
diff --git a/nifi-stateless/nifi-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessorWrapper.java b/nifi-stateless/nifi-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessorWrapper.java
index 108b2e2f6c..8ecb17188c 100644
--- a/nifi-stateless/nifi-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessorWrapper.java
+++ b/nifi-stateless/nifi-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessorWrapper.java
@@ -66,7 +66,7 @@ public class StatelessProcessorWrapper extends AbstractStatelessComponent implem
     private volatile boolean initialized = false;
 
 
-    StatelessProcessorWrapper(final String id, final Processor processor, final StatelessProcessorWrapper parent, final StatelessControllerServiceLookup lookup, final VariableRegistry registry,
+    public StatelessProcessorWrapper(final String id, final Processor processor, final StatelessProcessorWrapper parent, final StatelessControllerServiceLookup lookup, final VariableRegistry registry,
                               final boolean materializeContent, final ClassLoader classLoader) throws InvocationTargetException, IllegalAccessException {
 
         this.processor = processor;