HDFS-9597. BaseReplicationPolicyTest should update data node stats after adding a data node. Contributed by Wei-Chiu Chuang.

(cherry picked from commit 8cfd672397efd91c471c417afef3aff85f64b506)

hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/server/blockmanagement/BaseReplicationPolicyTest.java

@@ -98,6 +98,8 @@ abstract public class BaseReplicationPolicyTest {
       cluster.add(dataNodes[i]);
       bm.getDatanodeManager().getHeartbeatManager().addDatanode(
           dataNodes[i]);
+      bm.getDatanodeManager().getHeartbeatManager().updateDnStat(
+          dataNodes[i]);
     }
     updateHeartbeatWithUsage();
   }