How to use

testCreateAppRunning

method

in

org.apache.hadoop.yarn.server.resourcemanager.rmapp.TestRMAppTransitions

protected RMApp testCreateAppFinalSaving(
  ApplicationSubmissionContext submissionContext) throws IOException {
 RMApp application = testCreateAppRunning(submissionContext);
 RMAppEvent finishingEvent =
   new RMAppEvent(application.getApplicationId(),
    RMAppEventType.ATTEMPT_UNREGISTERED);
 application.handle(finishingEvent);
 assertAppState(RMAppState.FINAL_SAVING, application);
 assertAppFinalStateSaved(application);
 return application;
}

protected RMApp testCreateAppFinalSaving(
  ApplicationSubmissionContext submissionContext) throws IOException {
 RMApp application = testCreateAppRunning(submissionContext);
 RMAppEvent finishingEvent =
   new RMAppEvent(application.getApplicationId(),
    RMAppEventType.ATTEMPT_UNREGISTERED);
 application.handle(finishingEvent);
 assertAppState(RMAppState.FINAL_SAVING, application);
 assertAppFinalStateSaved(application);
 return application;
}

@Test
public void testUnmanagedApp() throws IOException {
 ApplicationSubmissionContext subContext = new ApplicationSubmissionContextPBImpl();
 subContext.setUnmanagedAM(true);
 // test success path
 LOG.info("--- START: testUnmanagedAppSuccessPath ---");
 final String diagMsg = "some diagnostics";
 RMApp application = testCreateAppFinished(subContext, diagMsg);
 Assert.assertTrue("Finished app missing diagnostics",
   application.getDiagnostics().indexOf(diagMsg) != -1);
 // reset the counter of Mockito.verify
 reset(writer);
 reset(publisher);
 // test app fails after 1 app attempt failure
 LOG.info("--- START: testUnmanagedAppFailPath ---");
 application = testCreateAppRunning(subContext);
 RMAppEvent event = new RMAppFailedAttemptEvent(
   application.getApplicationId(), RMAppEventType.ATTEMPT_FAILED, "", false);
 application.handle(event);
 rmDispatcher.await();
 RMAppAttempt appAttempt = application.getCurrentAppAttempt();
 Assert.assertEquals(1, appAttempt.getAppAttemptId().getAttemptId());
 sendAppUpdateSavedEvent(application);
 assertFailed(application,
   ".*Unmanaged application.*Failing the application.*");
 assertAppFinalStateSaved(application);
}

@Test
public void testUnmanagedApp() throws IOException {
 ApplicationSubmissionContext subContext = new ApplicationSubmissionContextPBImpl();
 subContext.setUnmanagedAM(true);
 // test success path
 LOG.info("--- START: testUnmanagedAppSuccessPath ---");
 final String diagMsg = "some diagnostics";
 RMApp application = testCreateAppFinished(subContext, diagMsg);
 Assert.assertTrue("Finished app missing diagnostics",
   application.getDiagnostics().indexOf(diagMsg) != -1);
 // reset the counter of Mockito.verify
 reset(writer);
 reset(publisher);
 // test app fails after 1 app attempt failure
 LOG.info("--- START: testUnmanagedAppFailPath ---");
 application = testCreateAppRunning(subContext);
 RMAppEvent event = new RMAppFailedAttemptEvent(
   application.getApplicationId(), RMAppEventType.ATTEMPT_FAILED, "", false);
 application.handle(event);
 rmDispatcher.await();
 RMAppAttempt appAttempt = application.getCurrentAppAttempt();
 Assert.assertEquals(1, appAttempt.getAppAttemptId().getAttemptId());
 sendAppUpdateSavedEvent(application);
 assertFailed(application,
   ".*Unmanaged application.*Failing the application.*");
 assertAppFinalStateSaved(application);
 verifyRMAppFieldsForFinalTransitions(application);
}

