public void registerTask(
AbstractTask task,
@SuppressWarnings("rawtypes") Class<? extends PactDriver> driver,
Class<? extends Stub> stubClass) {
final TaskConfig config = new TaskConfig(this.mockEnv.getTaskConfiguration());
config.setDriver(driver);
config.setStubWrapper(new UserCodeClassWrapper<Stub>(stubClass));
task.setEnvironment(this.mockEnv);
if (task instanceof RegularPactTask<?, ?>) {
((RegularPactTask<?, ?>) task).setUserCodeClassLoader(getClass().getClassLoader());
}
task.registerInputOutput();
}
public void registerTask(AbstractTask task) {
task.setEnvironment(this.mockEnv);
task.registerInputOutput();
}