How to use

getPipeline

method

in

org.guvnor.ala.pipeline.execution.PipelineExecutorTaskDef

@JsonIgnore
@Override
public String getPipelineId() {
  return getTask().getTaskDef().getPipeline();
}

@JsonIgnore
@Override
public String getPipelineId() {
  return getTask().getTaskDef().getPipeline();
}

private void prepareExecution() {
  //mock the execution inputs
  pipeline = mock(Pipeline.class);
  stages = mockStages(PIPELINE_STAGES_SIZE);
  when(pipeline.getStages()).thenReturn(stages);
  when(pipeline.getName()).thenReturn(PIPELINE_ID);
  when(pipelineRegistry.getPipelineByName(PIPELINE_ID)).thenReturn(pipeline);
  taskDef = mock(PipelineExecutorTaskDef.class);
  input = mock(Input.class);
  when(taskDef.getInput()).thenReturn(input);
  when(taskDef.getPipeline()).thenReturn(PIPELINE_ID);
}

private void prepareExecution() {
  //mock the execution inputs
  pipeline = mock(Pipeline.class);
  stages = mockStages(PIPELINE_STAGES_SIZE);
  when(pipeline.getStages()).thenReturn(stages);
  when(pipeline.getName()).thenReturn(PIPELINE_ID);
  when(pipelineRegistry.getPipelineByName(PIPELINE_ID)).thenReturn(pipeline);
  taskDef = mock(PipelineExecutorTaskDef.class);
  input = mock(Input.class);
  when(taskDef.getInput()).thenReturn(input);
  when(taskDef.getPipeline()).thenReturn(PIPELINE_ID);
}

@Before
public void setUp() {
  super.setUp();
  error = mock(Throwable.class);
  when(error.getMessage()).thenReturn(ERROR_MESSAGE);
  pipelineExecutorExceptionCaptor = ArgumentCaptor.forClass(PipelineExecutorException.class);
  pipeline = mock(Pipeline.class);
  stages = mockStages(PIPELINE_STAGES_SIZE);
  when(pipeline.getStages()).thenReturn(stages);
  when(pipeline.getName()).thenReturn(PIPELINE_ID);
  when(pipelineRegistry.getPipelineByName(PIPELINE_ID)).thenReturn(pipeline);
  //pick an arbitrary stage for events testing.
  stage = stages.get(0);
  taskDef = mock(PipelineExecutorTaskDef.class);
  input = mock(Input.class);
  when(taskDef.getInput()).thenReturn(input);
  when(taskDef.getPipeline()).thenReturn(PIPELINE_ID);
  task = spy(taskManagerHelper.createTask(taskDef,
                      TASK_ID));
  taskEntry = mock(PipelineExecutorTaskManagerImpl.TaskEntry.class);
  when(taskEntry.getTask()).thenReturn(task);
  //emulate there's a running task.
  taskManager.currentTasks.put(TASK_ID,
                 taskEntry);
  taskManager.init();
}

@Before
public void setUp() {
  super.setUp();
  error = mock(Throwable.class);
  when(error.getMessage()).thenReturn(ERROR_MESSAGE);
  pipelineExecutorErrorCaptor = ArgumentCaptor.forClass(PipelineExecutorError.class);
  pipeline = mock(Pipeline.class);
  stages = mockStages(PIPELINE_STAGES_SIZE);
  when(pipeline.getStages()).thenReturn(stages);
  when(pipeline.getName()).thenReturn(PIPELINE_ID);
  when(pipelineRegistry.getPipelineByName(PIPELINE_ID)).thenReturn(pipeline);
  //pick an arbitrary stage for events testing.
  stage = stages.get(0);
  taskDef = mock(PipelineExecutorTaskDef.class);
  input = mock(Input.class);
  when(taskDef.getInput()).thenReturn(input);
  when(taskDef.getPipeline()).thenReturn(PIPELINE_ID);
  task = spy(taskManagerHelper.createTask(taskDef,
                      TASK_ID));
  taskEntry = mock(PipelineExecutorTaskManagerImpl.TaskEntry.class);
  when(taskEntry.getTask()).thenReturn(task);
  //emulate there's a running task.
  taskManager.currentTasks.put(TASK_ID,
                 taskEntry);
  taskManager.init();
}

Pipeline pipeline = mock(Pipeline.class);
when(pipeline.getStages()).thenReturn(mock(List.class));
when(taskDef.getPipeline()).thenReturn(PIPELINE_ID);

