fix the race condition in test reset action start time

Signed-off-by: bowenlan-amzn <[email protected]>
bowenlan-amzn · Oct 13, 2023 · a848789 · a848789
1 parent b7dcd7a
commit a848789
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/...h/indexmanagement/indexstatemanagement/resthandler/RestRetryFailedManagedIndexActionIT.kt b/...h/indexmanagement/indexstatemanagement/resthandler/RestRetryFailedManagedIndexActionIT.kt
@@ -22,6 +22,7 @@ import org.opensearch.indexmanagement.spi.indexstatemanagement.model.ManagedInde
 import org.opensearch.indexmanagement.waitFor
 import org.opensearch.rest.RestRequest
 import org.opensearch.core.rest.RestStatus
+import org.opensearch.indexmanagement.indexstatemanagement.step.forcemerge.AttemptSetReadOnlyStep
 import java.time.Instant
 import java.util.Locale
 
@@ -257,20 +258,19 @@ class RestRetryFailedManagedIndexActionIT : IndexStateManagementRestTestCase() {
         val action = randomForceMergeActionConfig(maxNumSegments = 1)
         action.configRetry = ActionRetry(0)
         val policy = randomPolicy(states = listOf(randomState(actions = listOf(action))))
+
         createPolicy(policy, policyId = policyID)
         createIndex(indexName, policyID)
 
         val managedIndexConfig = getExistingManagedIndexConfig(indexName)
 
         // init policy on job
         updateManagedIndexConfigStartTime(managedIndexConfig)
-
         // verify we have policy
         waitFor { assertEquals(policyID, getExplainManagedIndexMetaData(indexName).policyID) }
 
         // speed up to execute set read only force merge step
         updateManagedIndexConfigStartTime(managedIndexConfig)
-
         waitFor {
             assertPredicatesOnMetaData(
                 listOf(
@@ -294,7 +294,6 @@ class RestRetryFailedManagedIndexActionIT : IndexStateManagementRestTestCase() {
 
         // speed up to execute attempt call force merge step
         updateManagedIndexConfigStartTime(managedIndexConfig)
-
         // verify failed and save the startTime
         var firstStartTime: Long = Long.MAX_VALUE
         waitFor {
@@ -331,7 +330,6 @@ class RestRetryFailedManagedIndexActionIT : IndexStateManagementRestTestCase() {
             FAILED_INDICES to emptyList<Map<String, Any>>()
         )
         assertAffectedIndicesResponseIsEqual(expectedErrorMessage, response.asMap())
-
         // verify actionStartTime was reset to null
         assertPredicatesOnMetaData(
             listOf(
@@ -348,7 +346,6 @@ class RestRetryFailedManagedIndexActionIT : IndexStateManagementRestTestCase() {
 
         // should execute and set the startTime again
         updateManagedIndexConfigStartTime(managedIndexConfig)
-
         // the new startTime should be greater than the first start time
         waitFor {
             assertPredicatesOnMetaData(
@@ -363,6 +360,10 @@ class RestRetryFailedManagedIndexActionIT : IndexStateManagementRestTestCase() {
                 ),
                 getExplainMap(indexName), false
             )
+            assertEquals(
+                AttemptSetReadOnlyStep.getSuccessMessage(indexName),
+                getExplainManagedIndexMetaData(indexName).info?.get("message")
+            )
         }
     }
 }