unit test mocking client

Signed-off-by: bowenlan-amzn <bowenlan23@gmail.com>
bowenlan-amzn · Feb 22, 2025 · 81c5626 · 81c5626
1 parent 9032f4d
commit 81c5626
Show file tree

Hide file tree

Showing 2 changed files with 203 additions and 200 deletions.
diff --git a/...h/indexmanagement/indexstatemanagement/step/stopreplication/AttemptStopReplicationStep.kt b/...h/indexmanagement/indexstatemanagement/step/stopreplication/AttemptStopReplicationStep.kt
@@ -31,7 +31,7 @@ class AttemptStopReplicationStep : Step(name) {
         val indexName = context.metadata.index
         try {
             val stopIndexReplicationRequestObj = StopIndexReplicationRequest(indexName)
-            val response: AcknowledgedResponse = context.client.admin().indices().suspendUntil {
+            val response: AcknowledgedResponse = context.client.suspendUntil {
                 replicationPluginInterface.stopReplication(
                     context.client,
                     stopIndexReplicationRequestObj,