Add SimpleIndexTemplate tests

Signed-off-by: Arpit Bandejiya <abandeji@amazon.com>
opensearch-project · Dec 27, 2022 · ebe089a · ebe089a
1 parent f563943
commit ebe089a
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/...r/src/internalClusterTest/java/org/opensearch/indices/template/SimpleIndexTemplateIT.java b/...r/src/internalClusterTest/java/org/opensearch/indices/template/SimpleIndexTemplateIT.java
@@ -1031,22 +1031,30 @@ public void testPartitionedTemplate() throws Exception {
     }
 
     public void testAwarenessReplicaBalance() throws IOException {
-        manageReplicaBalanceSetting(true);
+        manageUseZoneForReplicaSetting(true);
         int updated = 0;
         try {
             client().admin()
                 .indices()
                 .preparePutTemplate("template_1")
                 .setPatterns(Arrays.asList("a*", "b*"))
-                .setSettings(Settings.builder().put(IndexMetadata.SETTING_NUMBER_OF_REPLICAS, 1))
+                .setSettings(Settings.builder().put(IndexMetadata.SETTING_NUMBER_OF_REPLICAS, 2))
                 .get();
             updated++;
 
             client().admin()
                 .indices()
                 .preparePutTemplate("template_1")
                 .setPatterns(Arrays.asList("a*", "b*"))
-                .setSettings(Settings.builder().put(IndexMetadata.SETTING_AUTO_EXPAND_REPLICAS, "0-1"))
+                .setSettings(Settings.builder().put(IndexMetadata.SETTING_NUMBER_OF_SHARDS, 4))
+                .get();
+            updated++;
+
+            client().admin()
+                .indices()
+                .preparePutTemplate("template_1")
+                .setPatterns(Arrays.asList("a*", "b*"))
+                .setSettings(Settings.builder().put(IndexMetadata.SETTING_AUTO_EXPAND_REPLICAS, "0-2"))
                 .get();
             updated++;
 
@@ -1060,12 +1068,12 @@ public void testAwarenessReplicaBalance() throws IOException {
             fail("should have thrown an exception about the replica  count");
         } catch (InvalidIndexTemplateException e) {
             assertEquals(
-                "index_template [template_1] invalid, cause [Validation Failed: 1: expected total copies needs to be a multiple of total awareness attributes [2];]",
+                "index_template [template_1] invalid, cause [Validation Failed: 1: expected total copies needs to be a multiple of total awareness attributes [3];]",
                 e.getMessage()
             );
-            assertEquals(2, updated);
+            assertEquals(3, updated);
         } finally {
-            manageReplicaBalanceSetting(false);
+            manageUseZoneForReplicaSetting(false);
         }
     }