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()); }