Preserve user callbacks through DocumentOperationExecutorImpl

1 files changed, 10 insertions, 0 deletions

diff --git a/vespaclient-container-plugin/src/test/java/com/yahoo/document/restapi/DocumentOperationExecutorTest.java b/vespaclient-container-plugin/src/test/java/com/yahoo/document/restapi/DocumentOperationExecutorTest.java
index b57a0478ec2..331ad18e63c 100644
--- a/vespaclient-container-plugin/src/test/java/com/yahoo/document/restapi/DocumentOperationExecutorTest.java
+++ b/vespaclient-container-plugin/src/test/java/com/yahoo/document/restapi/DocumentOperationExecutorTest.java
@@ -232,6 +232,16 @@ public class DocumentOperationExecutorTest {
     }
 
     @Test
+    public void testCallback() {
+        AtomicBoolean called = new AtomicBoolean();
+        executor.get(doc1.getId(), parameters().withResponseHandler(__ -> called.set(true)), operationContext());
+        assertTrue(called.get());
+        assertEquals(List.of(), messages);
+        assertEquals(List.of(), errors);
+        assertEquals(List.of(), received);
+    }
+
+    @Test
     public void testVisit() throws InterruptedException {
         executor.put(new DocumentPut(doc1), parameters(), operationContext());
         executor.put(new DocumentPut(doc2), parameters(), operationContext());