Fixes test

Signed-off-by: Clay Downs <[email protected]>
opensearch-project · Apr 15, 2022 · b0730ce · b0730ce
1 parent 18dc35d
commit b0730ce
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/test/kotlin/org/opensearch/indexmanagement/indexstatemanagement/action/ShrinkActionIT.kt b/src/test/kotlin/org/opensearch/indexmanagement/indexstatemanagement/action/ShrinkActionIT.kt
@@ -298,6 +298,7 @@ class ShrinkActionIT : IndexStateManagementRestTestCase() {
         }
     }
 
+    @Suppress("UNCHECKED_CAST")
     fun `test allocation block picks correct node`() {
         val logger = LogManager.getLogger(::ShrinkActionIT)
         val nodes = getNodes()
@@ -394,7 +395,9 @@ class ShrinkActionIT : IndexStateManagementRestTestCase() {
                     WaitForShrinkStep.SUCCESS_MESSAGE,
                     getExplainManagedIndexMetaData(indexName).info?.get("message")
                 )
-                assertEquals("Write block setting was not reset after successful shrink", "false", getIndexBlocksWriteSetting(indexName))
+                val indexSettings = getIndexSettings(indexName) as Map<String, Map<String, Map<String, Any?>>>
+                val writeBlock = indexSettings[indexName]!!["settings"]!![IndexMetadata.SETTING_BLOCKS_WRITE] as String?
+                assertNull("Write block setting was not reset after successful shrink", writeBlock)
             }
         }
     }