test/jdk/java/net/httpclient/websocket/WebSocketTest.java
author pconcannon
Mon, 23 Sep 2019 16:53:16 +0100
changeset 58289 3a79d4cccbcb
parent 53521 41fa3e6f2785
permissions -rw-r--r--
8217825: Verify @AfterTest is used correctly in WebSocket tests Summary: Remove @AfterTest tags; added in explicit closing of resources e.g. httpServer, webSocket, etc Reviewed-by: dfuchs, prappo
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
     1
/*
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
     2
 * Copyright (c) 2018, 2019, Oracle and/or its affiliates. All rights reserved.
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
     4
 *
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
     7
 * published by the Free Software Foundation.
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
     8
 *
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
     9
 * This code is distributed in the hope that it will be useful, but WITHOUT
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    10
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    11
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    12
 * version 2 for more details (a copy is included in the LICENSE file that
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    13
 * accompanied this code).
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    14
 *
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License version
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    16
 * 2 along with this work; if not, write to the Free Software Foundation,
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    17
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    18
 *
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    19
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    20
 * or visit www.oracle.com if you need additional information or have any
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    21
 * questions.
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    22
 */
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    23
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    24
/*
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    25
 * @test
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    26
 * @bug 8217429
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    27
 * @build DummyWebSocketServer
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    28
 * @run testng/othervm
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    29
 *       WebSocketTest
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    30
 */
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    31
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
    32
import org.testng.annotations.DataProvider;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    33
import org.testng.annotations.Test;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    34
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    35
import java.io.IOException;
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    36
import java.net.Authenticator;
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    37
import java.net.PasswordAuthentication;
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    38
import java.net.http.HttpResponse;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    39
import java.net.http.WebSocket;
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    40
import java.net.http.WebSocketHandshakeException;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    41
import java.nio.ByteBuffer;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    42
import java.nio.charset.StandardCharsets;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    43
import java.util.ArrayList;
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    44
import java.util.Base64;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    45
import java.util.List;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    46
import java.util.concurrent.CompletableFuture;
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    47
import java.util.concurrent.CompletionException;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    48
import java.util.concurrent.CompletionStage;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    49
import java.util.concurrent.TimeUnit;
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
    50
import java.util.concurrent.atomic.AtomicBoolean;
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    51
import java.util.function.Function;
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
    52
import java.util.function.Supplier;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    53
import java.util.stream.Collectors;
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
    54
49944
4690a2871b44 8202423: Small HTTP Client refresh
chegar
parents: 49765
diff changeset
    55
import static java.net.http.HttpClient.Builder.NO_PROXY;
4690a2871b44 8202423: Small HTTP Client refresh
chegar
parents: 49765
diff changeset
    56
import static java.net.http.HttpClient.newBuilder;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    57
import static java.net.http.WebSocket.NORMAL_CLOSURE;
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    58
import static java.nio.charset.StandardCharsets.UTF_8;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    59
import static org.testng.Assert.assertEquals;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    60
import static org.testng.Assert.assertThrows;
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
    61
