|
@@ -1199,6 +1199,7 @@ public class MockRM extends ResourceManager {
|
|
nm.nodeHeartbeat(true);
|
|
nm.nodeHeartbeat(true);
|
|
((AbstractYarnScheduler)rm.getResourceScheduler()).update();
|
|
((AbstractYarnScheduler)rm.getResourceScheduler()).update();
|
|
rm.drainEventsImplicitly();
|
|
rm.drainEventsImplicitly();
|
|
|
|
+ nm.nodeHeartbeat(true);
|
|
MockAM am = rm.sendAMLaunched(attempt.getAppAttemptId());
|
|
MockAM am = rm.sendAMLaunched(attempt.getAppAttemptId());
|
|
rm.waitForState(attempt.getAppAttemptId(), RMAppAttemptState.LAUNCHED);
|
|
rm.waitForState(attempt.getAppAttemptId(), RMAppAttemptState.LAUNCHED);
|
|
return am;
|
|
return am;
|
|
@@ -1215,6 +1216,7 @@ public class MockRM extends ResourceManager {
|
|
nm.nodeHeartbeat(true);
|
|
nm.nodeHeartbeat(true);
|
|
((AbstractYarnScheduler)rm.getResourceScheduler()).update();
|
|
((AbstractYarnScheduler)rm.getResourceScheduler()).update();
|
|
rm.drainEventsImplicitly();
|
|
rm.drainEventsImplicitly();
|
|
|
|
+ nm.nodeHeartbeat(true);
|
|
MockAM am = new MockAM(rm.getRMContext(), rm.masterService,
|
|
MockAM am = new MockAM(rm.getRMContext(), rm.masterService,
|
|
attempt.getAppAttemptId());
|
|
attempt.getAppAttemptId());
|
|
rm.waitForState(attempt.getAppAttemptId(), RMAppAttemptState.LAUNCHED);
|
|
rm.waitForState(attempt.getAppAttemptId(), RMAppAttemptState.LAUNCHED);
|