diff --git a/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/tools/TestJMXGet.java b/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/tools/TestJMXGet.java index f83e7d007f5..7abe5fd36f6 100644 --- a/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/tools/TestJMXGet.java +++ b/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/tools/TestJMXGet.java @@ -31,6 +31,7 @@ import java.io.PrintStream; import java.lang.management.ManagementFactory; import java.util.Random; import java.util.Set; +import java.util.concurrent.TimeoutException; import javax.management.MBeanServerConnection; import javax.management.ObjectName; @@ -55,6 +56,7 @@ import org.junit.Test; * */ public class TestJMXGet { + public static final String WRONG_METRIC_VALUE_ERROR_MSG = "Unable to get the correct value for %s."; private Configuration config; private MiniDFSCluster cluster; @@ -106,9 +108,12 @@ public class TestJMXGet { assertTrue("error printAllValues", checkPrintAllValues(jmx)); //get some data from different source - DFSTestUtil.waitForMetric(jmx, "NumLiveDataNodes", numDatanodes); - assertEquals(numDatanodes, Integer.parseInt( + try { + DFSTestUtil.waitForMetric(jmx, "NumLiveDataNodes", numDatanodes); + } catch (TimeoutException e) { + assertEquals(String.format(WRONG_METRIC_VALUE_ERROR_MSG, "NumLiveDataNodes"),numDatanodes, Integer.parseInt( jmx.getValue("NumLiveDataNodes"))); + } assertGauge("CorruptBlocks", Long.parseLong(jmx.getValue("CorruptBlocks")), getMetrics("FSNamesystem")); @@ -158,8 +163,11 @@ public class TestJMXGet { String serviceName = "DataNode"; jmx.setService(serviceName); jmx.init(); - DFSTestUtil.waitForMetric(jmx, "BytesWritten", fileSize); - assertEquals(fileSize, Integer.parseInt(jmx.getValue("BytesWritten"))); + try { + DFSTestUtil.waitForMetric(jmx, "BytesWritten", fileSize); + } catch (TimeoutException e) { + assertEquals(String.format(WRONG_METRIC_VALUE_ERROR_MSG, "BytesWritten"), fileSize, Integer.parseInt(jmx.getValue("BytesWritten"))); + } cluster.shutdown(); MBeanServerConnection mbsc = ManagementFactory.getPlatformMBeanServer();