public void registerFileInputTask(
AbstractInputTask<?> inTask,
Class<? extends DelimitedInputFormat> stubClass,
String inPath,
String delimiter) {
DelimitedInputFormat format;
try {
format = stubClass.newInstance();
} catch (Throwable t) {
throw new RuntimeException("Could not instantiate test input format.", t);
}
format.setFilePath(inPath);
format.setDelimiter(delimiter);
TaskConfig dsConfig = new TaskConfig(this.mockEnv.getTaskConfiguration());
dsConfig.setStubWrapper(new UserCodeObjectWrapper<DelimitedInputFormat>(format));
this.inputSplitProvider.addInputSplits(inPath, 5);
inTask.setEnvironment(this.mockEnv);
if (inTask instanceof DataSourceTask<?>) {
((DataSourceTask<?>) inTask).setUserCodeClassLoader(getClass().getClassLoader());
}
inTask.registerInputOutput();
}
