http-client-branch: (WebSocket) more tests http-client-branch
authorprappo
Tue, 30 Jan 2018 17:28:12 +0000
branchhttp-client-branch
changeset 56045 5c6e3b76d2ad
parent 56044 a8423a38386e
child 56046 75d9bc7febf5
http-client-branch: (WebSocket) more tests
src/jdk.incubator.httpclient/share/classes/jdk/incubator/http/internal/websocket/WebSocketImpl.java
test/jdk/java/net/httpclient/websocket/DummyWebSocketServer.java
test/jdk/java/net/httpclient/websocket/Exceptionally.java
--- a/src/jdk.incubator.httpclient/share/classes/jdk/incubator/http/internal/websocket/WebSocketImpl.java	Tue Jan 30 13:52:40 2018 +0000
+++ b/src/jdk.incubator.httpclient/share/classes/jdk/incubator/http/internal/websocket/WebSocketImpl.java	Tue Jan 30 17:28:12 2018 +0000
@@ -38,6 +38,7 @@
 import java.net.ProtocolException;
 import java.net.URI;
 import java.nio.ByteBuffer;
+import java.util.Objects;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.CompletionStage;
 import java.util.concurrent.TimeoutException;
@@ -148,6 +149,7 @@
     @Override
     public CompletableFuture<WebSocket> sendText(CharSequence message,
                                                  boolean isLast) {
+        Objects.requireNonNull(message);
         if (!outstandingSend.compareAndSet(false, true)) {
             return failedFuture(new IllegalStateException("Send pending"));
         }
@@ -158,6 +160,7 @@
     @Override
     public CompletableFuture<WebSocket> sendBinary(ByteBuffer message,
                                                    boolean isLast) {
+        Objects.requireNonNull(message);
         if (!outstandingSend.compareAndSet(false, true)) {
             return failedFuture(new IllegalStateException("Send pending"));
         }
--- a/test/jdk/java/net/httpclient/websocket/DummyWebSocketServer.java	Tue Jan 30 13:52:40 2018 +0000
+++ b/test/jdk/java/net/httpclient/websocket/DummyWebSocketServer.java	Tue Jan 30 17:28:12 2018 +0000
@@ -80,7 +80,7 @@
  *     Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=
  *     Sec-WebSocket-Protocol: chat
  */
-public final class DummyWebSocketServer implements Closeable {
+public class DummyWebSocketServer implements Closeable {
 
     private final AtomicBoolean started = new AtomicBoolean();
     private final Thread thread;
@@ -108,12 +108,7 @@
                         List<String> strings = asList(request.toString().split("\r\n"));
                         List<String> response = mapping.apply(strings);
                         writeResponse(channel, response);
-                        // Read until the thread is interrupted or an error occurred
-                        // or the input is shutdown
-                        ByteBuffer b = ByteBuffer.allocate(1024);
-                        while (channel.read(b) != -1) {
-                            b.clear();
-                        }
+                        serve(channel);
                     } catch (IOException e) {
                         err.println("Error in connection: " + channel + ", " + e);
                     } finally {
@@ -133,6 +128,15 @@
         thread.setDaemon(false);
     }
 
+    protected void serve(SocketChannel channel) throws IOException {
+        // Read until the thread is interrupted or an error occurred
+        // or the input is shutdown
+        ByteBuffer b = ByteBuffer.allocate(1024);
+        while (channel.read(b) != -1) {
+            b.clear();
+        }
+    }
+
     public void open() throws IOException {
         err.println("Starting");
         if (!started.compareAndSet(false, true)) {
--- a/test/jdk/java/net/httpclient/websocket/Exceptionally.java	Tue Jan 30 13:52:40 2018 +0000
+++ b/test/jdk/java/net/httpclient/websocket/Exceptionally.java	Tue Jan 30 17:28:12 2018 +0000
@@ -32,8 +32,13 @@
 
 import java.io.IOException;
 import java.nio.ByteBuffer;
+import java.nio.CharBuffer;
+import java.nio.channels.SocketChannel;
+import java.nio.charset.StandardCharsets;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.CompletionException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
 
 import static jdk.incubator.http.HttpClient.newHttpClient;
 import static jdk.incubator.http.WebSocket.NORMAL_CLOSURE;
@@ -41,7 +46,10 @@
 
 public class Exceptionally {
 
-    static final Class<NullPointerException> NPE = NullPointerException.class;
+    private static final Class<NullPointerException> NPE
+            = NullPointerException.class;
+    private static final Class<IllegalArgumentException> IAE
+            = IllegalArgumentException.class;
 
     @Test
     public void testNull() throws IOException {
@@ -52,21 +60,16 @@
                     .buildAsync(server.getURI(), new WebSocket.Listener() { })
                     .join();
 
+            assertThrows(NPE, () -> ws.sendText(null, false));
+            assertThrows(NPE, () -> ws.sendText(null, true));
+            assertThrows(NPE, () -> ws.sendBinary(null, false));
+            assertThrows(NPE, () -> ws.sendBinary(null, true));
             assertThrows(NPE, () -> ws.sendPing(null));
             assertThrows(NPE, () -> ws.sendPong(null));
             assertThrows(NPE, () -> ws.sendClose(NORMAL_CLOSURE, null));
-
-            // ... add more NPE scenarios
         }
     }
 
-    private static String stringWithNBytes(int n) {
-        StringBuilder sb = new StringBuilder(n);
-        for (int i = 0; i < n; i++)
-            sb.append("A");
-        return sb.toString();
-    }
-
     @Test
     public void testIllegalArgument() throws IOException {
         try (DummyWebSocketServer server = new DummyWebSocketServer()) {
@@ -79,23 +82,184 @@
             assertIAE(ws.sendPing(ByteBuffer.allocate(126)));
             assertIAE(ws.sendPing(ByteBuffer.allocate(127)));
             assertIAE(ws.sendPing(ByteBuffer.allocate(128)));
-            assertIAE(ws.sendPing(ByteBuffer.allocate(150)));
+            assertIAE(ws.sendPing(ByteBuffer.allocate(129)));
+            assertIAE(ws.sendPing(ByteBuffer.allocate(256)));
 
             assertIAE(ws.sendPong(ByteBuffer.allocate(126)));
             assertIAE(ws.sendPong(ByteBuffer.allocate(127)));
             assertIAE(ws.sendPong(ByteBuffer.allocate(128)));
-            assertIAE(ws.sendPong(ByteBuffer.allocate(150)));
+            assertIAE(ws.sendPong(ByteBuffer.allocate(129)));
+            assertIAE(ws.sendPong(ByteBuffer.allocate(256)));
+
+            assertIAE(ws.sendText(incompleteString(), true));
+            assertIAE(ws.sendText(incompleteString(), false));
+            assertIAE(ws.sendText(malformedString(), true));
+            assertIAE(ws.sendText(malformedString(), false));
 
             assertIAE(ws.sendClose(NORMAL_CLOSURE, stringWithNBytes(124)));
-            assertIAE(ws.sendClose(NORMAL_CLOSURE, stringWithNBytes(150)));
-            assertIAE(ws.sendClose(NORMAL_CLOSURE - 1, "a reason"));
+            assertIAE(ws.sendClose(NORMAL_CLOSURE, stringWithNBytes(125)));
+            assertIAE(ws.sendClose(NORMAL_CLOSURE, stringWithNBytes(128)));
+            assertIAE(ws.sendClose(NORMAL_CLOSURE, stringWithNBytes(256)));
+            assertIAE(ws.sendClose(NORMAL_CLOSURE, stringWithNBytes(257)));
+            assertIAE(ws.sendClose(NORMAL_CLOSURE, stringWith2NBytes((123 / 2) + 1)));
+            assertIAE(ws.sendClose(NORMAL_CLOSURE, malformedString()));
+            assertIAE(ws.sendClose(NORMAL_CLOSURE, incompleteString()));
 
-            // ... add more CF complete exceptionally scenarios
+            assertIAE(ws.sendClose(-2, "a reason"));
+            assertIAE(ws.sendClose(-1, "a reason"));
+            assertIAE(ws.sendClose(0, "a reason"));
+            assertIAE(ws.sendClose(1, "a reason"));
+            assertIAE(ws.sendClose(500, "a reason"));
+            assertIAE(ws.sendClose(998, "a reason"));
+            assertIAE(ws.sendClose(999, "a reason"));
+            assertIAE(ws.sendClose(1002, "a reason"));
+            assertIAE(ws.sendClose(1003, "a reason"));
+            assertIAE(ws.sendClose(1006, "a reason"));
+            assertIAE(ws.sendClose(1007, "a reason"));
+            assertIAE(ws.sendClose(1009, "a reason"));
+            assertIAE(ws.sendClose(1010, "a reason"));
+            assertIAE(ws.sendClose(1012, "a reason"));
+            assertIAE(ws.sendClose(1013, "a reason"));
+            assertIAE(ws.sendClose(1015, "a reason"));
+            assertIAE(ws.sendClose(5000, "a reason"));
+            assertIAE(ws.sendClose(32768, "a reason"));
+            assertIAE(ws.sendClose(65535, "a reason"));
+            assertIAE(ws.sendClose(65536, "a reason"));
+            assertIAE(ws.sendClose(Integer.MAX_VALUE, "a reason"));
+            assertIAE(ws.sendClose(Integer.MIN_VALUE, "a reason"));
+
+            assertThrows(IAE, () -> ws.request(Integer.MIN_VALUE));
+            assertThrows(IAE, () -> ws.request(-1));
+            assertThrows(IAE, () -> ws.request(0));
+        }
+    }
+
+    @Test
+    public void testIllegalStateOutstanding1() throws Exception {
+        try (DummyWebSocketServer server = notReadingServer()) {
+            server.open();
+            WebSocket ws = newHttpClient()
+                    .newWebSocketBuilder()
+                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
+                    .join();
+
+            ByteBuffer data = ByteBuffer.allocate(65536);
+            for (int i = 0; ; i++) {
+                System.out.println("cycle #" + i);
+                try {
+                    ws.sendBinary(data, true).get(10, TimeUnit.SECONDS);
+                    data.clear();
+                } catch (TimeoutException e) {
+                    break;
+                }
+            }
+            assertISE(ws.sendBinary(ByteBuffer.allocate(0), true));
+            assertISE(ws.sendText("", true));
         }
     }
 
     @Test
-    public void testIllegalState() throws IOException {
+    public void testIllegalStateOutstanding2() throws Exception {
+        try (DummyWebSocketServer server = notReadingServer()) {
+            server.open();
+            WebSocket ws = newHttpClient()
+                    .newWebSocketBuilder()
+                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
+                    .join();
+
+            CharBuffer data = CharBuffer.allocate(65536);
+            for (int i = 0; ; i++) {
+                System.out.println("cycle #" + i);
+                try {
+                    ws.sendText(data, true).get(10, TimeUnit.SECONDS);
+                    data.clear();
+                } catch (TimeoutException e) {
+                    break;
+                }
+            }
+            assertISE(ws.sendText("", true));
+            assertISE(ws.sendBinary(ByteBuffer.allocate(0), true));
+        }
+    }
+
+    private static DummyWebSocketServer notReadingServer() {
+        return new DummyWebSocketServer() {
+            @Override
+            protected void serve(SocketChannel channel) throws IOException {
+                try {
+                    Thread.sleep(Long.MAX_VALUE);
+                } catch (InterruptedException e) {
+                    throw new IOException(e);
+                }
+            }
+        };
+    }
+
+    @Test
+    public void testIllegalStateIntermixed1() throws IOException {
+        try (DummyWebSocketServer server = new DummyWebSocketServer()) {
+            server.open();
+            WebSocket ws = newHttpClient()
+                    .newWebSocketBuilder()
+                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
+                    .join();
+
+            ws.sendBinary(ByteBuffer.allocate(16), false).join();
+            assertISE(ws.sendText("text", false));
+            assertISE(ws.sendText("text", true));
+        }
+    }
+
+    @Test
+    public void testIllegalStateIntermixed2() throws IOException {
+        try (DummyWebSocketServer server = new DummyWebSocketServer()) {
+            server.open();
+            WebSocket ws = newHttpClient()
+                    .newWebSocketBuilder()
+                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
+                    .join();
+
+            ws.sendText("text", false).join();
+            assertISE(ws.sendBinary(ByteBuffer.allocate(16), false));
+            assertISE(ws.sendBinary(ByteBuffer.allocate(16), true));
+        }
+    }
+
+    private static String malformedString() {
+        return new String(new char[]{0xDC00, 0xD800});
+    }
+
+    private static String incompleteString() {
+        return new String(new char[]{0xD800});
+    }
+
+    private static String stringWithNBytes(int n) {
+        StringBuilder sb = new StringBuilder(n);
+        for (int i = 0; i < n; i++) {
+            sb.append("A");
+        }
+        return sb.toString();
+    }
+
+    private static String stringWith2NBytes(int n) {
+        // Russian alphabet repeated cyclically
+        char FIRST = '\u0410';
+        char LAST = '\u042F';
+        StringBuilder sb = new StringBuilder(n);
+        char c = FIRST;
+        for (int i = 0; i < n; i++) {
+            if (++c > LAST) {
+                c = FIRST;
+            }
+            sb.append(c);
+        }
+        String s = sb.toString();
+        assert s.length() == n && s.getBytes(StandardCharsets.UTF_8).length == 2 * n;
+        return s;
+    }
+
+    @Test
+    public void testIllegalStateSendClose() throws IOException {
         try (DummyWebSocketServer server = new DummyWebSocketServer()) {
             server.open();
             WebSocket ws = newHttpClient()
@@ -105,30 +269,37 @@
 
             ws.sendClose(NORMAL_CLOSURE, "normal close").join();
 
+            assertISE(ws.sendText("", true));
+            assertISE(ws.sendText("", false));
+            assertISE(ws.sendText("abc", true));
+            assertISE(ws.sendText("abc", false));
+            assertISE(ws.sendBinary(ByteBuffer.allocate(0), true));
+            assertISE(ws.sendBinary(ByteBuffer.allocate(0), false));
+            assertISE(ws.sendBinary(ByteBuffer.allocate(1), true));
+            assertISE(ws.sendBinary(ByteBuffer.allocate(1), false));
+
             assertISE(ws.sendPing(ByteBuffer.allocate(125)));
             assertISE(ws.sendPing(ByteBuffer.allocate(124)));
-            assertISE(ws.sendPing(ByteBuffer.allocate(  1)));
-            assertISE(ws.sendPing(ByteBuffer.allocate(  0)));
+            assertISE(ws.sendPing(ByteBuffer.allocate(1)));
+            assertISE(ws.sendPing(ByteBuffer.allocate(0)));
 
             assertISE(ws.sendPong(ByteBuffer.allocate(125)));
             assertISE(ws.sendPong(ByteBuffer.allocate(124)));
-            assertISE(ws.sendPong(ByteBuffer.allocate(  1)));
-            assertISE(ws.sendPong(ByteBuffer.allocate(  0)));
-
-            // ... add more CF complete exceptionally scenarios
+            assertISE(ws.sendPong(ByteBuffer.allocate(1)));
+            assertISE(ws.sendPong(ByteBuffer.allocate(0)));
         }
     }
 
-    private void assertIAE(CompletableFuture<?> stage) {
-        assertExceptionally(IllegalArgumentException.class, stage);
+    private static void assertIAE(CompletableFuture<?> stage) {
+        assertExceptionally(IAE, stage);
     }
 
-    private void assertISE(CompletableFuture<?> stage) {
+    private static void assertISE(CompletableFuture<?> stage) {
         assertExceptionally(IllegalStateException.class, stage);
     }
 
-    private void assertExceptionally(Class<? extends Throwable> clazz,
-                                     CompletableFuture<?> stage) {
+    private static void assertExceptionally(Class<? extends Throwable> clazz,
+                                            CompletableFuture<?> stage) {
         stage.handle((result, error) -> {
             if (error instanceof CompletionException) {
                 Throwable cause = error.getCause();
@@ -143,7 +314,8 @@
         }).join();
     }
 
-    private void assertException(Class<? extends Throwable> clazz, Throwable t) {
+    private static void assertException(Class<? extends Throwable> clazz,
+                                        Throwable t) {
         if (t == null) {
             throw new AssertionError("Expected " + clazz + ", caught nothing");
         }
@@ -151,6 +323,4 @@
             throw new AssertionError("Expected " + clazz + ", caught " + t);
         }
     }
-
-    // ... more API assertions ???
 }