How to use

getStages

method

in

org.guvnor.ala.pipeline.execution.PipelineExecutorTaskDef

public PipelineExecutorTaskImpl(final PipelineExecutorTaskDef taskDef,
                final String executionId) {
  this.taskDef = taskDef;
  setId(executionId);
  taskDef.getStages().forEach(stage -> setStageStatus(stage,
                            Status.SCHEDULED));
}

public PipelineExecutorTaskImpl(final PipelineExecutorTaskDef taskDef,
                final String executionId) {
  this.taskDef = taskDef;
  setId(executionId);
  taskDef.getStages().forEach(stage -> setStageStatus(stage,
                            Status.SCHEDULED));
}

  public void setTaskInStoppedStatus(final PipelineExecutorTaskImpl task) {
    task.setPipelineStatus(PipelineExecutorTask.Status.STOPPED);
    task.getTaskDef().getStages()
        .stream()
        .filter(stage -> PipelineExecutorTask.Status.RUNNING.equals(task.getStageStatus(stage)) ||
            PipelineExecutorTask.Status.SCHEDULED.equals(task.getStageStatus(stage)))
        .forEach(stage -> task.setStageStatus(stage,
                           PipelineExecutorTask.Status.STOPPED));
    task.clearErrors();
    task.setOutput(null);
  }
}

  public void setTaskInStoppedStatus(final PipelineExecutorTaskImpl task) {
    task.setPipelineStatus(PipelineExecutorTask.Status.STOPPED);
    task.getTaskDef().getStages()
        .stream()
        .filter(stage -> PipelineExecutorTask.Status.RUNNING.equals(task.getStageStatus(stage)) ||
            PipelineExecutorTask.Status.SCHEDULED.equals(task.getStageStatus(stage)))
        .forEach(stage -> task.setStageStatus(stage,
                           PipelineExecutorTask.Status.STOPPED));
    task.clearErrors();
    task.setOutput(null);
  }
}

Input input = mock(Input.class);
when(taskDef.getInput()).thenReturn(input);
when(taskDef.getStages()).thenReturn(stages);

Input input = mock(Input.class);
when(taskDef.getInput()).thenReturn(input);
when(taskDef.getStages()).thenReturn(stages);

@Override
public PipelineExecutorTraceImpl getValue() {
  //only the stage names are used by the taskDef, so the pipeline can be mocked.
  List<Stage> stages = mockStages(STAGE_COUNT);
  Pipeline pipeline = new BasePipeline(PIPELINE_NAME,
                     stages) {
  };
  Input input = mockInput(PIPELINE_INPUT_SIZE);
  PipelineExecutorTaskDefImpl taskDef = new PipelineExecutorTaskDefImpl(pipeline,
                                     input);
  PipelineExecutorTaskImpl taskImpl = new PipelineExecutorTaskImpl(taskDef,
                                   PIPELINE_EXECUTION_ID);
  taskImpl.setPipelineStatus(PipelineExecutorTask.Status.SCHEDULED);
  taskImpl.setPipelineError(mockError(PIPELINE_ERROR,
                    PIPELINE_ERROR_DETAIL));
  taskImpl.getTaskDef().getStages().forEach(stage -> taskImpl.setStageError(stage,
                                       mockStageError(stage)));
  taskImpl.setOutput(new MockPipelineOutput(PIPELINE_OUTPUT));
  return new PipelineExecutorTraceImpl(taskImpl);
}

@Override
public PipelineExecutorTraceImpl getValue() {
  //only the stage names are used by the taskDef, so the pipeline can be mocked.
  List<Stage> stages = mockStages(STAGE_COUNT);
  Pipeline pipeline = new BasePipeline(PIPELINE_NAME,
                     stages);
  Input input = mockInput(PIPELINE_INPUT_SIZE);
  PipelineExecutorTaskDefImpl taskDef = new PipelineExecutorTaskDefImpl(pipeline,
                                     input);
  PipelineExecutorTaskImpl taskImpl = new PipelineExecutorTaskImpl(taskDef,
                                   PIPELINE_EXECUTION_ID);
  taskImpl.setPipelineStatus(PipelineExecutorTask.Status.SCHEDULED);
  taskImpl.setPipelineError(mockError(PIPELINE_ERROR));
  taskImpl.getTaskDef().getStages().forEach(stage -> taskImpl.setStageError(stage,
                                       mockStageError(stage)));
  taskImpl.setOutput(new MockPipelineOutput(PIPELINE_OUTPUT));
  return new PipelineExecutorTraceImpl(taskImpl);
}

