Adding unit test for RoutingNode

Signed-off-by: Ankit Jain <jain.ankitk@gmail.com>
opensearch-project · Nov 3, 2021 · 8e7c453 · 8e7c453
1 parent 9f5b306
commit 8e7c453
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/server/src/test/java/org/opensearch/cluster/routing/RoutingNodeTests.java b/server/src/test/java/org/opensearch/cluster/routing/RoutingNodeTests.java
@@ -41,6 +41,7 @@
 import org.opensearch.test.OpenSearchTestCase;
 
 import java.net.InetAddress;
+import java.util.Iterator;
 
 import static java.util.Collections.emptyMap;
 import static java.util.Collections.emptySet;
@@ -86,6 +87,32 @@ public void testAdd() {
         assertThat(routingNode.getByShardId(new ShardId("test", IndexMetadata.INDEX_UUID_NA_VALUE, 4)), equalTo(relocatingShard0));
     }
 
+    public void testPrimaryFirstIterator() {
+        ShardRouting initializingShard3 = TestShardRouting.newShardRouting("test", 3, "node-1", false, ShardRoutingState.INITIALIZING);
+        ShardRouting relocatingShard4 = TestShardRouting.newShardRouting(
+            "test",
+            4,
+            "node-1",
+            "node-2",
+            true,
+            ShardRoutingState.RELOCATING
+        );
+        routingNode.add(initializingShard3);
+        routingNode.add(relocatingShard4);
+        final Iterator<ShardRouting> shardRoutingIterator = routingNode.iterator();
+        assertTrue(shardRoutingIterator.hasNext());
+        assertThat(shardRoutingIterator.next(), equalTo(relocatingShard4));
+        assertTrue(shardRoutingIterator.hasNext());
+        assertThat(shardRoutingIterator.next(), equalTo(unassignedShard0));
+        assertTrue(shardRoutingIterator.hasNext());
+        assertThat(shardRoutingIterator.next(), equalTo(initializingShard0));
+        assertTrue(shardRoutingIterator.hasNext());
+        assertThat(shardRoutingIterator.next(), equalTo(relocatingShard0));
+        assertTrue(shardRoutingIterator.hasNext());
+        assertThat(shardRoutingIterator.next(), equalTo(initializingShard3));
+        assertFalse(shardRoutingIterator.hasNext());
+    }
+
     public void testUpdate() {
         ShardRouting startedShard0 = TestShardRouting.newShardRouting("test", 0, "node-1", false, ShardRoutingState.STARTED);
         ShardRouting startedShard1 = TestShardRouting.newShardRouting("test", 1, "node-1", "node-2", false, ShardRoutingState.RELOCATING);