Potential fix

Signed-off-by: Emmanuel Hugonnet <[email protected]>

Author: ehsavoie

transport/jsonrpc/src/test/java/io/a2a/transport/jsonrpc/handler/JSONRPCHandlerTest.java

@@ -186,7 +186,8 @@ public void testOnMessageNewMessageSuccessMocks() {
         SendMessageResponse response;
         try (MockedConstruction<EventConsumer> mocked = Mockito.mockConstruction(
                 EventConsumer.class,
-                (mock, context) -> {Mockito.doReturn(ZeroPublisher.fromItems(wrapEvent(MINIMAL_TASK))).when(mock).consumeAll();})){
+                (mock, context) -> {Mockito.doReturn(ZeroPublisher.fromItems(wrapEvent(MINIMAL_TASK))).when(mock).consumeAll();
+                Mockito.doCallRealMethod().when(mock).createAgentRunnableDoneCallback();})){
             response = handler.onMessageSend(request, callContext);
         }
         assertNull(response.getError());