Examples with IBlockheadClient org.eclipse.jetty.websocket.common.test.IBlockheadClient used on opensource projects

Example 21 with IBlockheadClient

use of org.eclipse.jetty.websocket.common.test.IBlockheadClient in project jetty.project by eclipse.

the class WebSocketCloseTest method testFastClose.

/**
     * Test fast close (bug #403817)
     * 
     * @throws Exception
     *             on test failure
     */
@Test
public void testFastClose() throws Exception {
    try (IBlockheadClient client = new BlockheadClient(server.getServerUri())) {
        client.setProtocols("fastclose");
        client.setTimeout(5, TimeUnit.SECONDS);
        client.connect();
        client.sendStandardRequest();
        client.expectUpgradeResponse();
        // Verify that client got close frame
        EventQueue<WebSocketFrame> frames = client.readFrames(1, 5, TimeUnit.SECONDS);
        WebSocketFrame frame = frames.poll();
        assertThat("frames[0].opcode", frame.getOpCode(), is(OpCode.CLOSE));
        CloseInfo close = new CloseInfo(frame);
        assertThat("Close Status Code", close.getStatusCode(), is(StatusCode.NORMAL));
        // Notify server of close handshake
        // respond with close
        client.write(close.asFrame());
        // ensure server socket got close event
        assertThat("Fast Close Latch", closeSocket.closeLatch.await(5, TimeUnit.SECONDS), is(true));
        assertThat("Fast Close.statusCode", closeSocket.closeStatusCode, is(StatusCode.NORMAL));
    }
}

Example 22 with IBlockheadClient

use of org.eclipse.jetty.websocket.common.test.IBlockheadClient in project jetty.project by eclipse.

the class ManyConnectionsCleanupTest method testOpenSessionCleanup.

/**
     * Test session open session cleanup (bug #474936)
     * 
     * @throws Exception
     *             on test failure
     */
@Test
public void testOpenSessionCleanup() throws Exception {
    int iterationCount = 100;
    StdErrLog.getLogger(FastFailSocket.class).setLevel(StdErrLog.LEVEL_OFF);
    StdErrLog sessLog = StdErrLog.getLogger(WebSocketSession.class);
    int oldLevel = sessLog.getLevel();
    sessLog.setLevel(StdErrLog.LEVEL_OFF);
    for (int requests = 0; requests < iterationCount; requests++) {
        fastFail();
        fastClose();
        dropConnection();
    }
    sessLog.setLevel(oldLevel);
    try (IBlockheadClient client = new BlockheadClient(server.getServerUri())) {
        client.setProtocols("container");
        client.setTimeout(1, TimeUnit.SECONDS);
        client.connect();
        client.sendStandardRequest();
        client.expectUpgradeResponse();
        client.write(new TextFrame().setPayload("calls"));
        client.write(new TextFrame().setPayload("openSessions"));
        EventQueue<WebSocketFrame> frames = client.readFrames(3, 6, TimeUnit.SECONDS);
        WebSocketFrame frame;
        String resp;
        frame = frames.poll();
        assertThat("frames[0].opcode", frame.getOpCode(), is(OpCode.TEXT));
        resp = frame.getPayloadAsUTF8();
        assertThat("Should only have 1 open session", resp, containsString("calls=" + ((iterationCount * 2) + 1)));
        frame = frames.poll();
        assertThat("frames[1].opcode", frame.getOpCode(), is(OpCode.TEXT));
        resp = frame.getPayloadAsUTF8();
        assertThat("Should only have 1 open session", resp, containsString("openSessions.size=1\n"));
        frame = frames.poll();
        assertThat("frames[2].opcode", frame.getOpCode(), is(OpCode.CLOSE));
        CloseInfo close = new CloseInfo(frame);
        assertThat("Close Status Code", close.getStatusCode(), is(StatusCode.NORMAL));
        // respond with close
        client.write(close.asFrame());
        // ensure server socket got close event
        assertThat("Open Sessions Latch", closeSocket.closeLatch.await(1, TimeUnit.SECONDS), is(true));
        assertThat("Open Sessions.statusCode", closeSocket.closeStatusCode, is(StatusCode.NORMAL));
        assertThat("Open Sessions.errors", closeSocket.errors.size(), is(0));
    }
}

Example 23 with IBlockheadClient

use of org.eclipse.jetty.websocket.common.test.IBlockheadClient in project jetty.project by eclipse.

the class ManyConnectionsCleanupTest method dropConnection.

private void dropConnection() throws Exception {
    try (IBlockheadClient client = new BlockheadClient(server.getServerUri())) {
        client.setProtocols("container");
        client.setTimeout(1, TimeUnit.SECONDS);
        try (StacklessLogging scope = new StacklessLogging(WebSocketSession.class)) {
            client.connect();
            client.sendStandardRequest();
            client.expectUpgradeResponse();
            client.disconnect();
        }
    }
}

Example 24 with IBlockheadClient

use of org.eclipse.jetty.websocket.common.test.IBlockheadClient in project jetty.project by eclipse.

the class FirefoxTest method testConnectionKeepAlive.

@Test
public void testConnectionKeepAlive() throws Exception {
    try (IBlockheadClient client = new BlockheadClient(server.getServerUri())) {
        // Odd Connection Header value seen in Firefox
        client.setConnectionValue("keep-alive, Upgrade");
        client.connect();
        client.sendStandardRequest();
        client.expectUpgradeResponse();
        // Generate text frame
        String msg = "this is an echo ... cho ... ho ... o";
        client.write(new TextFrame().setPayload(msg));
        // Read frame (hopefully text frame)
        EventQueue<WebSocketFrame> frames = client.readFrames(1, 30, TimeUnit.SECONDS);
        WebSocketFrame tf = frames.poll();
        Assert.assertThat("Text Frame.status code", tf.getPayloadAsUTF8(), is(msg));
    }
}