co.cask.cdap.etl.mock.action

Class MockActionContext

java.lang.Object

co.cask.cdap.etl.mock.action.MockActionContext

All Implemented Interfaces:

MetadataReader, MetadataWriter, SecureStore, SecureStoreManager, ServiceDiscoverer, Transactional, ActionContext, LineageRecorder, StageContext

public class MockActionContext
extends Object
implements ActionContext

Mock ActionContext for CustomAction tests.

Constructor Summary

Constructors

Constructor and Description
MockActionContext()

Method Summary

Modifier and Type	Method and Description
void	addProperties(MetadataEntity metadataEntity, Map<String,String> properties)
void	addTags(MetadataEntity metadataEntity, Iterable<String> tags)
void	addTags(MetadataEntity metadataEntity, String... tags)
void	deleteSecureData(String namespace, String name)
void	execute(int timeoutInSeconds, TxRunnable runnable)
void	execute(TxRunnable runnable)
SettableArguments	getArguments()
co.cask.cdap.api.data.schema.Schema	getInputSchema()
Map<String,co.cask.cdap.api.data.schema.Schema>	getInputSchemas()
long	getLogicalStartTime()
Map<MetadataScope,Metadata>	getMetadata(MetadataEntity metadataEntity)
Metadata	getMetadata(MetadataScope scope, MetadataEntity metadataEntity)
StageMetrics	getMetrics()
String	getNamespace()
Map<String,co.cask.cdap.api.data.schema.Schema>	getOutputPortSchemas()
co.cask.cdap.api.data.schema.Schema	getOutputSchema()
String	getPipelineName()
PluginProperties	getPluginProperties()
PluginProperties	getPluginProperties(String pluginId)
SecureStoreData	getSecureData(String namespace, String name)
URL	getServiceURL(String serviceId)
URL	getServiceURL(String applicationId, String serviceId)
String	getStageName()
Map<String,String>	listSecureData(String namespace)
<T> Class<T>	loadPluginClass(String pluginId)
<T> T	newPluginInstance(String pluginId)
void	putSecureData(String namespace, String name, String data, String description, Map<String,String> properties)
void	record(List<FieldOperation> fieldOperations)
void	removeMetadata(MetadataEntity metadataEntity)
void	removeProperties(MetadataEntity metadataEntity)
void	removeProperties(MetadataEntity metadataEntity, String... keys)
void	removeTags(MetadataEntity metadataEntity)
void	removeTags(MetadataEntity metadataEntity, String... tags)

Methods inherited from class java.lang.Object

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

Constructor Detail

MockActionContext

public MockActionContext()

Method Detail

getLogicalStartTime

public long getLogicalStartTime()

Specified by:

getLogicalStartTime in interface StageContext

getArguments

public SettableArguments getArguments()

Specified by:

getArguments in interface ActionContext

Specified by:

getArguments in interface StageContext

getStageName

public String getStageName()

Specified by:

getStageName in interface StageContext

getMetrics

public StageMetrics getMetrics()

Specified by:

getMetrics in interface StageContext

getPluginProperties

public PluginProperties getPluginProperties()

Specified by:

getPluginProperties in interface StageContext

getPluginProperties

public PluginProperties getPluginProperties(String pluginId)

Specified by:

getPluginProperties in interface StageContext

loadPluginClass

public <T> Class<T> loadPluginClass(String pluginId)

Specified by:

loadPluginClass in interface StageContext

newPluginInstance

public <T> T newPluginInstance(String pluginId)
                        throws InstantiationException

Specified by:

newPluginInstance in interface StageContext

Throws:

InstantiationException

getInputSchema

@Nullable
public co.cask.cdap.api.data.schema.Schema getInputSchema()

Specified by:

getInputSchema in interface StageContext

getInputSchemas

public Map<String,co.cask.cdap.api.data.schema.Schema> getInputSchemas()

Specified by:

getInputSchemas in interface StageContext

getOutputSchema

@Nullable
public co.cask.cdap.api.data.schema.Schema getOutputSchema()

Specified by:

getOutputSchema in interface StageContext

getOutputPortSchemas

public Map<String,co.cask.cdap.api.data.schema.Schema> getOutputPortSchemas()

Specified by:

getOutputPortSchemas in interface StageContext

getServiceURL

@Nullable
public URL getServiceURL(String applicationId,
                                   String serviceId)

Specified by:

getServiceURL in interface ServiceDiscoverer

getServiceURL

@Nullable
public URL getServiceURL(String serviceId)

Specified by:

getServiceURL in interface ServiceDiscoverer

getNamespace

public String getNamespace()

Specified by:

getNamespace in interface StageContext

getPipelineName

public String getPipelineName()

Specified by:

getPipelineName in interface StageContext

listSecureData

public Map<String,String> listSecureData(String namespace)

Specified by:

listSecureData in interface SecureStore

getSecureData

public SecureStoreData getSecureData(String namespace,
                                     String name)

Specified by:

getSecureData in interface SecureStore

putSecureData

public void putSecureData(String namespace,
                          String name,
                          String data,
                          String description,
                          Map<String,String> properties)

Specified by:

putSecureData in interface SecureStoreManager

deleteSecureData

public void deleteSecureData(String namespace,
                             String name)

Specified by:

deleteSecureData in interface SecureStoreManager

execute

public void execute(TxRunnable runnable)

Specified by:

execute in interface Transactional

execute

public void execute(int timeoutInSeconds,
                    TxRunnable runnable)

Specified by:

execute in interface Transactional

getMetadata

public Map<MetadataScope,Metadata> getMetadata(MetadataEntity metadataEntity)

Specified by:

getMetadata in interface MetadataReader

getMetadata

public Metadata getMetadata(MetadataScope scope,
                            MetadataEntity metadataEntity)

Specified by:

getMetadata in interface MetadataReader

addProperties

public void addProperties(MetadataEntity metadataEntity,
                          Map<String,String> properties)

Specified by:

addProperties in interface MetadataWriter

addTags

public void addTags(MetadataEntity metadataEntity,
                    String... tags)

Specified by:

addTags in interface MetadataWriter

addTags

public void addTags(MetadataEntity metadataEntity,
                    Iterable<String> tags)

Specified by:

addTags in interface MetadataWriter

removeMetadata

public void removeMetadata(MetadataEntity metadataEntity)

Specified by:

removeMetadata in interface MetadataWriter

removeProperties

public void removeProperties(MetadataEntity metadataEntity)

Specified by:

removeProperties in interface MetadataWriter

removeProperties

public void removeProperties(MetadataEntity metadataEntity,
                             String... keys)

Specified by:

removeProperties in interface MetadataWriter

removeTags

public void removeTags(MetadataEntity metadataEntity)

Specified by:

removeTags in interface MetadataWriter

removeTags

public void removeTags(MetadataEntity metadataEntity,
                       String... tags)

Specified by:

removeTags in interface MetadataWriter

record

public void record(List<FieldOperation> fieldOperations)

Specified by:

record in interface LineageRecorder