[grid] keep the code and reason for closing the websocket

SeleniumHQ · Oct 25, 2023 · 43170d2 · 43170d2
1 parent 5d44d12
commit 43170d2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/java/src/org/openqa/selenium/netty/server/MessageOutboundConverter.java b/java/src/org/openqa/selenium/netty/server/MessageOutboundConverter.java
@@ -45,7 +45,9 @@ public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise)
     Message seMessage = (Message) msg;
 
     if (seMessage instanceof CloseMessage) {
-      ctx.writeAndFlush(new CloseWebSocketFrame(true, 0));
+      CloseMessage closeMessage = (CloseMessage) seMessage;
+      ctx.writeAndFlush(
+          new CloseWebSocketFrame(true, 0, closeMessage.code(), closeMessage.reason()));
     } else if (seMessage instanceof BinaryMessage) {
       ctx.writeAndFlush(
           new BinaryWebSocketFrame(