fixes

Signed-off-by: Oleh Dokuka <odokuka@vmware.com>
Signed-off-by: Oleh Dokuka <oleh.dokuka@icloud.com>
Signed-off-by: OlegDokuka <odokuka@vmware.com>

Author: OlegDokuka

rsocket-examples/src/test/java/io/rsocket/integration/GracefulShutdownIntegrationTest.java

@@ -79,11 +79,14 @@ public void testCompleteWithoutNext() {
     final AtomicBoolean gracefullyDisposedClient = new AtomicBoolean();
     final AtomicBoolean disposedServer = new AtomicBoolean();
     final AtomicBoolean disposedClient = new AtomicBoolean();
+    final Sinks.Empty<Void> requestHandled = Sinks.unsafe().empty();
     handler =
         new RSocket() {
           @Override
           public Flux<Payload> requestStream(Payload payload) {
             payload.release();
+
+            requestHandled.tryEmitEmpty();
             return Flux.<Payload>never().takeUntilOther(serverGracefulShutdownSink.asMono());
           }
 
@@ -135,6 +138,8 @@ public void disposeGracefully() {
             .expectComplete()
             .verifyLater();
 
+    requestHandled.asMono().block(Duration.ofSeconds(5));
+
     assertThat(gracefullyDisposedServer).isFalse();
     assertThat(gracefullyDisposedClient).isFalse();