diff --git hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestNMClient.java hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestNMClient.java index 0d4a271..59d7a47 100644 --- hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestNMClient.java +++ hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestNMClient.java @@ -67,6 +67,7 @@ import org.apache.hadoop.yarn.server.resourcemanager.rmapp.attempt.RMAppAttemptState; import org.apache.hadoop.yarn.util.Records; import org.junit.After; +import org.junit.Assert; import org.junit.Before; import org.junit.Test; @@ -82,7 +83,7 @@ NMTokenCache nmTokenCache = null; @Before - public void setup() throws YarnException, IOException { + public void setup() throws YarnException, IOException, InterruptedException { // start minicluster conf = new YarnConfiguration(); yarnCluster = @@ -99,7 +100,9 @@ public void setup() throws YarnException, IOException { assertNotNull(yarnClient); assertEquals(STATE.STARTED, yarnClient.getServiceState()); - // get node info + // get node info after nodemanagers are up + assertTrue("Wait for nodemanagers to connect failed on yarn startup", + yarnCluster.waitForNodeManagersToConnect(5000)); nodeReports = yarnClient.getNodeReports(NodeState.RUNNING); // submit new app