@Before public void setUp() { clearDefaultProperties(); configExecutors = mockConfigExecutors(CONFIG_EXECUTORS_SIZE); when(configExecutorsInstance.iterator()).thenReturn(configExecutors.iterator()); externalListeners = mockEventListeners(PIPELINE_EVENT_LISTENERS); when(eventListenersInstance.iterator()).thenReturn(externalListeners.iterator()); taskManagerHelper = spy(new PipelineExecutorTaskManagerImplHelper(configExecutorsInstance, eventListenersInstance)); }
@Before public void setUp() { super.setUp(); }
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)); }); }
@Before public void setUp() { clearDefaultProperties(); configExecutors = mockConfigExecutors(CONFIG_EXECUTORS_SIZE); when(configExecutorsInstance.iterator()).thenReturn(configExecutors.iterator()); externalListeners = mockEventListeners(PIPELINE_EVENT_LISTENERS); when(eventListenersInstance.iterator()).thenReturn(externalListeners.iterator()); taskManagerHelper = spy(new PipelineExecutorTaskManagerImplHelper(configExecutorsInstance, eventListenersInstance)); }
@Before public void setUp() { super.setUp(); }
@Before public void setUp() { pipelineExecutorTraceCaptor = ArgumentCaptor.forClass(PipelineExecutorTrace.class); taskEntryCaptor = ArgumentCaptor.forClass(PipelineExecutorTaskManagerImpl.TaskEntry.class); stringCaptor = ArgumentCaptor.forClass(String.class); configExecutors = mockConfigExecutors(CONFIG_EXECUTORS_SIZE); when(configExecutorsInstance.iterator()).thenReturn(configExecutors.iterator()); externalListeners = mockEventListeners(PIPELINE_EVENT_LISTENERS); when(eventListenersInstance.iterator()).thenReturn(externalListeners.iterator()); taskManagerHelper = spy(new PipelineExecutorTaskManagerImplHelper(configExecutorsInstance, eventListenersInstance)); doReturn(executorService).when(taskManagerHelper).createExecutorService(); doReturn(pipelineExecutor).when(taskManagerHelper).createPipelineExecutor(); taskManager = spy(new PipelineExecutorTaskManagerImpl(pipelineRegistry, configExecutorsInstance, eventListenersInstance, pipelineExecutorRegistry) { { super.taskManagerHelper = PipelineExecutorTaskManagerImplTestBase.this.taskManagerHelper; } @Override protected void init() { super.init(); super.futureTaskMap = spy(super.futureTaskMap); } }); }
@Before public void setUp() { super.setUp(); }
@Before public void setUp() { pipelineExecutorTraceCaptor = ArgumentCaptor.forClass(PipelineExecutorTrace.class); taskEntryCaptor = ArgumentCaptor.forClass(PipelineExecutorTaskManagerImpl.TaskEntry.class); stringCaptor = ArgumentCaptor.forClass(String.class); configExecutors = mockConfigExecutors(CONFIG_EXECUTORS_SIZE); when(configExecutorsInstance.iterator()).thenReturn(configExecutors.iterator()); externalListeners = mockEventListeners(PIPELINE_EVENT_LISTENERS); when(eventListenersInstance.iterator()).thenReturn(externalListeners.iterator()); taskManagerHelper = spy(new PipelineExecutorTaskManagerImplHelper(configExecutorsInstance, eventListenersInstance)); doReturn(executorService).when(taskManagerHelper).createExecutorService(); doReturn(pipelineExecutor).when(taskManagerHelper).createPipelineExecutor(); taskManager = spy(new PipelineExecutorTaskManagerImpl(pipelineRegistry, configExecutorsInstance, eventListenersInstance, pipelineExecutorRegistry) { { super.taskManagerHelper = PipelineExecutorTaskManagerImplTestBase.this.taskManagerHelper; } @Override protected void init() { super.init(); super.futureTaskMap = spy(super.futureTaskMap); } }); }
@Before public void setUp() { super.setUp(); }
@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(); }
@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(); }