Deduplicate test host message response handling

Agent-Logs-Url: https://github.com/larohra/azure-functions-java-worker/sessions/db76de91-28a8-4391-b3a2-d8796d6725e4

Co-authored-by: larohra <41490930+larohra@users.noreply.github.com>

Author: Copilot

src/test/java/com/microsoft/azure/functions/worker/test/utilities/FunctionsTestHost.java

@@ -203,14 +203,7 @@ void handleMessage(String requestId, Function<StreamingMessage, StreamingMessage
                     this.getResponseCondition(requestId).await(RESPONSE_POLL_MILLIS, TimeUnit.MILLISECONDS);
                     FunctionsTestHost.this.throwIfListeningFailed();
                 }
-                StreamingMessage message = this.respValue.get(requestId);
-                StreamingMessage response = null;
-                if (handler != null) {
-                    response = handler.apply(message);
-                }
-                if (response != null) {
-                    this.responder.get(requestId).onNext(response);
-                }
+                this.respondToMessage(requestId, handler);
             } finally {
                 this.lock.unlock();
             }
@@ -230,19 +223,23 @@ void handleMessageWithTimeout(String requestId, Function<StreamingMessage, Strea
                     long waitMillis = Math.max(1L, Math.min(TimeUnit.NANOSECONDS.toMillis(remainingNanos), RESPONSE_POLL_MILLIS));
                     this.getResponseCondition(requestId).await(waitMillis, TimeUnit.MILLISECONDS);
                 }
-                StreamingMessage message = this.respValue.get(requestId);
-                StreamingMessage response = null;
-                if (handler != null) {
-                    response = handler.apply(message);
-                }
-                if (response != null) {
-                    this.responder.get(requestId).onNext(response);
-                }
+                this.respondToMessage(requestId, handler);
             } finally {
                 this.lock.unlock();
             }
         }
 
+        private void respondToMessage(String requestId, Function<StreamingMessage, StreamingMessage> handler) {
+            StreamingMessage message = this.respValue.get(requestId);
+            StreamingMessage response = null;
+            if (handler != null) {
+                response = handler.apply(message);
+            }
+            if (response != null) {
+                this.responder.get(requestId).onNext(response);
+            }
+        }
+
         private StreamingMessage initWorker() {
             WorkerInitRequest.Builder request = WorkerInitRequest.newBuilder().setHostVersion("2.0.0");
             return StreamingMessage.newBuilder().setRequestId(INITWORKER_REQID).setWorkerInitRequest(request).build();