fix test

Author: arturobernalg

httpclient5-testing/src/test/java/org/apache/hc/client5/testing/websocket/WebSocketClientTest.java

@@ -93,18 +93,12 @@ private static URI uri(final int port, final String path) {
         return URI.create("ws://localhost:" + port + path);
     }
 
-    /**
-     * Build a client and ensure it is started before use.
-     */
     private static CloseableWebSocketClient newClient() {
         final CloseableWebSocketClient client = WebSocketClientBuilder.create().build();
         client.start(); // start reactor threads
         return client;
     }
 
-    /**
-     * Standard echo (uncompressed); ensures basic end-to-end is healthy.
-     */
     @Test
     void echo_uncompressed() throws Exception {
         final URI uri = uri(port, "/echo");
@@ -263,12 +257,13 @@ void max_message_1009() throws Exception {
             client.connect(u, new WebSocketListener() {
                 @Override
                 public void onOpen(final WebSocket ws) {
-                    // Client sends nothing; server pushes an oversized message.
+                    // Trigger the server to send an oversized text message.
+                    ws.sendText("trigger-too-big", true);
                 }
 
                 @Override
                 public void onText(final CharBuffer text, final boolean last) {
-                    // Depending on timing, we may or may not see text before the 1009 close.
+                    // We may or may not see some text before the 1009 close.
                 }
 
                 @Override
@@ -325,8 +320,6 @@ public void onError(final Throwable ex) {
         }
     }
 
-    // -------------------- Embedded servlets/sockets ---------------------------
-
     public static final class EchoServlet extends WebSocketServlet {
         @Override
         public void configure(final WebSocketServletFactory factory) {
@@ -401,17 +394,20 @@ public void configure(final WebSocketServletFactory factory) {
 
     public static final class TooBigSocket extends WebSocketAdapter {
         @Override
-        public void onWebSocketConnect(final Session sess) {
-            super.onWebSocketConnect(sess);
+        public void onWebSocketText(final String msg) {
+            final Session sess = getSession();
+            if (sess == null || !sess.isOpen()) {
+                return;
+            }
             final StringBuilder sb = new StringBuilder();
             final String chunk = "1234567890abcdef-";
             // Build something comfortably larger than the maxMessage (2 KiB in the test)
             while (sb.length() <= 8192) {
                 sb.append(chunk);
             }
-            final String msg = sb.toString();
-            sess.getRemote().sendString(msg, null);
-            // Do not send CLOSE here; the client should close with 1009.
+            final String big = sb.toString();
+            sess.getRemote().sendString(big, null);
+            // No CLOSE here; the client must decide to close with 1009.
         }
     }
 }