From 80cd1be3df3eb9032a907b2aab211655b3abbfc1 Mon Sep 17 00:00:00 2001
From: Sanjay Kumar <sanjay.kumar_p@nokia.com>
Date: Fri, 12 Apr 2024 13:59:22 +0530
Subject: [PATCH] Addition of test case on valid indexPatterns and Addresal of
 review comments

Signed-off-by: Sanjay Kumar <sanjay.kumar_p@nokia.com>
---
 .../UpdateAutoFollowPatternRequest.kt         |  4 ++--
 .../replication/util/ValidationUtil.kt        |  2 +-
 .../integ/rest/UpdateAutoFollowPatternIT.kt   | 19 +++++++++++++++++++
 3 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/src/main/kotlin/org/opensearch/replication/action/autofollow/UpdateAutoFollowPatternRequest.kt b/src/main/kotlin/org/opensearch/replication/action/autofollow/UpdateAutoFollowPatternRequest.kt
index d145eda2..7ee53c83 100644
--- a/src/main/kotlin/org/opensearch/replication/action/autofollow/UpdateAutoFollowPatternRequest.kt
+++ b/src/main/kotlin/org/opensearch/replication/action/autofollow/UpdateAutoFollowPatternRequest.kt
@@ -14,7 +14,7 @@ package org.opensearch.replication.action.autofollow
 import org.opensearch.replication.action.index.ReplicateIndexRequest
 import org.opensearch.replication.metadata.store.KEY_SETTINGS
 import org.opensearch.replication.util.ValidationUtil.validateName
-import org.opensearch.replication.util.ValidationUtil.validatePattern
+import org.opensearch.replication.util.ValidationUtil.validateAutoFollowPattern
 import org.opensearch.action.ActionRequestValidationException
 import org.opensearch.action.support.master.AcknowledgedRequest
 import org.opensearch.core.ParseField
@@ -118,7 +118,7 @@ class UpdateAutoFollowPatternRequest: AcknowledgedRequest<UpdateAutoFollowPatter
             if(pattern == null)
                validationException.addValidationError("Missing pattern")
             else
-               validatePattern(pattern, validationException)
+               validateAutoFollowPattern(pattern, validationException)
         }
 
         return if(validationException.validationErrors().isEmpty()) return null else validationException
diff --git a/src/main/kotlin/org/opensearch/replication/util/ValidationUtil.kt b/src/main/kotlin/org/opensearch/replication/util/ValidationUtil.kt
index 58bf1ebc..c9329968 100644
--- a/src/main/kotlin/org/opensearch/replication/util/ValidationUtil.kt
+++ b/src/main/kotlin/org/opensearch/replication/util/ValidationUtil.kt
@@ -109,7 +109,7 @@ object ValidationUtil {
      * Validate the pattern against the rules that we have for indexPattern name.
      */
 
-    fun validatePattern(pattern: String?, validationException: ValidationException) {
+    fun validateAutoFollowPattern(pattern: String?, validationException: ValidationException) {
 
         if (!Strings.validFileNameExcludingAstrix(pattern))
             validationException.addValidationError("Autofollow pattern: $pattern must not contain the following characters ${Strings.INVALID_FILENAME_CHARS}")
diff --git a/src/test/kotlin/org/opensearch/replication/integ/rest/UpdateAutoFollowPatternIT.kt b/src/test/kotlin/org/opensearch/replication/integ/rest/UpdateAutoFollowPatternIT.kt
index 7634667b..39980035 100644
--- a/src/test/kotlin/org/opensearch/replication/integ/rest/UpdateAutoFollowPatternIT.kt
+++ b/src/test/kotlin/org/opensearch/replication/integ/rest/UpdateAutoFollowPatternIT.kt
@@ -311,6 +311,7 @@ class UpdateAutoFollowPatternIT: MultiClusterRestTestCase() {
                 "Autofollow pattern: -leader must not start with '_' or '-'")
         assertPatternValidation(followerClient, "",
                 "Autofollow pattern:  must not be empty")
+
     }
     private fun assertPatternValidation(followerClient: RestHighLevelClient, pattern: String,
                                         errorMsg: String) {
@@ -320,6 +321,24 @@ class UpdateAutoFollowPatternIT: MultiClusterRestTestCase() {
                 .hasMessageContaining(errorMsg)
     }
 
+    fun `test auto follow should succeed on valid indexPatterns`() {
+        val followerClient = getClientForCluster(FOLLOWER)
+        createConnectionBetweenClusters(FOLLOWER, LEADER, connectionAlias)
+        assertValidPatternValidation(followerClient, "test-leader")
+        assertValidPatternValidation(followerClient, "test*")
+        assertValidPatternValidation(followerClient, "leader-*")
+        assertValidPatternValidation(followerClient, "leader_test")
+        assertValidPatternValidation(followerClient, "Leader_Test-*")
+        assertValidPatternValidation(followerClient, "Leader_*")
+
+    }
+
+    private fun assertValidPatternValidation(followerClient: RestHighLevelClient, pattern: String) {
+        Assertions.assertThatCode {
+            followerClient.updateAutoFollowPattern(connectionAlias, indexPatternName, pattern)
+        }.doesNotThrowAnyException()
+    }
+
     fun `test removing autofollow pattern stop autofollow task`() {
         val followerClient = getClientForCluster(FOLLOWER)
         val leaderClient = getClientForCluster(LEADER)