Make verifier return type as ActionResponse (#7167)

Signed-off-by: Heemin Kim <heemin@amazon.com>
opensearch-project · May 1, 2023 · 8386d10 · 8386d10
1 parent 142ae46
commit 8386d10
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/framework/src/main/java/org/opensearch/test/rest/RestActionTestCase.java b/test/framework/src/main/java/org/opensearch/test/rest/RestActionTestCase.java
@@ -122,7 +122,7 @@ public void reset() {
          * @param verifier A function which is called in place of {@link #doExecute(ActionType, ActionRequest, ActionListener)}
          */
         public <Request extends ActionRequest, Response extends ActionResponse> void setExecuteVerifier(
-            BiFunction<ActionType<Response>, Request, Void> verifier
+            BiFunction<ActionType<Response>, Request, Response> verifier
         ) {
             executeVerifier.set(verifier);
         }
@@ -142,7 +142,7 @@ public <Request extends ActionRequest, Response extends ActionResponse> void doE
          * @param verifier A function which is called in place of {@link #executeLocally(ActionType, ActionRequest, TaskListener)}
          */
         public <Request extends ActionRequest, Response extends ActionResponse> void setExecuteLocallyVerifier(
-            BiFunction<ActionType<Response>, Request, Void> verifier
+            BiFunction<ActionType<Response>, Request, Response> verifier
         ) {
             executeLocallyVerifier.set(verifier);
         }