protected void assertHasSameInfo(PipelineExecutorTask expectedTask,
                 PipelineExecutorTask task) {
  assertEquals(expectedTask.getId(),
         task.getId());
  assertEquals(expectedTask.getPipelineStatus(),
         task.getPipelineStatus());
  assertEquals(expectedTask.getPipelineError(),
         task.getPipelineError());
  assertEquals(expectedTask.getOutput(),
         task.getOutput());
  assertHasSameInfo(expectedTask.getTaskDef(),
           task.getTaskDef());
  expectedTask.getTaskDef().getStages().forEach(stage -> {
    assertEquals(expectedTask.getStageStatus(stage),
           task.getStageStatus(stage));
    assertEquals(expectedTask.getStageError(stage),
           task.getStageError(stage));
  });
}

protected void assertHasSameInfo(PipelineExecutorTask expectedTask,
                 PipelineExecutorTask task) {
  assertEquals(expectedTask.getId(),
         task.getId());
  assertEquals(expectedTask.getPipelineStatus(),
         task.getPipelineStatus());
  assertEquals(expectedTask.getPipelineError(),
         task.getPipelineError());
  assertEquals(expectedTask.getOutput(),
         task.getOutput());
  assertHasSameInfo(expectedTask.getTaskDef(),
           task.getTaskDef());
  expectedTask.getTaskDef().getStages().forEach(stage -> {
    assertEquals(expectedTask.getStageStatus(stage),
           task.getStageStatus(stage));
    assertEquals(expectedTask.getStageError(stage),
           task.getStageError(stage));
  });
}

List<PipelineStageItem> stageItems = pipelineExecutorTrace.getTask().getTaskDef().getStages().stream()
    .map(stage -> {
      String stageError = null;

private void assertEqualsPipelineExecutorTrace(PipelineExecutorTraceImpl expectedValue,
                        PipelineExecutorTraceImpl value) {
  assertEquals(expectedValue.getTaskId(),
         value.getTaskId());
  assertEquals(expectedValue.getPipelineId(),
         value.getPipelineId());
  assertEquals(expectedValue.getTask().getTaskDef(),
         value.getTask().getTaskDef());
  assertEquals(expectedValue.getTask().getPipelineStatus(),
         value.getTask().getPipelineStatus());
  assertEquals(expectedValue.getTask().getPipelineError(),
         value.getTask().getPipelineError());
  for (String stage : expectedValue.getTask().getTaskDef().getStages()) {
    assertEquals(expectedValue.getTask().getStageStatus(stage),
           value.getTask().getStageStatus(stage));
    assertEquals(expectedValue.getTask().getStageError(stage),
           value.getTask().getStageError(stage));
  }
  assertEquals(expectedValue.getTask().getOutput(),
         value.getTask().getOutput());
}

private void assertEqualsPipelineExecutorTrace(PipelineExecutorTraceImpl expectedValue,
                        PipelineExecutorTraceImpl value) {
  assertEquals(expectedValue.getTaskId(),
         value.getTaskId());
  assertEquals(expectedValue.getPipelineId(),
         value.getPipelineId());
  assertEquals(expectedValue.getTask().getTaskDef(),
         value.getTask().getTaskDef());
  assertEquals(expectedValue.getTask().getPipelineStatus(),
         value.getTask().getPipelineStatus());
  assertEqualsPipelineExecutorException(expectedValue.getTask().getPipelineError(),
                     value.getTask().getPipelineError());
  for (String stage : expectedValue.getTask().getTaskDef().getStages()) {
    assertEquals(expectedValue.getTask().getStageStatus(stage),
           value.getTask().getStageStatus(stage));
    assertEqualsPipelineExecutorException(expectedValue.getTask().getStageError(stage),
                       value.getTask().getStageError(stage));
  }
  assertEquals(expectedValue.getTask().getOutput(),
         value.getTask().getOutput());
}