Fix connectionErrorUnopenedPortWithCallback

Author: heyams

instrumentation/play/play-mvc/play-mvc-2.4/javaagent/src/play24Test/java/io/opentelemetry/javaagent/instrumentation/play/v2_4/client/PlayWsClientTest.java

@@ -62,7 +62,15 @@ public void sendRequestWithCallback(
     internalSendRequest(wsRequest, method)
         .whenComplete(
             (wsResponse, throwable) -> {
-              httpClientResult.complete(wsResponse::getStatus, throwable);
+              if (wsResponse != null) {
+                httpClientResult.complete(wsResponse::getStatus, throwable);
+              } else {
+                httpClientResult.complete(
+                    () -> {
+                      throw new IllegalArgumentException("wsResponse is null!", throwable);
+                    },
+                    throwable);
+              }
             });
   }
 
@@ -78,8 +86,7 @@ protected void configure(HttpClientTestOptions.Builder optionsBuilder) {
           return attributes;
         });
 
-    optionsBuilder.setTestConnectionFailure(false);
-    optionsBuilder.setSpanEndsAfterType(null);
+    optionsBuilder.spanEndsAfterBody();
 
     // Play HTTP client uses AsyncHttpClient internally which does not support HTTP 1.1 pipelining
     // nor waiting for connection pool slots to free up. Therefore making a single connection test

instrumentation/play/play-mvc/play-mvc-2.4/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/play/v2_4/client/PlayWsClientTest.java

@@ -58,17 +58,23 @@ public void sendRequestWithCallback(
     internalSendRequest(wsRequest, method)
         .whenComplete(
             (wsResponse, throwable) -> {
-              httpClientResult.complete(wsResponse::getStatus, throwable);
+              if (wsResponse != null) {
+                httpClientResult.complete(wsResponse::getStatus, throwable);
+              } else {
+                httpClientResult.complete(
+                    () -> {
+                      throw new IllegalArgumentException("wsResponse is null!", throwable);
+                    },
+                    throwable);
+              }
             });
   }
 
   @Override
   protected void configure(HttpClientTestOptions.Builder optionsBuilder) {
     optionsBuilder.setTestRedirects(false);
     optionsBuilder.setTestReadTimeout(false);
-
-    optionsBuilder.setTestConnectionFailure(false);
-    optionsBuilder.setSpanEndsAfterType(null);
+    optionsBuilder.spanEndsAfterBody();
 
     // Play HTTP client uses AsyncHttpClient internally which does not support HTTP 1.1 pipelining
     // nor waiting for connection pool slots to free up. Therefore making a single connection test