diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-tests/src/test/java/org/apache/hadoop/yarn/server/TestMiniYarnClusterNodeUtilization.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-tests/src/test/java/org/apache/hadoop/yarn/server/TestMiniYarnClusterNodeUtilization.java index cf83e67..49a82c8 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-tests/src/test/java/org/apache/hadoop/yarn/server/TestMiniYarnClusterNodeUtilization.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-tests/src/test/java/org/apache/hadoop/yarn/server/TestMiniYarnClusterNodeUtilization.java @@ -214,8 +214,10 @@ private void verifySimulatedUtilization() throws InterruptedException { // We check if the nodeUtilization is up to date for (int i=0; i<100; i++) { for (RMNode ni : rmContext.getRMNodes().values()) { - if (ni.getNodeUtilization().equals(nodeUtilization)) { - break; + if (ni.getNodeUtilization() != null) { + if (ni.getNodeUtilization().equals(nodeUtilization)) { + break; + } } } Thread.sleep(100);