diff --git a/spring-integration-stomp/src/test/java/org/springframework/integration/stomp/outbound/StompMessageHandlerWebSocketIntegrationTests.java b/spring-integration-stomp/src/test/java/org/springframework/integration/stomp/outbound/StompMessageHandlerWebSocketIntegrationTests.java index 9599ef914d..c6103d0b8c 100644 --- a/spring-integration-stomp/src/test/java/org/springframework/integration/stomp/outbound/StompMessageHandlerWebSocketIntegrationTests.java +++ b/spring-integration-stomp/src/test/java/org/springframework/integration/stomp/outbound/StompMessageHandlerWebSocketIntegrationTests.java @@ -49,6 +49,7 @@ import org.springframework.integration.config.EnableIntegration; import org.springframework.integration.event.inbound.ApplicationEventListeningMessageProducer; import org.springframework.integration.stomp.StompSessionManager; import org.springframework.integration.stomp.WebSocketStompSessionManager; +import org.springframework.integration.stomp.event.StompConnectionFailedEvent; import org.springframework.integration.stomp.event.StompExceptionEvent; import org.springframework.integration.stomp.event.StompIntegrationEvent; import org.springframework.integration.stomp.event.StompReceiptEvent; @@ -145,6 +146,14 @@ public class StompMessageHandlerWebSocketIntegrationTests { Message failedMessage = messageDeliveryException.getFailedMessage(); assertThat((String) failedMessage.getPayload(), containsString("preSend intentional Exception")); + receive = this.stompEvents.receive(10000); + assertNotNull(receive); + assertThat(receive.getPayload(), instanceOf(StompConnectionFailedEvent.class)); + + receive = this.stompEvents.receive(10000); + assertNotNull(receive); + assertThat(receive.getPayload(), instanceOf(StompSessionConnectedEvent.class)); + receive = this.stompEvents.receive(10000); assertNotNull(receive); assertThat(receive.getPayload(), instanceOf(StompReceiptEvent.class));