Examples with Session com.developmentontheedge.be5.api.Session used on opensource projects

Example 36 with Session

use of com.developmentontheedge.be5.api.Session in project jetty.project by eclipse.

the class ConnectTimeoutTest method testConnectTimeout.

@Test
public void testConnectTimeout() throws Exception {
    final String host = "10.255.255.1";
    final int port = 80;
    int connectTimeout = 1000;
    assumeConnectTimeout(host, port, connectTimeout);
    start(new ServerSessionListener.Adapter());
    client.setConnectTimeout(connectTimeout);
    InetSocketAddress address = new InetSocketAddress(host, port);
    final CountDownLatch latch = new CountDownLatch(1);
    client.connect(address, new Session.Listener.Adapter(), new Promise.Adapter<Session>() {

        @Override
        public void failed(Throwable x) {
            Assert.assertTrue(x instanceof SocketTimeoutException);
            latch.countDown();
        }
    });
    Assert.assertTrue(latch.await(2 * connectTimeout, TimeUnit.MILLISECONDS));
}

Example 37 with Session

use of com.developmentontheedge.be5.api.Session in project jetty.project by eclipse.

the class FlowControlStalledTest method newClient.

protected Session newClient(Session.Listener listener) throws Exception {
    String host = "localhost";
    int port = connector.getLocalPort();
    InetSocketAddress address = new InetSocketAddress(host, port);
    FuturePromise<Session> promise = new FuturePromise<>();
    client.connect(address, listener, promise);
    return promise.get(5, TimeUnit.SECONDS);
}

Example 38 with Session

use of com.developmentontheedge.be5.api.Session in project jetty.project by eclipse.

the class StreamCloseTest method testPushedStreamIsClosed.

@Test
public void testPushedStreamIsClosed() throws Exception {
    final CountDownLatch serverLatch = new CountDownLatch(1);
    start(new ServerSessionListener.Adapter() {

        @Override
        public Stream.Listener onNewStream(Stream stream, HeadersFrame frame) {
            PushPromiseFrame pushFrame = new PushPromiseFrame(stream.getId(), 0, newRequest("GET", new HttpFields()));
            stream.push(pushFrame, new Promise.Adapter<Stream>() {

                @Override
                public void succeeded(final Stream pushedStream) {
                    // When created, pushed stream must be implicitly remotely closed.
                    Assert.assertTrue(((HTTP2Stream) pushedStream).isRemotelyClosed());
                    // Send some data with endStream = true.
                    pushedStream.data(new DataFrame(pushedStream.getId(), ByteBuffer.allocate(16), true), new Callback() {

                        @Override
                        public void succeeded() {
                            Assert.assertTrue(pushedStream.isClosed());
                            serverLatch.countDown();
                        }
                    });
                }
            }, new Stream.Listener.Adapter());
            HeadersFrame response = new HeadersFrame(stream.getId(), new MetaData.Response(HttpVersion.HTTP_2, 200, new HttpFields()), null, true);
            stream.headers(response, Callback.NOOP);
            return null;
        }
    });
    Session session = newClient(new Session.Listener.Adapter());
    HeadersFrame frame = new HeadersFrame(newRequest("GET", new HttpFields()), null, true);
    final CountDownLatch clientLatch = new CountDownLatch(1);
    session.newStream(frame, new Promise.Adapter<>(), new Stream.Listener.Adapter() {

        @Override
        public Stream.Listener onPush(Stream pushedStream, PushPromiseFrame frame) {
            Assert.assertTrue(((HTTP2Stream) pushedStream).isLocallyClosed());
            return new Adapter() {

                @Override
                public void onData(Stream pushedStream, DataFrame frame, Callback callback) {
                    Assert.assertTrue(pushedStream.isClosed());
                    callback.succeeded();
                    clientLatch.countDown();
                }
            };
        }
    });
    Assert.assertTrue(serverLatch.await(5, TimeUnit.SECONDS));
    Assert.assertTrue(clientLatch.await(5, TimeUnit.SECONDS));
}

Example 39 with Session

use of com.developmentontheedge.be5.api.Session in project jetty.project by eclipse.

