Skip to content

Commit 865594b

Browse files
authored
Retry open connection on lettuce 5 tests (#8145)
1 parent 1951e6f commit 865594b

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

dd-java-agent/instrumentation/lettuce-5/src/test/groovy/Lettuce5ReactiveClientTest.groovy

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,14 @@ import datadog.trace.api.DDSpanTypes
88
import datadog.trace.bootstrap.instrumentation.api.Tags
99
import io.lettuce.core.ClientOptions
1010
import io.lettuce.core.RedisClient
11-
import io.lettuce.core.api.StatefulConnection
11+
import io.lettuce.core.api.StatefulRedisConnection
1212
import io.lettuce.core.api.reactive.RedisReactiveCommands
1313
import io.lettuce.core.api.sync.RedisCommands
1414
import org.testcontainers.containers.wait.strategy.Wait
1515
import reactor.core.scheduler.Schedulers
1616
import spock.lang.Shared
1717
import spock.util.concurrent.AsyncConditions
18+
import spock.util.concurrent.PollingConditions
1819

1920
import java.util.function.Consumer
2021

@@ -40,7 +41,7 @@ abstract class Lettuce5ReactiveClientTest extends VersionedNamingTestBase {
4041
.waitingFor(Wait.forListeningPort())
4142

4243
RedisClient redisClient
43-
StatefulConnection connection
44+
StatefulRedisConnection connection
4445
RedisReactiveCommands<String, ?> reactiveCommands
4546
RedisCommands<String, ?> syncCommands
4647

@@ -54,7 +55,9 @@ abstract class Lettuce5ReactiveClientTest extends VersionedNamingTestBase {
5455

5556
redisClient.setOptions(CLIENT_OPTIONS)
5657

57-
connection = redisClient.connect()
58+
new PollingConditions(delay: 3, timeout: 15).eventually {
59+
(connection = redisClient.connect()) != null
60+
}
5861
reactiveCommands = connection.reactive()
5962
syncCommands = connection.sync()
6063

0 commit comments

Comments
 (0)