Examples with Completable io.servicetalk.concurrent.api.Completable used on opensource projects

Example 6 with Completable

use of io.servicetalk.concurrent.api.Completable in project servicetalk by apple.

the class HttpRequestEncoderTest method protocolPayloadEndOutboundShouldNotTriggerOnFailedFlush.

@Test
void protocolPayloadEndOutboundShouldNotTriggerOnFailedFlush() throws Exception {
    AtomicReference<CloseHandler> closeHandlerRef = new AtomicReference<>();
    try (CompositeCloseable resources = newCompositeCloseable()) {
        Processor serverCloseTrigger = newCompletableProcessor();
        CountDownLatch serverChannelLatch = new CountDownLatch(1);
        AtomicReference<Channel> serverChannelRef = new AtomicReference<>();
        ReadOnlyTcpServerConfig sConfig = new TcpServerConfig().asReadOnly();
        ServerContext serverContext = resources.prepend(TcpServerBinder.bind(localAddress(0), sConfig, false, SEC, null, (channel, observer) -> DefaultNettyConnection.initChannel(channel, SEC.bufferAllocator(), SEC.executor(), SEC.ioExecutor(), forPipelinedRequestResponse(false, channel.config()), defaultFlushStrategy(), null, new TcpServerChannelInitializer(sConfig, observer).andThen(channel2 -> {
            serverChannelRef.compareAndSet(null, channel2);
            serverChannelLatch.countDown();
        }), defaultStrategy(), mock(Protocol.class), observer, false, __ -> false), connection -> {
        }).toFuture().get());
        ReadOnlyHttpClientConfig cConfig = new HttpClientConfig().asReadOnly();
        assert cConfig.h1Config() != null;
        NettyConnection<Object, Object> conn = resources.prepend(TcpConnector.connect(null, serverHostAndPort(serverContext), cConfig.tcpConfig(), false, CEC, (channel, connectionObserver) -> {
            CloseHandler closeHandler = spy(forPipelinedRequestResponse(true, channel.config()));
            closeHandlerRef.compareAndSet(null, closeHandler);
            return DefaultNettyConnection.initChannel(channel, CEC.bufferAllocator(), CEC.executor(), CEC.ioExecutor(), closeHandler, defaultFlushStrategy(), null, new TcpClientChannelInitializer(cConfig.tcpConfig(), connectionObserver).andThen(new HttpClientChannelInitializer(getByteBufAllocator(CEC.bufferAllocator()), cConfig.h1Config(), closeHandler)).andThen(channel2 -> channel2.pipeline().addLast(new ChannelInboundHandlerAdapter() {

                @Override
                public void userEventTriggered(ChannelHandlerContext ctx, Object evt) {
                    // Propagate the user event in the pipeline before
                    // triggering the test condition.
                    ctx.fireUserEventTriggered(evt);
                    if (evt instanceof ChannelInputShutdownReadComplete) {
                        serverCloseTrigger.onComplete();
                    }
                }
            })), defaultStrategy(), HTTP_1_1, connectionObserver, true, __ -> false);
        }, NoopTransportObserver.INSTANCE).toFuture().get());
        // The server needs to wait to close the conneciton until after the client has established the connection.
        serverChannelLatch.await();
        Channel serverChannel = serverChannelRef.get();
        assertNotNull(serverChannel);
        assumeFalse(serverChannel instanceof NioSocketChannel, "Windows doesn't emit ChannelInputShutdownReadComplete. Investigation Required.");
        ((SocketChannel) serverChannel).config().setSoLinger(0);
        // Close and send RST concurrently with client write
        serverChannel.close();
        StreamingHttpRequest request = reqRespFactory.post("/closeme");
        fromSource(serverCloseTrigger).toFuture().get();
        Completable write = conn.write(from(request, allocator.fromAscii("Bye"), EmptyHttpHeaders.INSTANCE));
        assertThrows(ExecutionException.class, () -> write.toFuture().get());
        CloseHandler closeHandler = closeHandlerRef.get();
        assertNotNull(closeHandler);
        verify(closeHandler, never()).protocolPayloadEndOutbound(any(), any());
    }
}

Example 7 with Completable

use of io.servicetalk.concurrent.api.Completable in project servicetalk by apple.

the class RunnableCompletableTest method cancelInterrupts.

@Test
void cancelInterrupts() throws Exception {
    final Completable source = Completable.fromRunnable(factory);
    final CountDownLatch latch = new CountDownLatch(1);
    doAnswer(invocation -> {
        try {
            // await till interrupted.
            latch.await();
        } catch (InterruptedException e) {
            latch.countDown();
            Thread.currentThread().interrupt();
            throw e;
        }
        return 1;
    }).when(factory).run();
    toSource(source).subscribe(new CompletableSource.Subscriber() {

        @Override
        public void onSubscribe(final Cancellable cancellable) {
            cancellable.cancel();
        }

        @Override
        public void onComplete() {
        // noop
        }

        @Override
        public void onError(final Throwable t) {
        // noop
        }
    });
    latch.await();
}

Example 8 with Completable

use of io.servicetalk.concurrent.api.Completable in project servicetalk by apple.

the class RunnableCompletableTest method testOnError.

@Test
void testOnError() {
    doThrow(IllegalArgumentException.class).when(factory).run();
    Completable source = Completable.fromRunnable(factory);
    listenAndVerifyError(source);
    verify(factory).run();
}

Example 9 with Completable

use of io.servicetalk.concurrent.api.Completable in project servicetalk by apple.

the class RunnableCompletableTest method testEverySubscribeRuns.

@Test
void testEverySubscribeRuns() {
    Completable source = Completable.fromRunnable(factory);
    listenAndVerify(source);
    listenAndVerify(source);
    verify(factory, times(2)).run();
}

Example 10 with Completable

use of io.servicetalk.concurrent.api.Completable in project servicetalk by apple.

the class RunnableCompletableTest method onSubscribeThrows.

@Test
void onSubscribeThrows() {
    final Completable source = Completable.fromRunnable(factory);
    final CompletableSource.Subscriber subscriber = mock(CompletableSource.Subscriber.class);
    doThrow(DELIBERATE_EXCEPTION).when(subscriber).onSubscribe(any());
    toSource(source).subscribe(subscriber);
    verify(subscriber).onSubscribe(any());
    verify(subscriber).onError(DELIBERATE_EXCEPTION);
    verifyNoMoreInteractions(subscriber);
}