Increasing test coverage

Signed-off-by: Joshua Palis <[email protected]>
opensearch-project · Aug 16, 2024 · 558e374 · 558e374
1 parent d817902
commit 558e374
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/test/java/org/opensearch/flowframework/workflow/UpdateIndexStepTests.java b/src/test/java/org/opensearch/flowframework/workflow/UpdateIndexStepTests.java
@@ -79,6 +79,12 @@ public void testUpdateIndexStepWithUpdatedSettings() throws ExecutionException,
             return null;
         }).when(indicesAdminClient).getSettings(any(), any());
 
+        doAnswer(invocation -> {
+            ActionListener<AcknowledgedResponse> ackResponseListener = invocation.getArgument(1);
+            ackResponseListener.onResponse(new AcknowledgedResponse(true));
+            return null;
+        }).when(indicesAdminClient).updateSettings(any(), any());
+
         // validate update settings request content
         @SuppressWarnings({ "unchecked" })
         ArgumentCaptor<UpdateSettingsRequest> updateSettingsRequestCaptor = ArgumentCaptor.forClass(UpdateSettingsRequest.class);
@@ -106,6 +112,12 @@ public void testUpdateIndexStepWithUpdatedSettings() throws ExecutionException,
         assertEquals(2, settingsToUpdate.size());
         assertEquals("_none", settingsToUpdate.get("index.default_pipeline"));
         assertEquals("_none", settingsToUpdate.get("index.search.default_pipeline"));
+
+        assertTrue(future.isDone());
+        WorkflowData returnedData = (WorkflowData) future.get();
+        assertEquals(Map.ofEntries(Map.entry(INDEX_NAME, indexName)), returnedData.getContent());
+        assertEquals(data.getWorkflowId(), returnedData.getWorkflowId());
+        assertEquals(data.getNodeId(), returnedData.getNodeId());
     }
 
     public void testFailedToUpdateIndexSettings() throws ExecutionException, InterruptedException, IOException {