test/jdk/java/net/httpclient/websocket/PendingTextPongClose.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
 *       PendingTextPongClose
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.nio.CharBuffer;
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.CompletableFuture;
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.TimeUnit;
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    40
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
    41
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    42
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
    43
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    44
public class PendingTextPongClose extends PendingOperations {
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    45
56416
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    46
    CompletableFuture<WebSocket> cfText;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    47
    CompletableFuture<WebSocket> cfPong;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    48
    CompletableFuture<WebSocket> cfClose;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    49
56325
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    50
    @Test(dataProvider = "booleans")
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    51
    public void pendingTextPongClose(boolean last) throws Exception {
56416
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    52
        repeatable(() -> {
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    53
            server = Support.notReadingServer();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    54
            server.open();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    55
            webSocket = newHttpClient().newWebSocketBuilder()
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    56
                    .buildAsync(server.getURI(), new WebSocket.Listener() { })
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    57
                    .join();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    58
            CharBuffer data = CharBuffer.allocate(65536);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    59
            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
    60
                long start = System.currentTimeMillis();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    61
                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
    62
                cfText = webSocket.sendText(data, last);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    63
                try {
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    64
                    cfText.get(MAX_WAIT_SEC, TimeUnit.SECONDS);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    65
                    data.clear();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    66
                } catch (TimeoutException e) {
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    67
                    break;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    68
                } finally {
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    69
                    long stop = System.currentTimeMillis();
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    70
                    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
    71
                }
56325
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    72
            }
56416
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    73
            assertFails(ISE, webSocket.sendText("", true));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    74
            assertFails(ISE, webSocket.sendText("", false));
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), true));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    76
            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
    77
            cfPong = webSocket.sendPong(ByteBuffer.allocate(125));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    78
            assertHangs(cfPong);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    79
            assertFails(ISE, webSocket.sendPing(ByteBuffer.allocate(125)));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    80
            assertFails(ISE, webSocket.sendPong(ByteBuffer.allocate(125)));
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    81
            cfClose = webSocket.sendClose(WebSocket.NORMAL_CLOSURE, "ok");
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    82
            assertHangs(cfClose);
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    83
            return null;
db552808d149 http-client-branch: review comment - fix intermittently failing PendingXxx WebSocket tests
chegar
parents: 56409
diff changeset
    84
        }, () -> cfText.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
    85
        webSocket.abort();
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    86
        assertFails(IOE, cfText);
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    87
        assertFails(IOE, cfPong);
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    88
        assertFails(IOE, cfClose);
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    89
    }
195d2970d981 http-client-branch: (WebSocket) split pending operations tests, otherwise it took them too long to run
prappo
parents:
diff changeset
    90
}