How to use

InMemoryPipelineExecutorRegistry

in

org.guvnor.ala.registry.inmemory

@Test
public void testDeregister() {
  pipelineExecutorRegistry.register(trace);
  PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertEquals(trace,
         result);
  pipelineExecutorRegistry.deregister(trace.getTaskId());
  result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertNull(result);
}

@Test
public void testRegister() {
  pipelineExecutorRegistry.register(trace);
  PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertEquals(trace,
         result);
}

private void initializeRegistry() {
  try {
    final List<Object> traces = registryHelper.readEntries(registryRoot,
                                newFilter(TRACE_SUFFIX));
    traces.forEach(trace -> super.register((PipelineExecutorTrace) trace));
  } catch (Exception e) {
    logger.error("An error was produced during " + VFSPipelineExecutorRegistry.class.getName() + " initialization.",
           e);
  }
}

@Test
public void testGetExecutorTraces() {
  List<PipelineExecutorTrace> traces = new ArrayList<>();
  for (int i = 0; i < TRACES_COUNT; i++) {
    PipelineExecutorTrace trace = mock(PipelineExecutorTrace.class);
    when(trace.getTaskId()).thenReturn(PIPELINE_EXECUTION_ID + Integer.toString(i));
    traces.add(trace);
  }
  traces.forEach(trace -> pipelineExecutorRegistry.register(trace));
  Collection<PipelineExecutorTrace> result = pipelineExecutorRegistry.getExecutorTraces();
  assertEquals(traces.size(),
         result.size());
  for (PipelineExecutorTrace trace : traces) {
    assertTrue(result.contains(trace));
  }
}

@Override
public void deregister(final String taskId) {
  checkNotNull("taskId",
         taskId);
  final Path path = buildTracePath(taskId);
  registryHelper.deleteBatch(path);
  super.deregister(taskId);
}

@Test
public void testInit() throws Exception {
  traces = new ArrayList<>();
  for (int i = 0; i < TRACES_COUNT; i++) {
    PipelineExecutorTrace trace = mock(PipelineExecutorTrace.class);
    when(trace.getTaskId()).thenReturn(PIPELINE_EXECUTION_ID + Integer.toString(i));
    traces.add(trace);
  }
  when(registryHelper.readEntries(registryRoot,
                  VFSRegistryHelper.BySuffixFilter.newFilter(TRACE_SUFFIX))).thenReturn(traces);
  ((VFSPipelineExecutorRegistry) pipelineExecutorRegistry).init();
  verify(registryHelper,
      times(2)).ensureDirectory(PIPELINE_EXECUTOR_REGISTRY_PATH);
  verify(registryHelper,
      times(2)).readEntries(registryRoot,
                 VFSRegistryHelper.BySuffixFilter.newFilter(TRACE_SUFFIX));
  for (Object trace : traces) {
    PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(((PipelineExecutorTrace) trace).getTaskId());
    assertNotNull(result);
    assertEquals(trace,
           result);
  }
}

@Before
public void setUp() {
  trace = mock(PipelineExecutorTrace.class);
  when(trace.getTaskId()).thenReturn(PIPELINE_EXECUTION_ID);
  pipelineExecutorRegistry = new InMemoryPipelineExecutorRegistry();
}

@Test
public void testGetExecutorTrace() {
  pipelineExecutorRegistry.register(trace);
  PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertEquals(trace,
         result);
}

@Override
public void register(final PipelineExecutorTrace trace) {
  checkNotNull("trace",
         trace);
  final Path path = buildTracePath(trace.getTaskId());
  try {
    registryHelper.storeEntry(path,
                 trace);
  } catch (Exception e) {
    //uncommon error
    logger.error("Unexpected error was produced during trace marshalling/storing, trace: " + trace,
           e);
    throw new RuntimeException("Unexpected error was produced during trace marshalling/storing, trace: " + trace,
                  e);
  }
  super.register(trace);
}

