fix flaky kill process tests: removed sending response

Signed-off-by: Varun Deep Saini <varun.23bcs10048@ms.sst.scaler.com>

Author: varundeepsaini

acceptance/internal/prepare_server.go

@@ -221,7 +221,7 @@ func startLocalServer(t *testing.T,
 
 			if shouldKillCaller(stub, killCounters, killCountersMu) {
 				killCaller(t, stub.Pattern, req.Headers)
-				return testserver.Response{StatusCode: http.StatusOK}
+				return nil
 			}
 
 			return stub.Response
@@ -266,7 +266,14 @@ func killCaller(t *testing.T, pattern string, headers http.Header) {
 		return
 	}
 
-	t.Logf("KillCaller: killed PID %d (pattern: %s)", pid, pattern)
+	// Wait for process to actually exit before returning.
+	// This prevents the HTTP handler from sending a response while
+	// the process is still running, avoiding race conditions.
+	if waitForProcessExit(pid, 2*time.Second) {
+		t.Logf("KillCaller: killed PID %d (pattern: %s)", pid, pattern)
+	} else {
+		t.Logf("KillCaller: killed PID %d but timed out waiting for exit (pattern: %s)", pid, pattern)
+	}
 }
 
 func startProxyServer(t *testing.T,

libs/testserver/server.go

@@ -294,7 +294,7 @@ func (s *Server) Handle(method, path string, handler HandlerFunc) {
 			}
 		} else {
 			respAny := handler(request)
-			if respAny == nil && request.Context.Err() != nil {
+			if respAny == nil {
 				return
 			}
 			resp = normalizeResponse(s.t, respAny)