diff --git a/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/messaging/websocket/TraceWebSocketAutoConfigurationTest.java b/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/messaging/websocket/TraceWebSocketAutoConfigurationTest.java index ce8375fe4..442590476 100644 --- a/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/messaging/websocket/TraceWebSocketAutoConfigurationTest.java +++ b/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/messaging/websocket/TraceWebSocketAutoConfigurationTest.java @@ -46,13 +46,16 @@ public class TraceWebSocketAutoConfigurationTest { DelegatingWebSocketMessageBrokerConfiguration delegatingWebSocketMessageBrokerConfiguration; @Test - public void should_register_interceptors_for_inbound_and_outbound_channels() { + public void should_register_interceptors_for_all_channels() { then(this.delegatingWebSocketMessageBrokerConfiguration.clientInboundChannel() .getInterceptors()) .hasAtLeastOneElementOfType(TraceChannelInterceptor.class); then(this.delegatingWebSocketMessageBrokerConfiguration.clientOutboundChannel() .getInterceptors()) .hasAtLeastOneElementOfType(TraceChannelInterceptor.class); + then(this.delegatingWebSocketMessageBrokerConfiguration.brokerChannel() + .getInterceptors()) + .hasAtLeastOneElementOfType(TraceChannelInterceptor.class); } @EnableAutoConfiguration