test/jdk/java/net/httpclient/websocket/PendingBinaryPongClose.java
author chegar
Wed, 11 Apr 2018 14:21:11 +0100
branchhttp-client-branch
changeset 56416 db552808d149
parent 56409 b0c62cfc1d12
child 56451 9585061fdb04
permissions -rw-r--r--
http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
56325
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     1
/*
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     2
 * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     4
 *
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     7
 * published by the Free Software Foundation.
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     8
 *
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
     9
 * This code is distributed in the hope that it will be useful, but WITHOUT
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    10
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    11
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    12
 * version 2 for more details (a copy is included in the LICENSE file that
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    13
 * accompanied this code).
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    14
 *
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License version
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    16
 * 2 along with this work; if not, write to the Free Software Foundation,
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    17
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    18
 *
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    19
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    20
 * or visit www.oracle.com if you need additional information or have any
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    21
 * questions.
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    22
 */
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    23
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    24
/*
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    25
 * @test
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    26
 * @build DummyWebSocketServer
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    27
 * @run testng/othervm
56409
b0c62cfc1d12 http-client-branch: enable HTTP debugging in Pending[Text|Binary]* tests
chegar
parents: 56335
diff changeset
    28
 *      -Djdk.internal.httpclient.debug=true
56325
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    29
 *      -Djdk.internal.httpclient.websocket.debug=true
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    30
 *       PendingBinaryPongClose
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    31
 */
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    32
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    33
import org.testng.annotations.Test;
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    34
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    35
import java.net.http.WebSocket;
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    36
import java.nio.ByteBuffer;
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    37
import java.util.concurrent.CompletableFuture;
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    38
import java.util.concurrent.TimeUnit;
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    39
import java.util.concurrent.TimeoutException;
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    40
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    41
import static java.net.http.HttpClient.newHttpClient;
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    42
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    43
public class PendingBinaryPongClose extends PendingOperations {
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    44
56416
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    45
    CompletableFuture<WebSocket> cfBinary;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    46
    CompletableFuture<WebSocket> cfPong;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    47
    CompletableFuture<WebSocket> cfClose;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    48
56325
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    49
    @Test(dataProvider = "booleans")
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    50
    public void pendingBinaryPongClose(boolean last) throws Exception {
56416
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    51
        repeatable(() -> {
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    52
            server = Support.notReadingServer();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    53
            server.open();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    54
            webSocket = newHttpClient().newWebSocketBuilder()
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    55
                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    56
                    .join();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    57
            ByteBuffer data = ByteBuffer.allocate(65536);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    58
            for (int i = 0; ; i++) {  // fill up the send buffer
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    59
                long start = System.currentTimeMillis();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    60
                System.out.printf("begin cycle #%s at %s%n", i, start);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    61
                cfBinary = webSocket.sendBinary(data, last);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    62
                try {
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    63
                    cfBinary.get(MAX_WAIT_SEC, TimeUnit.SECONDS);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    64
                    data.clear();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    65
                } catch (TimeoutException e) {
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    66
                    break;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    67
                } finally {
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    68
                    long stop = System.currentTimeMillis();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    69
                    System.out.printf("end cycle #%s at %s (%s ms)%n", i, stop, stop - start);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    70
                }
56325
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    71
            }
56416
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    72
            assertFails(ISE, webSocket.sendText("", true));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    73
            assertFails(ISE, webSocket.sendText("", false));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    74
            assertFails(ISE, webSocket.sendBinary(ByteBuffer.allocate(0), true));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    75
            assertFails(ISE, webSocket.sendBinary(ByteBuffer.allocate(0), false));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    76
            cfPong = webSocket.sendPong(ByteBuffer.allocate(125));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    77
            assertHangs(cfPong);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    78
            assertFails(ISE, webSocket.sendPing(ByteBuffer.allocate(125)));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    79
            assertFails(ISE, webSocket.sendPong(ByteBuffer.allocate(125)));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    80
            cfClose = webSocket.sendClose(WebSocket.NORMAL_CLOSURE, "ok");
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    81
            assertHangs(cfClose);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    82
            return null;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    83
        }, () -> cfBinary.isDone() ? true : false);
56325
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    84
        webSocket.abort();
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    85
        assertFails(IOE, cfBinary);
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    86
        assertFails(IOE, cfPong);
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    87
        assertFails(IOE, cfClose);
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    88
    }
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    89
}