diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/applicationsmanager/TestAMRestart.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/applicationsmanager/TestAMRestart.java index bc09b37..389dbd9 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/applicationsmanager/TestAMRestart.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/applicationsmanager/TestAMRestart.java @@ -837,6 +837,7 @@ public void testAMRestartNotLostContainerCompleteMsg() throws Exception { // launch the new AM RMAppAttempt attempt2 = app1.getCurrentAppAttempt(); + rm1.waitForState(newAttemptId, RMAppAttemptState.SCHEDULED); nm1.nodeHeartbeat(true); MockAM am2 = rm1.sendAMLaunched(attempt2.getAppAttemptId()); am2.registerAppAttempt();