fixes test

Signed-off-by: Oleh Dokuka <[email protected]>
Signed-off-by: Oleh Dokuka <[email protected]>

Author: Oleh Dokuka

build.gradle

@@ -47,6 +47,7 @@ subprojects {
     ext['assertj.version'] = '3.21.0'
     ext['netflix.limits.version'] = '0.3.6'
     ext['bouncycastle-bcpkix.version'] = '1.68'
+    ext['awaitility.version'] = '4.1.1'
 
     group = "io.rsocket"
 
@@ -80,6 +81,7 @@ subprojects {
             dependency "org.assertj:assertj-core:${ext['assertj.version']}"
             dependency "org.hdrhistogram:HdrHistogram:${ext['hdrhistogram.version']}"
             dependency "org.slf4j:slf4j-api:${ext['slf4j.version']}"
+            dependency "org.awaitility:awaitility:${ext['awaitility.version']}"
             dependencySet(group: 'org.mockito', version: ext['mockito.version']) {
                 entry 'mockito-junit-jupiter'
                 entry 'mockito-core'

rsocket-core/build.gradle

@@ -35,6 +35,7 @@ dependencies {
     testImplementation 'org.junit.jupiter:junit-jupiter-api'
     testImplementation 'org.junit.jupiter:junit-jupiter-params'
     testImplementation 'org.mockito:mockito-core'
+    testImplementation 'org.awaitility:awaitility'
 
     testRuntimeOnly 'ch.qos.logback:logback-classic'
     testRuntimeOnly 'org.junit.jupiter:junit-jupiter-engine'

rsocket-core/src/test/java/io/rsocket/core/ReconnectMonoTests.java

@@ -17,6 +17,7 @@
 package io.rsocket.core;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.awaitility.Awaitility.await;
 
 import io.rsocket.RaceTestConstants;
 import io.rsocket.internal.subscriber.AssertSubscriber;
@@ -321,6 +322,7 @@ public String get() {
 
       assertThat(expired).hasSize(1).containsOnly("value_to_expire" + i);
       if (reconnectMono.resolvingInner.subscribers == ResolvingOperator.READY) {
+        await().atMost(Duration.ofSeconds(5)).until(() -> received.size() == 2);
         assertThat(received)
             .hasSize(2)
             .containsExactly(