org.apache.nifi.util

Class MockProcessContext

java.lang.Object

org.apache.nifi.util.MockControllerServiceLookup

org.apache.nifi.util.MockProcessContext

All Implemented Interfaces:

ControllerServiceLookup, ProcessContext, SchedulingContext

public class MockProcessContext
extends MockControllerServiceLookup
implements SchedulingContext, ControllerServiceLookup

Field Summary

Fields

Modifier and Type	Field and Description
private boolean	allowExpressionValidation
private String	annotationData
private ConfigurableComponent	component
private Set<Relationship>	connections
private boolean	enableExpressionValidation
private boolean	incomingConnection
private int	maxConcurrentTasks
private boolean	nonLoopConnection
private Map<PropertyDescriptor,String>	properties
private StateManager	stateManager
private Set<Relationship>	unavailableRelationships
private VariableRegistry	variableRegistry
private boolean	yieldCalled

Constructor Summary

Constructors

Constructor and Description
MockProcessContext(ConfigurableComponent component, StateManager stateManager, VariableRegistry variableRegistry) Creates a new MockProcessContext for the given Processor
MockProcessContext(ConfigurableComponent component, VariableRegistry variableRegistry)
MockProcessContext(ControllerService component, MockProcessContext context, StateManager stateManager, VariableRegistry variableRegistry)

Method Summary

Methods

Modifier and Type	Method and Description
void	addConnection(Relationship relationship)
void	addControllerService(String serviceIdentifier, ControllerService controllerService, Map<PropertyDescriptor,String> properties, String annotationData)
void	assertValid()
String	decrypt(String encrypted)
void	disableExpressionValidation()
void	enableExpressionValidation()
String	encrypt(String unencrypted)
String	getAnnotationData()
Set<Relationship>	getAvailableRelationships()
(package private) String	getControllerServiceAnnotationData(ControllerService controllerService)
ControllerServiceLookup	getControllerServiceLookup()
(package private) Map<PropertyDescriptor,String>	getControllerServiceProperties(ControllerService controllerService)
int	getMaxConcurrentTasks()
String	getName()
Map<PropertyDescriptor,String>	getProperties()
PropertyValue	getProperty(PropertyDescriptor descriptor)
PropertyValue	getProperty(String propertyName)
StateManager	getStateManager()
Set<Relationship>	getUnavailableRelationships()
boolean	hasConnection(Relationship relationship)
boolean	hasIncomingConnection()
boolean	hasNonLoopConnection()
boolean	isExpressionLanguagePresent(PropertyDescriptor property)
boolean	isValid()
boolean	isYieldCalled()
void	leaseControllerService(String identifier)
PropertyValue	newPropertyValue(String rawValue)
void	removeConnection(Relationship relationship)
boolean	removeProperty(PropertyDescriptor descriptor)
void	setAnnotationData(String annotationData)
void	setConnections(Set<Relationship> connections)
void	setIncomingConnection(boolean hasIncomingConnection)
protected void	setMaxConcurrentTasks(int maxConcurrentTasks)
void	setNonLoopConnection(boolean hasNonLoopConnection)
ValidationResult	setProperty(PropertyDescriptor descriptor, String value) Updates the value of the property with the given PropertyDescriptor to the specified value IF and ONLY IF the value is valid according to the descriptor's validator.
ValidationResult	setProperty(String propertyName, String propertyValue)
void	setUnavailableRelationships(Set<Relationship> relationships)
void	setValidateExpressionUsage(boolean validate)
Collection<ValidationResult>	validate() Validates the current properties, returning ValidationResults for any invalid properties.
void	yield()

Methods inherited from class org.apache.nifi.util.MockControllerServiceLookup

addControllerService, addControllerService, addControllerServices, getConfiguration, getControllerService, getControllerServiceIdentifiers, getControllerServiceName, isControllerServiceEnabled, isControllerServiceEnabled, isControllerServiceEnabling, removeControllerService

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.nifi.controller.ControllerServiceLookup

getControllerService, getControllerServiceIdentifiers, getControllerServiceName, isControllerServiceEnabled, isControllerServiceEnabled, isControllerServiceEnabling

Field Detail

component

private final ConfigurableComponent component

properties

private final Map<PropertyDescriptor,String> properties

stateManager

private final StateManager stateManager

variableRegistry

private final VariableRegistry variableRegistry

annotationData

private String annotationData

yieldCalled

private boolean yieldCalled

enableExpressionValidation

private boolean enableExpressionValidation

allowExpressionValidation

private boolean allowExpressionValidation

incomingConnection

private volatile boolean incomingConnection

nonLoopConnection

private volatile boolean nonLoopConnection

maxConcurrentTasks

private int maxConcurrentTasks

connections

private volatile Set<Relationship> connections

unavailableRelationships

private volatile Set<Relationship> unavailableRelationships

Constructor Detail

MockProcessContext

