@Override public PipelineExecutorTrace getExecutorTrace(final RuntimeId runtimeId) { checkNotNull("runtimeId", runtimeId); return recordsMap.values() .stream() .filter(trace -> (trace.getTask().getOutput() instanceof RuntimeId) && runtimeId.getId().equals(((RuntimeId) trace.getTask().getOutput()).getId()) ).findFirst().orElse(null); } }
@Override public PipelineExecutorTrace getExecutorTrace(final RuntimeId runtimeId) { checkNotNull("runtimeId", runtimeId); return recordsMap.values() .stream() .filter(trace -> (trace.getTask().getOutput() instanceof RuntimeId) && runtimeId.getId().equals(((RuntimeId) trace.getTask().getOutput()).getId()) ).findFirst().orElse(null); } }
@Override public boolean test(PipelineExecutorTrace pipelineExecutorTrace) { if (pipelineExecutionId != null) { return pipelineExecutionId.equals(pipelineExecutorTrace.getTaskId()); } if (runtimeId != null) { return (pipelineExecutorTrace.getTask().getOutput() instanceof RuntimeId) && runtimeId.equals(((RuntimeId) pipelineExecutorTrace.getTask().getOutput()).getId()); } if (runtimeName != null) { return (pipelineExecutorTrace.getTask().getOutput() instanceof RuntimeId) && runtimeName.equals(((RuntimeId) pipelineExecutorTrace.getTask().getOutput()).getName()); } if (providerId != null) { if (pipelineExecutorTrace.getTask().getTaskDef().getProviderId() == null) { return false; } else if (!providerId.equals(pipelineExecutorTrace.getTask().getTaskDef().getProviderId().getId())) { return false; } } if (pipelineId != null) { if (!pipelineId.equals(pipelineExecutorTrace.getPipelineId())) { return false; } } return true; } }
@Test public void getExecutorTraceByRuntimeId() { RuntimeIdMock runtimeId = mock(RuntimeIdMock.class); when(runtimeId.getId()).thenReturn(RUNTIME_ID); PipelineExecutorTask task = mock(PipelineExecutorTask.class); when(task.getOutput()).thenReturn(runtimeId); when(trace.getTask()).thenReturn(task); pipelineExecutorRegistry.register(trace); PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(runtimeId); assertEquals(trace, result); }
@Test public void getExecutorTraceByRuntimeId() { RuntimeIdMock runtimeId = mock(RuntimeIdMock.class); when(runtimeId.getId()).thenReturn(RUNTIME_ID); PipelineExecutorTask task = mock(PipelineExecutorTask.class); when(task.getOutput()).thenReturn(runtimeId); when(trace.getTask()).thenReturn(task); pipelineExecutorRegistry.register(trace); PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(runtimeId); assertEquals(trace, result); }
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)); }); }
item.setPipelineStageItems(new PipelineStageItemList(stageItems)); if (pipelineExecutorTrace.getTask().getOutput() instanceof Runtime) { Runtime runtime = runtimeRegistry.getRuntimeById(((Runtime) pipelineExecutorTrace.getTask().getOutput()).getId()); if (runtime != null) { item.setRuntimeId(runtime.getId());
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()); }