diff --git a/tests/brave/spring-cloud-sleuth-instrumentation-lettuce-tests/src/test/java/org/springframework/cloud/sleuth/brave/instrument/redis/BraveLettuceIntegrationTests.java b/tests/brave/spring-cloud-sleuth-instrumentation-lettuce-tests/src/test/java/org/springframework/cloud/sleuth/brave/instrument/redis/BraveLettuceIntegrationTests.java index 1c82644a9..631cb5de9 100644 --- a/tests/brave/spring-cloud-sleuth-instrumentation-lettuce-tests/src/test/java/org/springframework/cloud/sleuth/brave/instrument/redis/BraveLettuceIntegrationTests.java +++ b/tests/brave/spring-cloud-sleuth-instrumentation-lettuce-tests/src/test/java/org/springframework/cloud/sleuth/brave/instrument/redis/BraveLettuceIntegrationTests.java @@ -17,6 +17,8 @@ package org.springframework.cloud.sleuth.brave.instrument.redis; import brave.sampler.Sampler; +import io.lettuce.core.tracing.BraveTracing; +import io.lettuce.core.tracing.Tracing; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.cloud.sleuth.brave.BraveTestSpanHandler; @@ -30,6 +32,11 @@ import org.springframework.test.context.ContextConfiguration; public class BraveLettuceIntegrationTests extends org.springframework.cloud.sleuth.instrument.redis.LettuceIntegrationTests { + @Override + protected Class tracing() { + return BraveTracing.class; + } + @Configuration(proxyBeanMethods = false) static class Config { diff --git a/tests/common/src/main/java/org/springframework/cloud/sleuth/instrument/redis/LettuceIntegrationTests.java b/tests/common/src/main/java/org/springframework/cloud/sleuth/instrument/redis/LettuceIntegrationTests.java index 8ba0f862d..94b656cfa 100644 --- a/tests/common/src/main/java/org/springframework/cloud/sleuth/instrument/redis/LettuceIntegrationTests.java +++ b/tests/common/src/main/java/org/springframework/cloud/sleuth/instrument/redis/LettuceIntegrationTests.java @@ -17,7 +17,7 @@ package org.springframework.cloud.sleuth.instrument.redis; import io.lettuce.core.resource.ClientResources; -import io.lettuce.core.tracing.BraveTracing; +import io.lettuce.core.tracing.Tracing; import org.junit.jupiter.api.Test; import org.springframework.beans.factory.annotation.Autowired; @@ -42,15 +42,18 @@ public abstract class LettuceIntegrationTests { TraceLettuceClientResourcesBuilderCustomizer customizer; @Autowired - BraveTracing braveTracing; + Tracing tracing; @Test void tracing_should_be_set() { then(this.clientResources).isNull(); then(this.customizer).isNotNull(); - then(this.braveTracing.isEnabled()).isTrue(); + then(this.tracing.isEnabled()).isTrue(); + then(this.tracing).isInstanceOf(tracing()); } + protected abstract Class tracing(); + @Configuration(proxyBeanMethods = false) @EnableAutoConfiguration protected static class TestConfig {