@Test
public void testGetExecutorTraces() {
  List<PipelineExecutorTrace> traces = new ArrayList<>();
  for (int i = 0; i < TRACES_COUNT; i++) {
    PipelineExecutorTrace trace = mock(PipelineExecutorTrace.class);
    when(trace.getTaskId()).thenReturn(PIPELINE_EXECUTION_ID + Integer.toString(i));
    traces.add(trace);
  }
  traces.forEach(trace -> pipelineExecutorRegistry.register(trace));
  Collection<PipelineExecutorTrace> result = pipelineExecutorRegistry.getExecutorTraces();
  assertEquals(traces.size(),
         result.size());
  for (PipelineExecutorTrace trace : traces) {
    assertTrue(result.contains(trace));
  }
}

@Test
public void testInit() throws Exception {
  traces = new ArrayList<>();
  for (int i = 0; i < TRACES_COUNT; i++) {
    PipelineExecutorTrace trace = mock(PipelineExecutorTrace.class);
    when(trace.getTaskId()).thenReturn(PIPELINE_EXECUTION_ID + Integer.toString(i));
    traces.add(trace);
  }
  when(registryHelper.readEntries(registryRoot,
                  VFSRegistryHelper.BySuffixFilter.newFilter(TRACE_SUFFIX))).thenReturn(traces);
  ((VFSPipelineExecutorRegistry) pipelineExecutorRegistry).init();
  verify(registryHelper,
      times(2)).ensureDirectory(PIPELINE_EXECUTOR_REGISTRY_PATH);
  verify(registryHelper,
      times(2)).readEntries(registryRoot,
                 VFSRegistryHelper.BySuffixFilter.newFilter(TRACE_SUFFIX));
  for (Object trace : traces) {
    PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(((PipelineExecutorTrace) trace).getTaskId());
    assertNotNull(result);
    assertEquals(trace,
           result);
  }
}

@Before
public void setUp() {
  trace = mock(PipelineExecutorTrace.class);
  when(trace.getTaskId()).thenReturn(PIPELINE_EXECUTION_ID);
  pipelineExecutorRegistry = new InMemoryPipelineExecutorRegistry();
}

@Test
public void testDeregister() {
  pipelineExecutorRegistry.register(trace);
  PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertEquals(trace,
         result);
  pipelineExecutorRegistry.deregister(trace.getTaskId());
  result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertNull(result);
}

@Test
public void testGetExecutorTrace() {
  pipelineExecutorRegistry.register(trace);
  PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertEquals(trace,
         result);
}

@Test
public void testRegisterWhenMarshallingErrors() throws Exception {
  prepareTargetPath();
  expectedException.expectMessage("Unexpected error was produced during trace marshalling/storing, trace: " + trace);
  doThrow(new Exception("no matter the message here"))
      .when(registryHelper)
      .storeEntry(traceTargetPath,
            trace);
  pipelineExecutorRegistry.register(trace);
}

@Test
@Override
public void testDeregister() {
  prepareTargetPath();
  pipelineExecutorRegistry.register(trace);
  PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertEquals(trace,
         result);
  pipelineExecutorRegistry.deregister(PIPELINE_EXECUTION_ID);
  verify(registryHelper,
      times(1)).deleteBatch(traceTargetPath);
  result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertNull(result);
}

@Test
public void testRegister() {
  pipelineExecutorRegistry.register(trace);
  PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertEquals(trace,
         result);
}

@Test
public void testRegisterWhenMarshallingErrors() throws Exception {
  prepareTargetPath();
  expectedException.expectMessage("Unexpected error was produced during trace marshalling/storing, trace: " + trace);
  doThrow(new Exception("no matter the message here"))
      .when(registryHelper)
      .storeEntry(traceTargetPath,
            trace);
  pipelineExecutorRegistry.register(trace);
}

@Test
@Override
public void testDeregister() {
  prepareTargetPath();
  pipelineExecutorRegistry.register(trace);
  PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertEquals(trace,
         result);
  pipelineExecutorRegistry.deregister(PIPELINE_EXECUTION_ID);
  verify(registryHelper,
      times(1)).deleteBatch(traceTargetPath);
  result = pipelineExecutorRegistry.getExecutorTrace(PIPELINE_EXECUTION_ID);
  assertNull(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);
}