From 8db38c98a6c6ce9215ea998a2f544b2eabca4340 Mon Sep 17 00:00:00 2001
From: Ayush Saxena <ayushsaxena@apache.org>
Date: Sat, 27 Jun 2020 19:13:01 +0530
Subject: [PATCH] HDFS-15378.
 TestReconstructStripedFile#testErasureCodingWorkerXmitsWeight is failing on
 trunk. Contributed by hemanthboyina.

---
 .../org/apache/hadoop/hdfs/TestReconstructStripedFile.java     | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/TestReconstructStripedFile.java b/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/TestReconstructStripedFile.java
index b119e7855b7..6156c3da33c 100644
--- a/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/TestReconstructStripedFile.java
+++ b/hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/TestReconstructStripedFile.java
@@ -571,7 +571,8 @@ public class TestReconstructStripedFile {
       DataNodeFaultInjector.set(oldInjector);
       for (final DataNode curDn : cluster.getDataNodes()) {
         GenericTestUtils.waitFor(() -> curDn.getXceiverCount() <= 1, 10, 60000);
-        assertEquals(0, curDn.getXmitsInProgress());
+        GenericTestUtils.waitFor(() -> curDn.getXmitsInProgress() == 0, 10,
+            2500);
       }
     }
   }