How to use

BootstrapResult

in

co.cask.cdap.proto.bootstrap

/**
 * Execute the steps in the loaded bootstrap config.
 *
 * @param shouldSkip predicate that determines whether to skip a step
 * @return the result of executing the bootstrap steps.
 * @throws IllegalStateException if bootstrapping is already in progress
 * @throws InterruptedException if bootstrapping was interrupted
 */
public BootstrapResult bootstrap(Predicate<BootstrapStep> shouldSkip) throws InterruptedException {
 List<BootstrapStepResult> results = new ArrayList<>(config.getSteps().size());
 if (!bootstrapping.compareAndSet(false, true)) {
  throw new IllegalStateException("Bootstrap already in progress.");
 }
 try {
  for (BootstrapStep step : config.getSteps()) {
   results.add(executeStep(step, shouldSkip));
  }
  if (!config.getSteps().isEmpty()) {
   bootstrapStore.bootstrapped();
  }
  return new BootstrapResult(results);
 } finally {
  bootstrapping.set(false);
 }
}

/**
 * Execute the steps in the loaded bootstrap config.
 *
 * @param shouldSkip predicate that determines whether to skip a step
 * @return the result of executing the bootstrap steps.
 * @throws IllegalStateException if bootstrapping is already in progress
 * @throws InterruptedException if bootstrapping was interrupted
 */
public BootstrapResult bootstrap(Predicate<BootstrapStep> shouldSkip) throws InterruptedException {
 List<BootstrapStepResult> results = new ArrayList<>(config.getSteps().size());
 if (!bootstrapping.compareAndSet(false, true)) {
  throw new IllegalStateException("Bootstrap already in progress.");
 }
 try {
  for (BootstrapStep step : config.getSteps()) {
   results.add(executeStep(step, shouldSkip));
  }
  if (!config.getSteps().isEmpty()) {
   bootstrapStore.bootstrapped();
  }
  return new BootstrapResult(results);
 } finally {
  bootstrapping.set(false);
 }
}

@Test
public void testRetries() throws Exception {
 EXECUTOR1.setNumRetryableFailures(3);
 BootstrapResult result = bootstrapService.bootstrap();
 List<BootstrapStepResult> expectedStepResults = new ArrayList<>();
 for (BootstrapStep step : bootstrapConfig.getSteps()) {
  expectedStepResults.add(new BootstrapStepResult(step.getLabel(), BootstrapStepResult.Status.SUCCEEDED));
 }
 BootstrapResult expected = new BootstrapResult(expectedStepResults);
 Assert.assertEquals(expected, result);
}

@Test
public void testRunCondition() throws Exception {
 BootstrapResult result =
  bootstrapService.bootstrap(step -> step.getRunCondition() == BootstrapStep.RunCondition.ONCE);
 List<BootstrapStepResult> stepResults = new ArrayList<>(3);
 stepResults.add(new BootstrapStepResult(STEP1.getLabel(), BootstrapStepResult.Status.SKIPPED));
 stepResults.add(new BootstrapStepResult(STEP2.getLabel(), BootstrapStepResult.Status.SUCCEEDED));
 stepResults.add(new BootstrapStepResult(STEP3.getLabel(), BootstrapStepResult.Status.SKIPPED));
 Assert.assertEquals(new BootstrapResult(stepResults), result);
}

@Test
public void testContinuesAfterFailures() throws Exception {
 EXECUTOR1.shouldFail();
 BootstrapResult result = bootstrapService.bootstrap();
 List<BootstrapStepResult> stepResults = new ArrayList<>(3);
 stepResults.add(new BootstrapStepResult(STEP1.getLabel(), BootstrapStepResult.Status.FAILED));
 stepResults.add(new BootstrapStepResult(STEP2.getLabel(), BootstrapStepResult.Status.SUCCEEDED));
 stepResults.add(new BootstrapStepResult(STEP3.getLabel(), BootstrapStepResult.Status.SUCCEEDED));
 BootstrapResult expected = new BootstrapResult(stepResults);
 Assert.assertEquals(expected, result);
}

@Test
public void testAllSuccess() throws Exception  {
 BootstrapResult result = bootstrapService.bootstrap();
 List<BootstrapStepResult> expectedStepResults = new ArrayList<>();
 for (BootstrapStep step : bootstrapConfig.getSteps()) {
  expectedStepResults.add(new BootstrapStepResult(step.getLabel(), BootstrapStepResult.Status.SUCCEEDED));
 }
 BootstrapResult expected = new BootstrapResult(expectedStepResults);
 Assert.assertEquals(expected, result);
 Assert.assertTrue(bootstrapService.isBootstrapped());
}