the class StreamCountTest method testServerAllowsOneStreamEnforcedByServer.

@Test
public void testServerAllowsOneStreamEnforcedByServer() throws Exception {
    final CountDownLatch resetLatch = new CountDownLatch(1);
    start(new ServerSessionListener.Adapter() {

        @Override
        public Stream.Listener onNewStream(Stream stream, HeadersFrame frame) {
            HTTP2Session session = (HTTP2Session) stream.getSession();
            session.setMaxRemoteStreams(1);
            return new Stream.Listener.Adapter() {

                @Override
                public void onData(Stream stream, DataFrame frame, Callback callback) {
                    if (frame.isEndStream()) {
                        HttpFields fields = new HttpFields();
                        MetaData.Response metaData = new MetaData.Response(HttpVersion.HTTP_2, 200, fields);
                        stream.headers(new HeadersFrame(stream.getId(), metaData, null, true), callback);
                    } else {
                        callback.succeeded();
                    }
                }
            };
        }
    });
    Session session = newClient(new Session.Listener.Adapter());
    HttpFields fields = new HttpFields();
    MetaData.Request metaData = newRequest("GET", fields);
    HeadersFrame frame1 = new HeadersFrame(metaData, null, false);
    FuturePromise<Stream> streamPromise1 = new FuturePromise<>();
    final CountDownLatch responseLatch = new CountDownLatch(1);
    session.newStream(frame1, streamPromise1, new Stream.Listener.Adapter() {

        @Override
        public void onHeaders(Stream stream, HeadersFrame frame) {
            if (frame.isEndStream())
                responseLatch.countDown();
        }
    });
    Stream stream1 = streamPromise1.get(5, TimeUnit.SECONDS);
    HeadersFrame frame2 = new HeadersFrame(metaData, null, false);
    FuturePromise<Stream> streamPromise2 = new FuturePromise<>();
    session.newStream(frame2, streamPromise2, new Stream.Listener.Adapter() {

        @Override
        public void onReset(Stream stream, ResetFrame frame) {
            resetLatch.countDown();
        }
    });
    streamPromise2.get(5, TimeUnit.SECONDS);
    Assert.assertTrue(resetLatch.await(5, TimeUnit.SECONDS));
    stream1.data(new DataFrame(stream1.getId(), BufferUtil.EMPTY_BUFFER, true), Callback.NOOP);
    Assert.assertTrue(responseLatch.await(5, TimeUnit.SECONDS));
}

Example 40 with Session

use of com.developmentontheedge.be5.api.Session in project jetty.project by eclipse.

the class StreamResetTest method testClientResetConsumesQueuedData.

@Test
public void testClientResetConsumesQueuedData() throws Exception {
    start(new EmptyHttpServlet());
    Session client = newClient(new Session.Listener.Adapter());
    MetaData.Request request = newRequest("GET", new HttpFields());
    HeadersFrame frame = new HeadersFrame(request, null, false);
    FuturePromise<Stream> promise = new FuturePromise<>();
    client.newStream(frame, promise, new Stream.Listener.Adapter());
    Stream stream = promise.get(5, TimeUnit.SECONDS);
    ByteBuffer data = ByteBuffer.allocate(FlowControlStrategy.DEFAULT_WINDOW_SIZE);
    CountDownLatch dataLatch = new CountDownLatch(1);
    stream.data(new DataFrame(stream.getId(), data, false), new Callback() {

        @Override
        public void succeeded() {
            dataLatch.countDown();
        }
    });
    // The server does not read the data, so the flow control window should be zero.
    Assert.assertTrue(dataLatch.await(5, TimeUnit.SECONDS));
    Assert.assertEquals(0, ((ISession) client).updateSendWindow(0));
    // Now reset the stream.
    stream.reset(new ResetFrame(stream.getId(), ErrorCode.CANCEL_STREAM_ERROR.code), Callback.NOOP);
    // Wait for the server to receive the reset and process
    // it, and for the client to process the window updates.
    Thread.sleep(1000);
    Assert.assertThat(((ISession) client).updateSendWindow(0), Matchers.greaterThan(0));
}