public MockProcessContext(ConfigurableComponent component,
                  VariableRegistry variableRegistry)

MockProcessContext

public MockProcessContext(ConfigurableComponent component,
                  StateManager stateManager,
                  VariableRegistry variableRegistry)

Creates a new MockProcessContext for the given Processor

Parameters:

component - being mocked

variableRegistry - variableRegistry

MockProcessContext

public MockProcessContext(ControllerService component,
                  MockProcessContext context,
                  StateManager stateManager,
                  VariableRegistry variableRegistry)

Method Detail

getProperty

public PropertyValue getProperty(PropertyDescriptor descriptor)

Specified by:

getProperty in interface ProcessContext

getProperty

public PropertyValue getProperty(String propertyName)

Specified by:

getProperty in interface ProcessContext

newPropertyValue

public PropertyValue newPropertyValue(String rawValue)

Specified by:

newPropertyValue in interface ProcessContext

setProperty

public ValidationResult setProperty(String propertyName,
                           String propertyValue)

setProperty

public ValidationResult setProperty(PropertyDescriptor descriptor,
                           String value)

Updates the value of the property with the given PropertyDescriptor to the specified value IF and ONLY IF the value is valid according to the descriptor's validator. Otherwise, the property value is not updated. In either case, the ValidationResult is returned, indicating whether or not the property is valid

Parameters:

descriptor - of property to modify

value - new value

Returns:

result

removeProperty

public boolean removeProperty(PropertyDescriptor descriptor)

yield

public void yield()

Specified by:

yield in interface ProcessContext

isYieldCalled

public boolean isYieldCalled()

addControllerService

public void addControllerService(String serviceIdentifier,
                        ControllerService controllerService,
                        Map<PropertyDescriptor,String> properties,
                        String annotationData)

getMaxConcurrentTasks

public int getMaxConcurrentTasks()

Specified by:

getMaxConcurrentTasks in interface ProcessContext

setAnnotationData

public void setAnnotationData(String annotationData)

getAnnotationData

public String getAnnotationData()

Specified by:

getAnnotationData in interface ProcessContext

getProperties

public Map<PropertyDescriptor,String> getProperties()

Specified by:

getProperties in interface ProcessContext

validate

public Collection<ValidationResult> validate()

Validates the current properties, returning ValidationResults for any invalid properties. All processor defined properties will be validated. If they are not included in the in the purposed configuration, the default value will be used.

Returns:

Collection of validation result objects for any invalid findings only. If the collection is empty then the processor is valid. Guaranteed non-null

isValid

public boolean isValid()

assertValid

public void assertValid()

encrypt

public String encrypt(String unencrypted)

Specified by:

encrypt in interface ProcessContext

decrypt

public String decrypt(String encrypted)

Specified by:

decrypt in interface ProcessContext

setValidateExpressionUsage

public void setValidateExpressionUsage(boolean validate)

enableExpressionValidation

public void enableExpressionValidation()

disableExpressionValidation

public void disableExpressionValidation()

getControllerServiceProperties

Map<PropertyDescriptor,String> getControllerServiceProperties(ControllerService controllerService)

getControllerServiceAnnotationData

String getControllerServiceAnnotationData(ControllerService controllerService)

getControllerServiceLookup

public ControllerServiceLookup getControllerServiceLookup()

Specified by:

getControllerServiceLookup in interface ProcessContext

leaseControllerService

public void leaseControllerService(String identifier)

Specified by:

leaseControllerService in interface SchedulingContext

getAvailableRelationships

public Set<Relationship> getAvailableRelationships()

Specified by:

getAvailableRelationships in interface ProcessContext

setUnavailableRelationships

public void setUnavailableRelationships(Set<Relationship> relationships)

getUnavailableRelationships

public Set<Relationship> getUnavailableRelationships()

hasIncomingConnection

public boolean hasIncomingConnection()

Specified by:

hasIncomingConnection in interface ProcessContext

setIncomingConnection

public void setIncomingConnection(boolean hasIncomingConnection)

hasConnection

public boolean hasConnection(Relationship relationship)

Specified by:

hasConnection in interface ProcessContext

setNonLoopConnection

public void setNonLoopConnection(boolean hasNonLoopConnection)

hasNonLoopConnection

public boolean hasNonLoopConnection()

Specified by:

hasNonLoopConnection in interface ProcessContext

addConnection

public void addConnection(Relationship relationship)

removeConnection

public void removeConnection(Relationship relationship)

setConnections

public void setConnections(Set<Relationship> connections)

isExpressionLanguagePresent

public boolean isExpressionLanguagePresent(PropertyDescriptor property)

Specified by:

isExpressionLanguagePresent in interface ProcessContext

getStateManager

public StateManager getStateManager()

Specified by:

getStateManager in interface ProcessContext

getName

public String getName()

Specified by:

getName in interface ProcessContext

setMaxConcurrentTasks

protected void setMaxConcurrentTasks(int maxConcurrentTasks)