refactor: simplify response handling in RpcHandlerDispatcher and update tests for null output

Author: brunoborges

src/main/java/com/github/copilot/sdk/RpcHandlerDispatcher.java

@@ -6,6 +6,7 @@
 
 import java.io.IOException;
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.Map;
 import java.util.concurrent.CompletableFuture;
 import java.util.logging.Level;
@@ -282,11 +283,7 @@ private void handleHooksInvoke(JsonRpcClient rpc, String requestId, JsonNode par
 
                 session.handleHooksInvoke(hookType, input).thenAccept(output -> {
                     try {
-                        if (output != null) {
-                            rpc.sendResponse(Long.parseLong(requestId), Map.of("output", output));
-                        } else {
-                            rpc.sendResponse(Long.parseLong(requestId), Map.of("output", (Object) null));
-                        }
+                        rpc.sendResponse(Long.parseLong(requestId), Collections.singletonMap("output", output));
                     } catch (IOException e) {
                         LOG.log(Level.SEVERE, "Error sending hooks response", e);
                     }

src/test/java/com/github/copilot/sdk/RpcHandlerDispatcherTest.java

@@ -461,10 +461,8 @@ void hooksInvokeWithNullOutput() throws Exception {
         invokeHandler("hooks.invoke", "31", params);
 
         JsonNode response = readResponse();
-        // Null output triggers NPE in Map.of() → falls to .exceptionally() → error
-        // response
-        assertNotNull(response.get("error"));
-        assertEquals(-32603, response.get("error").get("code").asInt());
+        JsonNode output = response.get("result").get("output");
+        assertTrue(output == null || output.isNull(), "Output should be null when no hook handler is set");
     }
 
     @Test
@@ -522,9 +520,7 @@ void hooksInvokeWithNoHooksRegistered() throws Exception {
         invokeHandler("hooks.invoke", "34", params);
 
         JsonNode response = readResponse();
-        // Null output triggers NPE in Map.of() → falls to .exceptionally() → error
-        // response
-        assertNotNull(response.get("error"));
-        assertEquals(-32603, response.get("error").get("code").asInt());
+        JsonNode output = response.get("result").get("output");
+        assertTrue(output == null || output.isNull(), "Output should be null when no hooks registered");
     }
 }