Examples with WritableByteChannelMock org.apache.hc.core5.http2.WritableByteChannelMock used on opensource projects

Example 11 with WritableByteChannelMock

use of org.apache.hc.core5.http2.WritableByteChannelMock in project httpcomponents-core by apache.

the class TestFileAsyncEntityProducer method testTextContentRepeatable.

@Test
public void testTextContentRepeatable() throws Exception {
    final AsyncEntityProducer producer = new FileEntityProducer(tempFile, ContentType.TEXT_PLAIN);
    Assertions.assertEquals(6, producer.getContentLength());
    Assertions.assertEquals(ContentType.TEXT_PLAIN.toString(), producer.getContentType());
    Assertions.assertNull(producer.getContentEncoding());
    for (int i = 0; i < 3; i++) {
        final WritableByteChannelMock byteChannel = new WritableByteChannelMock(1024);
        final DataStreamChannel streamChannel = new BasicDataStreamChannel(byteChannel);
        producer.produce(streamChannel);
        producer.produce(streamChannel);
        Assertions.assertFalse(byteChannel.isOpen());
        Assertions.assertEquals("abcdef", byteChannel.dump(StandardCharsets.US_ASCII));
        producer.releaseResources();
    }
}

Example 12 with WritableByteChannelMock

use of org.apache.hc.core5.http2.WritableByteChannelMock in project httpcomponents-core by apache.

the class TestSharedOutputBuffer method testMultithreadingWriteStream.

@Test
public void testMultithreadingWriteStream() throws Exception {
    final Charset charset = StandardCharsets.US_ASCII;
    final SharedOutputBuffer outputBuffer = new SharedOutputBuffer(20);
    final WritableByteChannelMock channel = new WritableByteChannelMock(1024);
    final DataStreamChannelMock dataStreamChannel = new DataStreamChannelMock(channel);
    final ExecutorService executorService = Executors.newFixedThreadPool(2);
    final Future<Boolean> task1 = executorService.submit(() -> {
        final byte[] tmp = "1234567890".getBytes(charset);
        outputBuffer.write(tmp, 0, tmp.length);
        outputBuffer.write(tmp, 0, tmp.length);
        outputBuffer.write('1');
        outputBuffer.write('2');
        outputBuffer.write(tmp, 0, tmp.length);
        outputBuffer.write(tmp, 0, tmp.length);
        outputBuffer.write(tmp, 0, tmp.length);
        outputBuffer.writeCompleted();
        outputBuffer.writeCompleted();
        return Boolean.TRUE;
    });
    final Future<Boolean> task2 = executorService.submit(() -> {
        for (; ; ) {
            outputBuffer.flush(dataStreamChannel);
            if (outputBuffer.isEndStream()) {
                break;
            }
            if (!outputBuffer.hasData()) {
                dataStreamChannel.awaitOutputRequest();
            }
        }
        return Boolean.TRUE;
    });
    Assertions.assertEquals(Boolean.TRUE, task1.get(TIMEOUT.getDuration(), TIMEOUT.getTimeUnit()));
    Assertions.assertEquals(Boolean.TRUE, task2.get(TIMEOUT.getDuration(), TIMEOUT.getTimeUnit()));
    Assertions.assertEquals("1234567890123456789012123456789012345678901234567890", new String(channel.toByteArray(), charset));
}

Example 13 with WritableByteChannelMock

use of org.apache.hc.core5.http2.WritableByteChannelMock in project httpcomponents-core by apache.

the class TestSharedOutputBuffer method testBasis.

@Test
public void testBasis() throws Exception {
    final Charset charset = StandardCharsets.US_ASCII;
    final SharedOutputBuffer outputBuffer = new SharedOutputBuffer(30);
    final WritableByteChannelMock channel = new WritableByteChannelMock(1024);
    final DataStreamChannel dataStreamChannel = Mockito.spy(new DataStreamChannelMock(channel));
    outputBuffer.flush(dataStreamChannel);
    Mockito.verifyNoInteractions(dataStreamChannel);
    Assertions.assertEquals(0, outputBuffer.length());
    Assertions.assertEquals(30, outputBuffer.capacity());
    final byte[] tmp = "1234567890".getBytes(charset);
    outputBuffer.write(tmp, 0, tmp.length);
    outputBuffer.write(tmp, 0, tmp.length);
    outputBuffer.write('1');
    outputBuffer.write('2');
    Assertions.assertEquals(22, outputBuffer.length());
    Assertions.assertEquals(8, outputBuffer.capacity());
    Mockito.verifyNoInteractions(dataStreamChannel);
}

Example 14 with WritableByteChannelMock

use of org.apache.hc.core5.http2.WritableByteChannelMock in project httpcomponents-core by apache.

the class TestPathAsyncEntityProducer method testTextContent.

@Test
public void testTextContent() throws Exception {
    final Path tempPath = tempFile.toPath();
    final AsyncEntityProducer producer = new PathEntityProducer(tempPath, ContentType.TEXT_PLAIN, StandardOpenOption.READ);
    Assertions.assertEquals(6, producer.getContentLength());
    Assertions.assertEquals(ContentType.TEXT_PLAIN.toString(), producer.getContentType());
    Assertions.assertNull(producer.getContentEncoding());
    final WritableByteChannelMock byteChannel = new WritableByteChannelMock(1024);
    final DataStreamChannel streamChannel = new BasicDataStreamChannel(byteChannel);
    producer.produce(streamChannel);
    producer.produce(streamChannel);
    Assertions.assertFalse(byteChannel.isOpen());
    Assertions.assertEquals("abcdef", byteChannel.dump(StandardCharsets.US_ASCII));
}

Example 15 with WritableByteChannelMock

use of org.apache.hc.core5.http2.WritableByteChannelMock in project httpcomponents-core by apache.

the class TestLengthDelimitedEncoder method testCodingBeyondContentLimit.

@Test
public void testCodingBeyondContentLimit() throws Exception {
    final WritableByteChannelMock channel = new WritableByteChannelMock(64);
    final SessionOutputBuffer outbuf = new SessionOutputBufferImpl(1024, 128);
    final BasicHttpTransportMetrics metrics = new BasicHttpTransportMetrics();
    final LengthDelimitedEncoder encoder = new LengthDelimitedEncoder(channel, outbuf, metrics, 16);
    encoder.write(CodecTestUtils.wrap("stuff;"));
    encoder.write(CodecTestUtils.wrap("more stuff; and a lot more stuff"));
    final String s = channel.dump(StandardCharsets.US_ASCII);
    Assertions.assertTrue(encoder.isCompleted());
    Assertions.assertEquals("stuff;more stuff", s);
}