diff --git a/java/src/org/openqa/selenium/remote/http/jdk/JdkHttpClient.java b/java/src/org/openqa/selenium/remote/http/jdk/JdkHttpClient.java index e7d1d3569d7aa..f2d844a21b152 100644 --- a/java/src/org/openqa/selenium/remote/http/jdk/JdkHttpClient.java +++ b/java/src/org/openqa/selenium/remote/http/jdk/JdkHttpClient.java @@ -166,10 +166,13 @@ public WebSocket openSocket(HttpRequest request, WebSocket.Listener listener) { throw new ConnectionFailedException("JdkWebSocket initial request execution error", e); } + java.net.http.WebSocket.Builder builder = client.newWebSocketBuilder(); + + request.getHeaderNames().forEach(name -> builder.header(name, request.getHeader(name))); + CompletableFuture closed = new CompletableFuture<>(); CompletableFuture webSocketCompletableFuture = - client - .newWebSocketBuilder() + builder .connectTimeout(connectTimeout) .buildAsync( uri,