How to use

startRMs

method

in

org.apache.hadoop.yarn.server.resourcemanager.TestKillApplicationWithRMHA

private void startRMsWithCustomizedClientRMService() throws IOException {
 final Configuration conf1 = new Configuration(confForRM1);
 rm1 = new MockRM(conf1) {
  @Override
  protected ClientRMService createClientRMService() {
   return new MyClientRMService(this.rmContext, this.scheduler,
     this.rmAppManager, this.applicationACLsManager,
     this.queueACLsManager, getRMContext().getRMDelegationTokenSecretManager());
  }
 };
 rm2 = new MockRM(confForRM2);
 startRMs(rm1, conf1, rm2, confForRM2);
}

private void startRMsWithCustomizedClientRMService() throws IOException {
 final Configuration conf1 = new Configuration(confForRM1);
 rm1 = new MockRM(conf1) {
  @Override
  protected ClientRMService createClientRMService() {
   return new MyClientRMService(this.rmContext, this.scheduler,
     this.rmAppManager, this.applicationACLsManager,
     this.queueACLsManager, getRMContext().getRMDelegationTokenSecretManager());
  }
 };
 rm2 = new MockRM(confForRM2);
 startRMs(rm1, conf1, rm2, confForRM2);
}

@Test (timeout = 20000)
public void testKillAppWhenFailoverHappensAtRunningState()
  throws Exception {
 startRMs();
 MockNM nm1 = new MockNM("127.0.0.1:1234", 15120,
   rm1.getResourceTrackerService());
 nm1.registerNode();
 // create app and launch the AM
 RMApp app0 = rm1.submitApp(200);
 MockAM am0 = launchAM(app0, rm1, nm1);
 // failover and kill application
 // The application is at RUNNING State when failOver happens.
 // Since RMStateStore has already saved ApplicationState, the active RM
 // will load the ApplicationState. After that, the application will be at
 // ACCEPTED State. Because the application is not at Final State,
 // KillApplicationResponse.getIsKillCompleted is expected to return false.
 failOverAndKillApp(app0.getApplicationId(),
   am0.getApplicationAttemptId(), RMAppState.RUNNING,
   RMAppAttemptState.RUNNING, RMAppState.ACCEPTED);
}

@Test (timeout = 20000)
public void testKillAppWhenFailoverHappensAtRunningState()
  throws Exception {
 startRMs();
 MockNM nm1 = new MockNM("127.0.0.1:1234", 15120,
   rm1.getResourceTrackerService());
 nm1.registerNode();
 // create app and launch the AM
 RMApp app0 = rm1.submitApp(200);
 MockAM am0 = launchAM(app0, rm1, nm1);
 // failover and kill application
 // The application is at RUNNING State when failOver happens.
 // Since RMStateStore has already saved ApplicationState, the active RM
 // will load the ApplicationState. After that, the application will be at
 // ACCEPTED State. Because the application is not at Final State,
 // KillApplicationResponse.getIsKillCompleted is expected to return false.
 failOverAndKillApp(app0.getApplicationId(),
   am0.getApplicationAttemptId(), RMAppState.RUNNING,
   RMAppAttemptState.RUNNING, RMAppState.ACCEPTED);
}

@Test (timeout = 20000)
public void testKillAppWhenFailoverHappensAtFinalState()
  throws Exception {
 startRMs();
 MockNM nm1 = new MockNM("127.0.0.1:1234", 15120,
   rm1.getResourceTrackerService());
 nm1.registerNode();
 // create app and launch the AM
 RMApp app0 = rm1.submitApp(200);
 MockAM am0 = launchAM(app0, rm1, nm1);
 // kill the app.
 rm1.killApp(app0.getApplicationId());
 rm1.waitForState(app0.getApplicationId(), RMAppState.KILLED);
 rm1.waitForState(am0.getApplicationAttemptId(), RMAppAttemptState.KILLED);
 // failover and kill application
 // The application is at Killed State and RMStateStore has already
 // saved this applicationState. After failover happens, the current
 // active RM will load the ApplicationState whose RMAppState is killed.
 // Because this application is at Final State,
 // KillApplicationResponse.getIsKillCompleted is expected to return true.
 failOverAndKillApp(app0.getApplicationId(),
   am0.getApplicationAttemptId(), RMAppState.KILLED,
   RMAppAttemptState.KILLED, RMAppState.KILLED);
}

@Test (timeout = 20000)
public void testKillAppWhenFailoverHappensAtFinalState()
  throws Exception {
 startRMs();
 MockNM nm1 = new MockNM("127.0.0.1:1234", 15120,
   rm1.getResourceTrackerService());
 nm1.registerNode();
 // create app and launch the AM
 RMApp app0 = rm1.submitApp(200);
 MockAM am0 = launchAM(app0, rm1, nm1);
 // kill the app.
 rm1.killApp(app0.getApplicationId());
 rm1.waitForState(app0.getApplicationId(), RMAppState.KILLED);
 rm1.waitForState(am0.getApplicationAttemptId(), RMAppAttemptState.KILLED);
 // failover and kill application
 // The application is at Killed State and RMStateStore has already
 // saved this applicationState. After failover happens, the current
 // active RM will load the ApplicationState whose RMAppState is killed.
 // Because this application is at Final State,
 // KillApplicationResponse.getIsKillCompleted is expected to return true.
 failOverAndKillApp(app0.getApplicationId(),
   am0.getApplicationAttemptId(), RMAppState.KILLED,
   RMAppAttemptState.KILLED, RMAppState.KILLED);
}