protected RMApp testCreateAppFinished(
  ApplicationSubmissionContext submissionContext,
  String diagnostics) throws IOException {
 // unmanaged AMs don't use the FINISHING state
 RMApp application = null;
 if (submissionContext != null && submissionContext.getUnmanagedAM()) {
  application = testCreateAppRunning(submissionContext);
 } else {
  application = testCreateAppFinishing(submissionContext);
 }
 // RUNNING/FINISHING => FINISHED event RMAppEventType.ATTEMPT_FINISHED
 RMAppEvent finishedEvent = new RMAppEvent(application.getApplicationId(),
   RMAppEventType.ATTEMPT_FINISHED, diagnostics);
 application.handle(finishedEvent);
 assertAppState(RMAppState.FINISHED, application);
 assertTimesAtFinish(application);
 // finished without a proper unregister implies failed
 assertFinalAppStatus(FinalApplicationStatus.FAILED, application);
 Assert.assertTrue("Finished app missing diagnostics",
   application.getDiagnostics().indexOf(diagnostics) != -1);
 return application;
}

protected RMApp testCreateAppFinished(
  ApplicationSubmissionContext submissionContext,
  String diagnostics) throws IOException {
 // unmanaged AMs don't use the FINISHING state
 RMApp application = null;
 if (submissionContext != null && submissionContext.getUnmanagedAM()) {
  application = testCreateAppRunning(submissionContext);
 } else {
  application = testCreateAppFinishing(submissionContext);
 }
 // RUNNING/FINISHING => FINISHED event RMAppEventType.ATTEMPT_FINISHED
 RMAppEvent finishedEvent = new RMAppEvent(application.getApplicationId(),
   RMAppEventType.ATTEMPT_FINISHED, diagnostics);
 application.handle(finishedEvent);
 assertAppState(RMAppState.FINISHED, application);
 assertTimesAtFinish(application);
 // finished without a proper unregister implies failed
 assertFinalAppStatus(FinalApplicationStatus.FAILED, application);
 Assert.assertTrue("Finished app missing diagnostics",
   application.getDiagnostics().indexOf(diagnostics) != -1);
 return application;
}

@Test
public void testAppRunningKill() throws IOException {
 LOG.info("--- START: testAppRunningKill ---");
 UserGroupInformation fooUser = UserGroupInformation.createUserForTesting(
   "fooTestAppRunningKill", new String[] { "foo_group" });
 RMApp application = testCreateAppRunning(null);
 // RUNNING => KILLED event RMAppEventType.KILL
 RMAppEvent event = new RMAppKillByClientEvent(
   application.getApplicationId(), "Application killed by user.", fooUser,
   Server.getRemoteIp());
 application.handle(event);
 rmDispatcher.await();
 sendAttemptUpdateSavedEvent(application);
 sendAppUpdateSavedEvent(application);
 assertKilled(application);
 verifyApplicationFinished(RMAppState.KILLED);
 verifyAppRemovedSchedulerEvent(RMAppState.KILLED);
}

@Test
public void testAppRunningKill() throws IOException {
 LOG.info("--- START: testAppRunningKill ---");
 RMApp application = testCreateAppRunning(null);
 // RUNNING => KILLED event RMAppEventType.KILL
 UserGroupInformation fooUser = UserGroupInformation.createUserForTesting(
   "fooTestAppRunningKill", new String[] {"foo_group"});
 // SUBMITTED => KILLED event RMAppEventType.KILL
 RMAppEvent event = new RMAppKillByClientEvent(
   application.getApplicationId(), "Application killed by user.", fooUser,
   Server.getRemoteIp());
 application.handle(event);
 rmDispatcher.await();
 sendAttemptUpdateSavedEvent(application);
 sendAppUpdateSavedEvent(application);
 assertKilled(application);
 verifyApplicationFinished(RMAppState.KILLED);
 verifyAppRemovedSchedulerEvent(RMAppState.KILLED);
 verifyRMAppFieldsForFinalTransitions(application);
}

UserGroupInformation fooUser = UserGroupInformation.createUserForTesting(
  "fooTestAppKilledKill", new String[] { "foo_group" });
RMApp application = testCreateAppRunning(null);

LOG.info("--- START: testAppKilledKilled ---");
RMApp application = testCreateAppRunning(null);

LOG.info("--- START: testAppRunningFailed ---");
RMApp application = testCreateAppRunning(null);
RMAppAttempt appAttempt = application.getCurrentAppAttempt();
int expectedAttemptId = 1;

LOG.info("--- START: testAppRunningFailed ---");
RMApp application = testCreateAppRunning(null);
RMAppAttempt appAttempt = application.getCurrentAppAttempt();
int expectedAttemptId = 1;