HDFS-16831. Add one ut to test shuffler

apache · Dec 23, 2022 · d154c04 · d154c04
1 parent 2329cb7
commit d154c04
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/...src/test/java/org/apache/hadoop/hdfs/server/federation/resolver/TestNamenodeResolver.java b/...src/test/java/org/apache/hadoop/hdfs/server/federation/resolver/TestNamenodeResolver.java
@@ -165,6 +165,21 @@ public void testShuffleObserverNNs() throws Exception {
     assertEquals(FederationNamenodeServiceState.OBSERVER, observerList2.get(1).getState());
     assertEquals(FederationNamenodeServiceState.ACTIVE, observerList2.get(2).getState());
     assertEquals(FederationNamenodeServiceState.STANDBY, observerList2.get(3).getState());
+
+    // Test shuffler
+    List<? extends FederationNamenodeContext> observerList3;
+    boolean hit = false;
+    for (int i = 0; i < 1000; i++) {
+      observerList3 = namenodeResolver.getNamenodesForNameserviceId(NAMESERVICES[0], true);
+      assertEquals(FederationNamenodeServiceState.OBSERVER, observerList3.get(0).getState());
+      assertEquals(FederationNamenodeServiceState.OBSERVER, observerList3.get(1).getState());
+      if (observerList3.get(0).getNamenodeId().equals(observerList2.get(1).getNamenodeId()) &&
+          observerList3.get(1).getNamenodeId().equals(observerList2.get(0).getNamenodeId())) {
+        hit = true;
+        break;
+      }
+    }
+    assertTrue(hit);
   }
 
   @Test