fixing TransportSamlLogoutActionTests

masseyke · Dec 4, 2023 · 4f43e4b · 4f43e4b
1 parent d1c6e19
commit 4f43e4b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...est/java/org/elasticsearch/xpack/security/action/saml/TransportSamlLogoutActionTests.java b/...est/java/org/elasticsearch/xpack/security/action/saml/TransportSamlLogoutActionTests.java
@@ -196,6 +196,11 @@ public void setup() throws Exception {
             ((Runnable) inv.getArguments()[1]).run();
             return null;
         }).when(securityIndex).prepareIndexIfNeededThenExecute(any(Consumer.class), any(Runnable.class));
+        doAnswer(inv -> {
+            ((Runnable) inv.getArguments()[1]).run();
+            ((Runnable) inv.getArguments()[2]).run();
+            return null;
+        }).when(securityIndex).prepareIndexIfNeededThenExecute(any(Consumer.class), any(Runnable.class), any(Runnable.class));
         doAnswer(inv -> {
             ((Runnable) inv.getArguments()[1]).run();
             return null;