From c21a1e6ddaaad7ce48ca472c6adc72ccd3218881 Mon Sep 17 00:00:00 2001 From: Dave Syer Date: Wed, 18 Oct 2017 09:08:36 +0100 Subject: [PATCH] Test for intercepting broker channel --- .../websocket/TraceWebSocketAutoConfigurationTest.java | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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