Examples with GoAwayGenerator org.eclipse.jetty.http2.generator.GoAwayGenerator used on opensource projects

Example 1 with GoAwayGenerator

use of org.eclipse.jetty.http2.generator.GoAwayGenerator in project jetty.project by eclipse.

the class GoAwayGenerateParseTest method testGenerateParseOneByteAtATime.

@Test
public void testGenerateParseOneByteAtATime() throws Exception {
    GoAwayGenerator generator = new GoAwayGenerator(new HeaderGenerator());
    final List<GoAwayFrame> frames = new ArrayList<>();
    Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter() {

        @Override
        public void onGoAway(GoAwayFrame frame) {
            frames.add(frame);
        }
    }, 4096, 8192);
    int lastStreamId = 13;
    int error = 17;
    byte[] payload = new byte[16];
    new Random().nextBytes(payload);
    // Iterate a few times to be sure generator and parser are properly reset.
    for (int i = 0; i < 2; ++i) {
        ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
        generator.generateGoAway(lease, lastStreamId, error, payload);
        frames.clear();
        for (ByteBuffer buffer : lease.getByteBuffers()) {
            while (buffer.hasRemaining()) {
                parser.parse(ByteBuffer.wrap(new byte[] { buffer.get() }));
            }
        }
        Assert.assertEquals(1, frames.size());
        GoAwayFrame frame = frames.get(0);
        Assert.assertEquals(lastStreamId, frame.getLastStreamId());
        Assert.assertEquals(error, frame.getError());
        Assert.assertArrayEquals(payload, frame.getPayload());
    }
}

Example 2 with GoAwayGenerator

use of org.eclipse.jetty.http2.generator.GoAwayGenerator in project jetty.project by eclipse.

the class GoAwayGenerateParseTest method testGenerateParse.

@Test
public void testGenerateParse() throws Exception {
    GoAwayGenerator generator = new GoAwayGenerator(new HeaderGenerator());
    final List<GoAwayFrame> frames = new ArrayList<>();
    Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter() {

        @Override
        public void onGoAway(GoAwayFrame frame) {
            frames.add(frame);
        }
    }, 4096, 8192);
    int lastStreamId = 13;
    int error = 17;
    // Iterate a few times to be sure generator and parser are properly reset.
    for (int i = 0; i < 2; ++i) {
        ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
        generator.generateGoAway(lease, lastStreamId, error, null);
        frames.clear();
        for (ByteBuffer buffer : lease.getByteBuffers()) {
            while (buffer.hasRemaining()) {
                parser.parse(buffer);
            }
        }
    }
    Assert.assertEquals(1, frames.size());
    GoAwayFrame frame = frames.get(0);
    Assert.assertEquals(lastStreamId, frame.getLastStreamId());
    Assert.assertEquals(error, frame.getError());
    Assert.assertNull(frame.getPayload());
}