Pipeline pipeline = mock(Pipeline.class);
when(pipeline.getStages()).thenReturn(mock(List.class));
when(taskDef.getPipeline()).thenReturn(PIPELINE_ID);

/**
 * Executes a task in asynchronous mode.
 * @param task the task for execute.
 * @return the taskId of the task.
 */
private synchronized void startAsyncTask(final PipelineExecutorTask task) {
  final Future<?> future = executor.submit(() -> {
    final Pipeline pipeline = pipelineRegistry.getPipelineByName(task.getTaskDef().getPipeline());
    try {
      pipelineExecutor.execute(task.getTaskDef().getInput(),
                   pipeline,
                   output -> processPipelineOutput(task,
                                   output),
                   localListener);
    } catch (Exception e) {
      logger.error("An error was produced during pipeline execution for PipelineExecutorTask: " + task.getId(),
             e);
    } finally {
      removeTaskEntry(task.getId());
      removeFutureTask(task.getId());
    }
  });
  storeFutureTask(task.getId(),
          future);
}

/**
 * Executes a task in asynchronous mode.
 * @param task the task for execute.
 * @return the taskId of the task.
 */
private synchronized void startAsyncTask(final PipelineExecutorTask task) {
  final Future<?> future = executor.submit(() -> {
    final Pipeline pipeline = pipelineRegistry.getPipelineByName(task.getTaskDef().getPipeline());
    try {
      pipelineExecutor.execute(task.getTaskDef().getInput(),
                   pipeline,
                   output -> processPipelineOutput(task,
                                   output),
                   localListener);
    } catch (Exception e) {
      logger.error("An error was produced during pipeline execution for PipelineExecutorTask: " + task.getId(),
             e);
    } finally {
      removeTaskEntry(task.getId());
      removeFutureTask(task.getId());
    }
  });
  storeFutureTask(task.getId(),
          future);
}

  protected void assertHasSameInfo(PipelineExecutorTaskDef expectedTaskDef,
                   PipelineExecutorTaskDef taskDef) {
    assertEquals(expectedTaskDef.getInput(),
           taskDef.getInput());
    assertEquals(expectedTaskDef.getPipeline(),
           taskDef.getPipeline());
    assertEquals(expectedTaskDef.getProviderId(),
           taskDef.getProviderId());
    assertEquals(expectedTaskDef.getProviderType(),
           taskDef.getProviderType());
  }
}

  protected void assertHasSameInfo(PipelineExecutorTaskDef expectedTaskDef,
                   PipelineExecutorTaskDef taskDef) {
    assertEquals(expectedTaskDef.getInput(),
           taskDef.getInput());
    assertEquals(expectedTaskDef.getPipeline(),
           taskDef.getPipeline());
    assertEquals(expectedTaskDef.getProviderId(),
           taskDef.getProviderId());
    assertEquals(expectedTaskDef.getProviderType(),
           taskDef.getProviderType());
  }
}

/**
 * Executes a task definition in synchronous mode.
 * @param taskDef task definition for executing.
 * @return the taskId assigned to the executed task.
 */
private String executeSync(final PipelineExecutorTaskDef taskDef) {
  final PipelineExecutorTaskImpl task = taskManagerHelper.createTask(taskDef);
  storeTaskEntry(TaskEntry.newSyncEntry(task));
  final Pipeline pipeline = pipelineRegistry.getPipelineByName(taskDef.getPipeline());
  pipelineExecutor.execute(taskDef.getInput(),
               pipeline,
               output -> processPipelineOutput(task,
                               output),
               localListener);
  removeTaskEntry(task.getId());
  updateExecutorRegistry(task);
  return task.getId();
}

/**
 * Executes a task definition in synchronous mode.
 * @param taskDef task definition for executing.
 * @return the taskId assigned to the executed task.
 */
private String executeSync(final PipelineExecutorTaskDef taskDef) {
  final PipelineExecutorTaskImpl task = taskManagerHelper.createTask(taskDef);
  storeTaskEntry(TaskEntry.newSyncEntry(task));
  final Pipeline pipeline = pipelineRegistry.getPipelineByName(taskDef.getPipeline());
  pipelineExecutor.execute(taskDef.getInput(),
               pipeline,
               output -> processPipelineOutput(task,
                               output),
               localListener);
  removeTaskEntry(task.getId());
  updateExecutorRegistry(task);
  return task.getId();
}