import static org.testng.Assert.fail;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    62
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    63
public class WebSocketTest {
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    64
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    65
    private static final Class<IllegalArgumentException> IAE = IllegalArgumentException.class;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    66
    private static final Class<IllegalStateException> ISE = IllegalStateException.class;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    67
    private static final Class<IOException> IOE = IOException.class;
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    68
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    69
    /* shortcut */
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    70
    private static void assertFails(Class<? extends Throwable> clazz,
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    71
                                    CompletionStage<?> stage) {
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    72
        Support.assertCompletesExceptionally(clazz, stage);
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    73
    }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    74
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    75
    @Test
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    76
    public void illegalArgument() throws IOException {
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    77
        try (var server = new DummyWebSocketServer()) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    78
            server.open();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    79
            var webSocket = newBuilder().proxy(NO_PROXY).build()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    80
                    .newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    81
                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    82
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    83
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    84
                assertFails(IAE, webSocket.sendPing(ByteBuffer.allocate(126)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    85
                assertFails(IAE, webSocket.sendPing(ByteBuffer.allocate(127)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    86
                assertFails(IAE, webSocket.sendPing(ByteBuffer.allocate(128)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    87
                assertFails(IAE, webSocket.sendPing(ByteBuffer.allocate(129)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    88
                assertFails(IAE, webSocket.sendPing(ByteBuffer.allocate(256)));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    89
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    90
                assertFails(IAE, webSocket.sendPong(ByteBuffer.allocate(126)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    91
                assertFails(IAE, webSocket.sendPong(ByteBuffer.allocate(127)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    92
                assertFails(IAE, webSocket.sendPong(ByteBuffer.allocate(128)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    93
                assertFails(IAE, webSocket.sendPong(ByteBuffer.allocate(129)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    94
                assertFails(IAE, webSocket.sendPong(ByteBuffer.allocate(256)));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
    95
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    96
                assertFails(IOE, webSocket.sendText(Support.incompleteString(), true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    97
                assertFails(IOE, webSocket.sendText(Support.incompleteString(), false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    98
                assertFails(IOE, webSocket.sendText(Support.malformedString(), true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
    99
                assertFails(IOE, webSocket.sendText(Support.malformedString(), false));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   100
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   101
                assertFails(IAE, webSocket.sendClose(NORMAL_CLOSURE, Support.stringWithNBytes(124)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   102
                assertFails(IAE, webSocket.sendClose(NORMAL_CLOSURE, Support.stringWithNBytes(125)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   103
                assertFails(IAE, webSocket.sendClose(NORMAL_CLOSURE, Support.stringWithNBytes(128)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   104
                assertFails(IAE, webSocket.sendClose(NORMAL_CLOSURE, Support.stringWithNBytes(256)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   105
                assertFails(IAE, webSocket.sendClose(NORMAL_CLOSURE, Support.stringWithNBytes(257)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   106
                assertFails(IAE, webSocket.sendClose(NORMAL_CLOSURE, Support.stringWith2NBytes((123 / 2) + 1)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   107
                assertFails(IAE, webSocket.sendClose(NORMAL_CLOSURE, Support.malformedString()));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   108
                assertFails(IAE, webSocket.sendClose(NORMAL_CLOSURE, Support.incompleteString()));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   109
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   110
                assertFails(IAE, webSocket.sendClose(-2, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   111
                assertFails(IAE, webSocket.sendClose(-1, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   112
                assertFails(IAE, webSocket.sendClose(0, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   113
                assertFails(IAE, webSocket.sendClose(1, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   114
                assertFails(IAE, webSocket.sendClose(500, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   115
                assertFails(IAE, webSocket.sendClose(998, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   116
                assertFails(IAE, webSocket.sendClose(999, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   117
                assertFails(IAE, webSocket.sendClose(1002, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   118
                assertFails(IAE, webSocket.sendClose(1003, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   119
                assertFails(IAE, webSocket.sendClose(1006, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   120
                assertFails(IAE, webSocket.sendClose(1007, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   121
                assertFails(IAE, webSocket.sendClose(1009, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   122
                assertFails(IAE, webSocket.sendClose(1010, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   123
                assertFails(IAE, webSocket.sendClose(1012, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   124
                assertFails(IAE, webSocket.sendClose(1013, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   125
                assertFails(IAE, webSocket.sendClose(1015, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   126
                assertFails(IAE, webSocket.sendClose(5000, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   127
                assertFails(IAE, webSocket.sendClose(32768, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   128
                assertFails(IAE, webSocket.sendClose(65535, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   129
                assertFails(IAE, webSocket.sendClose(65536, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   130
                assertFails(IAE, webSocket.sendClose(Integer.MAX_VALUE, "a reason"));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   131
                assertFails(IAE, webSocket.sendClose(Integer.MIN_VALUE, "a reason"));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   132
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   133
                assertThrows(IAE, () -> webSocket.request(Integer.MIN_VALUE));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   134
                assertThrows(IAE, () -> webSocket.request(Long.MIN_VALUE));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   135
                assertThrows(IAE, () -> webSocket.request(-1));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   136
                assertThrows(IAE, () -> webSocket.request(0));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   137
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   138
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   139
                webSocket.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   140
            }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   141
        }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   142
    }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   143
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   144
    @Test
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   145
    public void partialBinaryThenText() throws IOException {
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   146
        try (var server = new DummyWebSocketServer()) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   147
            server.open();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   148
            var webSocket = newBuilder().proxy(NO_PROXY).build().newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   149
                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   150
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   151
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   152
                webSocket.sendBinary(ByteBuffer.allocate(16), false).join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   153
                assertFails(ISE, webSocket.sendText("text", false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   154
                assertFails(ISE, webSocket.sendText("text", true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   155
                // Pings & Pongs are fine
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   156
                webSocket.sendPing(ByteBuffer.allocate(125)).join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   157
                webSocket.sendPong(ByteBuffer.allocate(125)).join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   158
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   159
                webSocket.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   160
            }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   161
        }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   162
    }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   163
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   164
    @Test
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   165
    public void partialTextThenBinary() throws IOException {
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   166
        try (var server = new DummyWebSocketServer()) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   167
            server.open();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   168
            var webSocket = newBuilder().proxy(NO_PROXY).build().newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   169
                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   170
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   171
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   172
                webSocket.sendText("text", false).join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   173
                assertFails(ISE, webSocket.sendBinary(ByteBuffer.allocate(16), false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   174
                assertFails(ISE, webSocket.sendBinary(ByteBuffer.allocate(16), true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   175
                // Pings & Pongs are fine
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   176
                webSocket.sendPing(ByteBuffer.allocate(125)).join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   177
                webSocket.sendPong(ByteBuffer.allocate(125)).join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   178
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   179
                webSocket.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   180
            }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   181
        }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   182
    }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   183
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   184
    @Test
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   185
    public void sendMethodsThrowIOE1() throws IOException {
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   186
        try (var server = new DummyWebSocketServer()) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   187
            server.open();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   188
            var webSocket = newBuilder().proxy(NO_PROXY).build()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   189
                    .newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   190
                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   191
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   192
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   193
                webSocket.sendClose(NORMAL_CLOSURE, "ok").join();
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   194
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   195
                assertFails(IOE, webSocket.sendClose(WebSocket.NORMAL_CLOSURE, "ok"));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   196
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   197
                assertFails(IOE, webSocket.sendText("", true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   198
                assertFails(IOE, webSocket.sendText("", false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   199
                assertFails(IOE, webSocket.sendText("abc", true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   200
                assertFails(IOE, webSocket.sendText("abc", false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   201
                assertFails(IOE, webSocket.sendBinary(ByteBuffer.allocate(0), true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   202
                assertFails(IOE, webSocket.sendBinary(ByteBuffer.allocate(0), false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   203
                assertFails(IOE, webSocket.sendBinary(ByteBuffer.allocate(1), true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   204
                assertFails(IOE, webSocket.sendBinary(ByteBuffer.allocate(1), false));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   205
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   206
                assertFails(IOE, webSocket.sendPing(ByteBuffer.allocate(125)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   207
                assertFails(IOE, webSocket.sendPing(ByteBuffer.allocate(124)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   208
                assertFails(IOE, webSocket.sendPing(ByteBuffer.allocate(1)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   209
                assertFails(IOE, webSocket.sendPing(ByteBuffer.allocate(0)));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   210
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   211
                assertFails(IOE, webSocket.sendPong(ByteBuffer.allocate(125)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   212
                assertFails(IOE, webSocket.sendPong(ByteBuffer.allocate(124)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   213
                assertFails(IOE, webSocket.sendPong(ByteBuffer.allocate(1)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   214
                assertFails(IOE, webSocket.sendPong(ByteBuffer.allocate(0)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   215
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   216
                webSocket.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   217
            }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   218
        }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   219
    }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   220
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   221
    @DataProvider(name = "sequence")
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   222
    public Object[][] data1() {
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   223
        int[] CLOSE = {
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   224
                0x81, 0x00, // ""
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   225
                0x82, 0x00, // []
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   226
                0x89, 0x00, // <PING>
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   227
                0x8a, 0x00, // <PONG>
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   228
                0x88, 0x00, // <CLOSE>
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   229
        };
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   230
        int[] ERROR = {
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   231
                0x81, 0x00, // ""
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   232
                0x82, 0x00, // []
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   233
                0x89, 0x00, // <PING>
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   234
                0x8a, 0x00, // <PONG>
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   235
                0x8b, 0x00, // 0xB control frame (causes an error)
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   236
        };
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   237
        return new Object[][]{
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   238
                {CLOSE, 1},
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   239
                {CLOSE, 3},
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   240
                {CLOSE, 4},
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   241
                {CLOSE, Long.MAX_VALUE},
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   242
                {ERROR, 1},
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   243
                {ERROR, 3},
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   244
                {ERROR, 4},
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   245
                {ERROR, Long.MAX_VALUE},
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   246
        };
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   247
    }
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   248
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   249
    @Test(dataProvider = "sequence")
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   250
    public void listenerSequentialOrder(int[] binary, long requestSize)
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   251
            throws IOException
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   252
    {
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   253
        try (var server = Support.serverWithCannedData(binary)) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   254
            server.open();
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   255
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   256
            CompletableFuture<Void> violation = new CompletableFuture<>();
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   257
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   258
            MockListener listener = new MockListener(requestSize) {
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   259
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   260
                final AtomicBoolean guard = new AtomicBoolean();
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   261
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   262
                private <T> T checkRunExclusively(Supplier<T> action) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   263
                    if (guard.getAndSet(true)) {
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   264
                        violation.completeExceptionally(new RuntimeException());
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   265
                    }
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   266
                    try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   267
                        return action.get();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   268
                    } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   269
                        if (!guard.getAndSet(false)) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   270
                            violation.completeExceptionally(new RuntimeException());
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   271
                        }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   272
                    }
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   273
                }
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   274
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   275
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   276
                public void onOpen(WebSocket webSocket) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   277
                    checkRunExclusively(() -> {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   278
                        super.onOpen(webSocket);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   279
                        return null;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   280
                    });
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   281
                }
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   282
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   283
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   284
                public CompletionStage<?> onText(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   285
                                                 CharSequence data,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   286
                                                 boolean last) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   287
                    return checkRunExclusively(
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   288
                            () -> super.onText(webSocket, data, last));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   289
                }
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   290
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   291
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   292
                public CompletionStage<?> onBinary(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   293
                                                   ByteBuffer data,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   294
                                                   boolean last) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   295
                    return checkRunExclusively(
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   296
                            () -> super.onBinary(webSocket, data, last));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   297
                }
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   298
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   299
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   300
                public CompletionStage<?> onPing(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   301
                                                 ByteBuffer message) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   302
                    return checkRunExclusively(
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   303
                            () -> super.onPing(webSocket, message));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   304
                }
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   305
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   306
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   307
                public CompletionStage<?> onPong(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   308
                                                 ByteBuffer message) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   309
                    return checkRunExclusively(
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   310
                            () -> super.onPong(webSocket, message));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   311
                }
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   312
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   313
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   314
                public CompletionStage<?> onClose(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   315
                                                  int statusCode,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   316
                                                  String reason) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   317
                    return checkRunExclusively(
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   318
                            () -> super.onClose(webSocket, statusCode, reason));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   319
                }
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   320
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   321
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   322
                public void onError(WebSocket webSocket, Throwable error) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   323
                    checkRunExclusively(() -> {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   324
                        super.onError(webSocket, error);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   325
                        return null;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   326
                    });
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   327
                }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   328
            };
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   329
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   330
            var webSocket = newBuilder().proxy(NO_PROXY).build().newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   331
                    .buildAsync(server.getURI(), listener)
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   332
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   333
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   334
                listener.invocations();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   335
                violation.complete(null); // won't affect if completed exceptionally
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   336
                violation.join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   337
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   338
                webSocket.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   339
            }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   340
        }
50681
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   341
    }
4254bed3c09d 8204679: HTTP Client refresh
chegar
parents: 49944
diff changeset
   342
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   343
    @Test
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   344
    public void sendMethodsThrowIOE2() throws Exception {
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   345
        try (var server = Support.serverWithCannedData(0x88, 0x00)) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   346
            server.open();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   347
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   348
            CompletableFuture<Void> onCloseCalled = new CompletableFuture<>();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   349
            CompletableFuture<Void> canClose = new CompletableFuture<>();
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   350
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   351
            WebSocket.Listener listener = new WebSocket.Listener() {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   352
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   353
                public CompletionStage<?> onClose(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   354
                                                  int statusCode,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   355
                                                  String reason) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   356
                    System.out.printf("onClose(%s, '%s')%n", statusCode, reason);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   357
                    onCloseCalled.complete(null);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   358
                    return canClose;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   359
                }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   360
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   361
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   362
                public void onError(WebSocket webSocket, Throwable error) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   363
                    System.out.println("onError(" + error + ")");
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   364
                    onCloseCalled.completeExceptionally(error);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   365
                }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   366
            };
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   367
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   368
            var webSocket = newBuilder().proxy(NO_PROXY).build().newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   369
                    .buildAsync(server.getURI(), listener)
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   370
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   371
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   372
                onCloseCalled.join();      // Wait for onClose to be called
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   373
                canClose.complete(null);   // Signal to the WebSocket it can close the output
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   374
                TimeUnit.SECONDS.sleep(5); // Give canClose some time to reach the WebSocket
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   375
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   376
                assertFails(IOE, webSocket.sendClose(WebSocket.NORMAL_CLOSURE, "ok"));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   377
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   378
                assertFails(IOE, webSocket.sendText("", true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   379
                assertFails(IOE, webSocket.sendText("", false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   380
                assertFails(IOE, webSocket.sendText("abc", true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   381
                assertFails(IOE, webSocket.sendText("abc", false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   382
                assertFails(IOE, webSocket.sendBinary(ByteBuffer.allocate(0), true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   383
                assertFails(IOE, webSocket.sendBinary(ByteBuffer.allocate(0), false));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   384
                assertFails(IOE, webSocket.sendBinary(ByteBuffer.allocate(1), true));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   385
                assertFails(IOE, webSocket.sendBinary(ByteBuffer.allocate(1), false));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   386
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   387
                assertFails(IOE, webSocket.sendPing(ByteBuffer.allocate(125)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   388
                assertFails(IOE, webSocket.sendPing(ByteBuffer.allocate(124)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   389
                assertFails(IOE, webSocket.sendPing(ByteBuffer.allocate(1)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   390
                assertFails(IOE, webSocket.sendPing(ByteBuffer.allocate(0)));
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   391
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   392
                assertFails(IOE, webSocket.sendPong(ByteBuffer.allocate(125)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   393
                assertFails(IOE, webSocket.sendPong(ByteBuffer.allocate(124)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   394
                assertFails(IOE, webSocket.sendPong(ByteBuffer.allocate(1)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   395
                assertFails(IOE, webSocket.sendPong(ByteBuffer.allocate(0)));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   396
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   397
                webSocket.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   398
            }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   399
        }
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   400
    }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   401
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   402
    // Used to verify a server requiring Authentication
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   403
    private static final String USERNAME = "chegar";
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   404
    private static final String PASSWORD = "a1b2c3";
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   405
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   406
    static class WSAuthenticator extends Authenticator {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   407
        @Override
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   408
        protected PasswordAuthentication getPasswordAuthentication() {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   409
            return new PasswordAuthentication(USERNAME, PASSWORD.toCharArray());
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   410
        }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   411
    }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   412
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   413
    static final Function<int[],DummyWebSocketServer> SERVER_WITH_CANNED_DATA =
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   414
        new Function<>() {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   415
            @Override public DummyWebSocketServer apply(int[] data) {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   416
                return Support.serverWithCannedData(data); }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   417
            @Override public String toString() { return "SERVER_WITH_CANNED_DATA"; }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   418
        };
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   419
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   420
    static final Function<int[],DummyWebSocketServer> AUTH_SERVER_WITH_CANNED_DATA =
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   421
        new Function<>() {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   422
            @Override public DummyWebSocketServer apply(int[] data) {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   423
                return Support.serverWithCannedDataAndAuthentication(USERNAME, PASSWORD, data); }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   424
            @Override public String toString() { return "AUTH_SERVER_WITH_CANNED_DATA"; }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   425
        };
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   426
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   427
    @DataProvider(name = "servers")
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   428
    public Object[][] servers() {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   429
        return new Object[][] {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   430
            { SERVER_WITH_CANNED_DATA },
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   431
            { AUTH_SERVER_WITH_CANNED_DATA },
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   432
        };
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   433
    }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   434
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   435
    @Test(dataProvider = "servers")
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   436
    public void simpleAggregatingBinaryMessages
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   437
            (Function<int[],DummyWebSocketServer> serverSupplier)
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   438
        throws IOException
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   439
    {
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   440
        List<byte[]> expected = List.of("alpha", "beta", "gamma", "delta")
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   441
                .stream()
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   442
                .map(s -> s.getBytes(StandardCharsets.US_ASCII))
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   443
                .collect(Collectors.toList());
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   444
        int[] binary = new int[]{
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   445
                0x82, 0x05, 0x61, 0x6c, 0x70, 0x68, 0x61, // [alpha]
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   446
                0x02, 0x02, 0x62, 0x65,                   // [be
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   447
                0x80, 0x02, 0x74, 0x61,                   // ta]
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   448
                0x02, 0x01, 0x67,                         // [g
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   449
                0x00, 0x01, 0x61,                         // a
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   450
                0x00, 0x00,                               //
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   451
                0x00, 0x00,                               //
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   452
                0x00, 0x01, 0x6d,                         // m
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   453
                0x00, 0x01, 0x6d,                         // m
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   454
                0x80, 0x01, 0x61,                         // a]
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   455
                0x8a, 0x00,                               // <PONG>
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   456
                0x02, 0x04, 0x64, 0x65, 0x6c, 0x74,       // [delt
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   457
                0x00, 0x01, 0x61,                         // a
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   458
                0x80, 0x00,                               // ]
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   459
                0x88, 0x00                                // <CLOSE>
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   460
        };
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   461
        CompletableFuture<List<byte[]>> actual = new CompletableFuture<>();
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   462
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   463
        try (var server = serverSupplier.apply(binary)) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   464
            server.open();
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   465
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   466
            WebSocket.Listener listener = new WebSocket.Listener() {
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   467
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   468
                List<byte[]> collectedBytes = new ArrayList<>();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   469
                ByteBuffer buffer = ByteBuffer.allocate(1024);
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   470
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   471
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   472
                public CompletionStage<?> onBinary(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   473
                                                   ByteBuffer message,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   474
                                                   boolean last) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   475
                    System.out.printf("onBinary(%s, %s)%n", message, last);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   476
                    webSocket.request(1);
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   477
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   478
                    append(message);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   479
                    if (last) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   480
                        buffer.flip();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   481
                        byte[] bytes = new byte[buffer.remaining()];
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   482
                        buffer.get(bytes);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   483
                        buffer.clear();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   484
                        processWholeBinary(bytes);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   485
                    }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   486
                    return null;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   487
                }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   488
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   489
                private void append(ByteBuffer message) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   490
                    if (buffer.remaining() < message.remaining()) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   491
                        assert message.remaining() > 0;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   492
                        int cap = (buffer.capacity() + message.remaining()) * 2;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   493
                        ByteBuffer b = ByteBuffer.allocate(cap);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   494
                        b.put(buffer.flip());
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   495
                        buffer = b;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   496
                    }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   497
                    buffer.put(message);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   498
                }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   499
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   500
                private void processWholeBinary(byte[] bytes) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   501
                    String stringBytes = new String(bytes, UTF_8);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   502
                    System.out.println("processWholeBinary: " + stringBytes);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   503
                    collectedBytes.add(bytes);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   504
                }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   505
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   506
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   507
                public CompletionStage<?> onClose(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   508
                                                  int statusCode,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   509
                                                  String reason) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   510
                    actual.complete(collectedBytes);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   511
                    return null;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   512
                }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   513
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   514
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   515
                public void onError(WebSocket webSocket, Throwable error) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   516
                    actual.completeExceptionally(error);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   517
                }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   518
            };
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   519
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   520
            var webSocket = newBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   521
                    .proxy(NO_PROXY)
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   522
                    .authenticator(new WSAuthenticator())
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   523
                    .build().newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   524
                    .buildAsync(server.getURI(), listener)
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   525
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   526
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   527
                List<byte[]> a = actual.join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   528
                assertEquals(a, expected);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   529
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   530
                webSocket.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   531
            }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   532
        }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   533
    }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   534
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   535
    @Test(dataProvider = "servers")
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   536
    public void simpleAggregatingTextMessages
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   537
            (Function<int[],DummyWebSocketServer> serverSupplier)
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   538
        throws IOException
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   539
    {
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   540
        List<String> expected = List.of("alpha", "beta", "gamma", "delta");
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   541
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   542
        int[] binary = new int[]{
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   543
                0x81, 0x05, 0x61, 0x6c, 0x70, 0x68, 0x61, // "alpha"
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   544
                0x01, 0x02, 0x62, 0x65,                   // "be
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   545
                0x80, 0x02, 0x74, 0x61,                   // ta"
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   546
                0x01, 0x01, 0x67,                         // "g
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   547
                0x00, 0x01, 0x61,                         // a
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   548
                0x00, 0x00,                               //
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   549
                0x00, 0x00,                               //
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   550
                0x00, 0x01, 0x6d,                         // m
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   551
                0x00, 0x01, 0x6d,                         // m
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   552
                0x80, 0x01, 0x61,                         // a"
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   553
                0x8a, 0x00,                               // <PONG>
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   554
                0x01, 0x04, 0x64, 0x65, 0x6c, 0x74,       // "delt
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   555
                0x00, 0x01, 0x61,                         // a
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   556
                0x80, 0x00,                               // "
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   557
                0x88, 0x00                                // <CLOSE>
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   558
        };
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   559
        CompletableFuture<List<String>> actual = new CompletableFuture<>();
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   560
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   561
        try (var server = serverSupplier.apply(binary)) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   562
            server.open();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   563
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   564
            WebSocket.Listener listener = new WebSocket.Listener() {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   565
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   566
                List<String> collectedStrings = new ArrayList<>();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   567
                StringBuilder text = new StringBuilder();
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   568
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   569
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   570
                public CompletionStage<?> onText(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   571
                                                 CharSequence message,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   572
                                                 boolean last) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   573
                    System.out.printf("onText(%s, %s)%n", message, last);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   574
                    webSocket.request(1);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   575
                    text.append(message);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   576
                    if (last) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   577
                        String str = text.toString();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   578
                        text.setLength(0);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   579
                        processWholeText(str);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   580
                    }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   581
                    return null;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   582
                }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   583
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   584
                private void processWholeText(String string) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   585
                    System.out.println(string);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   586
                    collectedStrings.add(string);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   587
                }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   588
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   589
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   590
                public CompletionStage<?> onClose(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   591
                                                  int statusCode,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   592
                                                  String reason) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   593
                    actual.complete(collectedStrings);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   594
                    return null;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   595
                }
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   596
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   597
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   598
                public void onError(WebSocket webSocket, Throwable error) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   599
                    actual.completeExceptionally(error);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   600
                }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   601
            };
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   602
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   603
            var webSocket = newBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   604
                    .proxy(NO_PROXY)
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   605
                    .authenticator(new WSAuthenticator())
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   606
                    .build().newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   607
                    .buildAsync(server.getURI(), listener)
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   608
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   609
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   610
                List<String> a = actual.join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   611
                assertEquals(a, expected);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   612
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   613
                webSocket.abort();
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   614
            }
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   615
        }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   616
    }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   617
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   618
    /*
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   619
     * Exercises the scenario where requests for more messages are made prior to
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   620
     * completing the returned CompletionStage instances.
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   621
     */
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   622
    @Test(dataProvider = "servers")
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   623
    public void aggregatingTextMessages
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   624
        (Function<int[],DummyWebSocketServer> serverSupplier)
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   625
        throws IOException
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   626
    {
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   627
        List<String> expected = List.of("alpha", "beta", "gamma", "delta");
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   628
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   629
        int[] binary = new int[]{
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   630
                0x81, 0x05, 0x61, 0x6c, 0x70, 0x68, 0x61, // "alpha"
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   631
                0x01, 0x02, 0x62, 0x65,                   // "be
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   632
                0x80, 0x02, 0x74, 0x61,                   // ta"
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   633
                0x01, 0x01, 0x67,                         // "g
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   634
                0x00, 0x01, 0x61,                         // a
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   635
                0x00, 0x00,                               //
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   636
                0x00, 0x00,                               //
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   637
                0x00, 0x01, 0x6d,                         // m
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   638
                0x00, 0x01, 0x6d,                         // m
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   639
                0x80, 0x01, 0x61,                         // a"
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   640
                0x8a, 0x00,                               // <PONG>
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   641
                0x01, 0x04, 0x64, 0x65, 0x6c, 0x74,       // "delt
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   642
                0x00, 0x01, 0x61,                         // a
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   643
                0x80, 0x00,                               // "
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   644
                0x88, 0x00                                // <CLOSE>
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   645
        };
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   646
        CompletableFuture<List<String>> actual = new CompletableFuture<>();
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   647
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   648
        try (var server = serverSupplier.apply(binary)) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   649
            server.open();
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   650
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   651
            WebSocket.Listener listener = new WebSocket.Listener() {
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   652
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   653
                List<CharSequence> parts = new ArrayList<>();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   654
                /*
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   655
                 * A CompletableFuture which will complete once the current
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   656
                 * message has been fully assembled. Until then the listener
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   657
                 * returns this instance for every call.
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   658
                 */
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   659
                CompletableFuture<?> currentCf = new CompletableFuture<>();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   660
                List<String> collected = new ArrayList<>();
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   661
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   662
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   663
                public CompletionStage<?> onText(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   664
                                                 CharSequence message,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   665
                                                 boolean last) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   666
                    parts.add(message);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   667
                    if (!last) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   668
                        webSocket.request(1);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   669
                    } else {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   670
                        this.currentCf.thenRun(() -> webSocket.request(1));
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   671
                        CompletableFuture<?> refCf = this.currentCf;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   672
                        processWholeMessage(new ArrayList<>(parts), refCf);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   673
                        currentCf = new CompletableFuture<>();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   674
                        parts.clear();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   675
                        return refCf;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   676
                    }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   677
                    return currentCf;
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   678
                }
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   679
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   680
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   681
                public CompletionStage<?> onClose(WebSocket webSocket,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   682
                                                  int statusCode,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   683
                                                  String reason) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   684
                    actual.complete(collected);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   685
                    return null;
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   686
                }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   687
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   688
                @Override
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   689
                public void onError(WebSocket webSocket, Throwable error) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   690
                    actual.completeExceptionally(error);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   691
                }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   692
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   693
                public void processWholeMessage(List<CharSequence> data,
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   694
                                                CompletableFuture<?> cf) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   695
                    StringBuilder b = new StringBuilder();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   696
                    data.forEach(b::append);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   697
                    String s = b.toString();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   698
                    System.out.println(s);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   699
                    cf.complete(null);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   700
                    collected.add(s);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   701
                }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   702
            };
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   703
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   704
            var webSocket = newBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   705
                    .proxy(NO_PROXY)
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   706
                    .authenticator(new WSAuthenticator())
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   707
                    .build().newWebSocketBuilder()
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   708
                    .buildAsync(server.getURI(), listener)
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   709
                    .join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   710
            try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   711
                List<String> a = actual.join();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   712
                assertEquals(a, expected);
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   713
            } finally {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   714
                webSocket.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   715
            }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   716
        }
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   717
    }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   718
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   719
    // -- authentication specific tests
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   720
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   721
    /*
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   722
     * Ensures authentication succeeds when an Authenticator set on client builder.
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   723
     */
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   724
    @Test
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   725
    public void clientAuthenticate() throws IOException  {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   726
        try (var server = new DummyWebSocketServer(USERNAME, PASSWORD)){
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   727
            server.open();
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   728
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   729
            var webSocket = newBuilder()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   730
                    .proxy(NO_PROXY)
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   731
                    .authenticator(new WSAuthenticator())
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   732
                    .build()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   733
                    .newWebSocketBuilder()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   734
                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   735
                    .join();
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   736
            webSocket.abort();
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   737
        }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   738
    }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   739
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   740
    /*
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   741
     * Ensures authentication succeeds when an `Authorization` header is explicitly set.
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   742
     */
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   743
    @Test
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   744
    public void explicitAuthenticate() throws IOException  {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   745
        try (var server = new DummyWebSocketServer(USERNAME, PASSWORD)) {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   746
            server.open();
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   747
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   748
            String hv = "Basic " + Base64.getEncoder().encodeToString(
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   749
                    (USERNAME + ":" + PASSWORD).getBytes(UTF_8));
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   750
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   751
            var webSocket = newBuilder()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   752
                    .proxy(NO_PROXY).build()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   753
                    .newWebSocketBuilder()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   754
                    .header("Authorization", hv)
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   755
                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   756
                    .join();
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   757
            webSocket.abort();
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   758
        }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   759
    }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   760
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   761
    /*
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   762
     * Ensures authentication does not succeed when no authenticator is present.
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   763
     */
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   764
    @Test
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   765
    public void failNoAuthenticator() throws IOException  {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   766
        try (var server = new DummyWebSocketServer(USERNAME, PASSWORD)) {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   767
            server.open();
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   768
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   769
            CompletableFuture<WebSocket> cf = newBuilder()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   770
                    .proxy(NO_PROXY).build()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   771
                    .newWebSocketBuilder()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   772
                    .buildAsync(server.getURI(), new WebSocket.Listener() { });
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   773
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   774
            try {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   775
                var webSocket = cf.join();
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   776
                silentAbort(webSocket);
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   777
                fail("Expected exception not thrown");
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   778
            } catch (CompletionException expected) {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   779
                WebSocketHandshakeException e = (WebSocketHandshakeException)expected.getCause();
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   780
                HttpResponse<?> response = e.getResponse();
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   781
                assertEquals(response.statusCode(), 401);
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   782
            }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   783
        }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   784
    }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   785
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   786
    /*
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   787
     * Ensures authentication does not succeed when the authenticator presents
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   788
     * unauthorized credentials.
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   789
     */
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   790
    @Test
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   791
    public void failBadCredentials() throws IOException  {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   792
        try (var server = new DummyWebSocketServer(USERNAME, PASSWORD)) {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   793
            server.open();
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   794
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   795
            Authenticator authenticator = new Authenticator() {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   796
                @Override protected PasswordAuthentication getPasswordAuthentication() {
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   797
                    return new PasswordAuthentication("BAD" + USERNAME, "".toCharArray());
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   798
                }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   799
            };
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   800
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   801
            CompletableFuture<WebSocket> cf = newBuilder()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   802
                    .proxy(NO_PROXY)
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   803
                    .authenticator(authenticator)
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   804
                    .build()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   805
                    .newWebSocketBuilder()
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   806
                    .buildAsync(server.getURI(), new WebSocket.Listener() { });
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   807
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   808
            try {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   809
                var webSocket = cf.join();
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   810
                silentAbort(webSocket);
53521
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   811
                fail("Expected exception not thrown");
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   812
            } catch (CompletionException expected) {
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   813
                System.out.println("caught expected exception:" + expected);
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   814
            }
41fa3e6f2785 8217429: WebSocket over authenticating proxy fails to send Upgrade headers
chegar
parents: 50681
diff changeset
   815
        }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   816
    }
58289
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   817
    private static void silentAbort(WebSocket ws) {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   818
        try {
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   819
            ws.abort();
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   820
        } catch (Throwable t) { }
3a79d4cccbcb 8217825: Verify @AfterTest is used correctly in WebSocket tests
pconcannon
parents: 53521
diff changeset
   821
    }
49765
ee6f7a61f3a5 8197564: HTTP Client implementation
chegar
parents:
diff changeset
   822
}