Merge pull request #3 from wntmddus/Add-Convert-Index-To-Remote

Fixing IT Test
opensearch-project · Dec 17, 2024 · cf02c08 · cf02c08
2 parents 1eecabf + aa26147
commit cf02c08
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/...n/org/opensearch/indexmanagement/indexstatemanagement/IndexStateManagementRestTestCase.kt b/...n/org/opensearch/indexmanagement/indexstatemanagement/IndexStateManagementRestTestCase.kt
@@ -101,6 +101,37 @@ abstract class IndexStateManagementRestTestCase : IndexManagementRestTestCase()
     protected open fun enableValidationService() {
         updateValidationServiceSetting(true)
     }
+    protected fun createOrUpdateRole() {
+        val roleBody = """
+        {
+          "cluster_permissions": ["cluster:admin/snapshot/restore", "cluster:admin/snapshot/create"],
+          "index_permissions": [
+            {
+              "index_patterns": ["*"],
+              "allowed_actions": ["indices:*"]
+            }
+          ]
+        }
+        """.trimIndent()
+
+        client().makeRequest(
+            method = "PUT",
+            endpoint = "/_plugins/_security/api/roles/snapshot_restore_role",
+            entity = StringEntity(roleBody, ContentType.APPLICATION_JSON),
+        )
+
+        // Map your test user to snapshot_restore_role
+        val roleMappingBody = """
+        {
+          "users": ["testUser"]
+        }
+        """.trimIndent()
+        client().makeRequest(
+            method = "PUT",
+            endpoint = "/_plugins/_security/api/rolesmapping/snapshot_restore_role",
+            entity = StringEntity(roleMappingBody, ContentType.APPLICATION_JSON),
+        )
+    }
 
     protected fun createPolicy(
         policy: Policy,

diff --git a/...rg/opensearch/indexmanagement/indexstatemanagement/action/ConvertIndexToRemoteActionIT.kt b/...rg/opensearch/indexmanagement/indexstatemanagement/action/ConvertIndexToRemoteActionIT.kt
@@ -25,6 +25,7 @@ class ConvertIndexToRemoteActionIT : IndexStateManagementRestTestCase() {
         val policyID = "${testIndexName}_policy_basic"
         val repository = "repository"
 
+        createOrUpdateRole()
         // Step 1: Create an index and index some data
         createIndex(indexName, null)
         indexDoc(indexName, "1", """{"field": "value1"}""")