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 1688d40423..a8c64b22ca 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 @@ -44,6 +44,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; @@ -140,6 +141,14 @@ public class StompMessageHandlerWebSocketIntegrationTests { Message failedMessage = messageDeliveryException.getFailedMessage(); assertThat((String) failedMessage.getPayload()).contains("preSend intentional Exception"); + receive = this.stompEvents.receive(10000); + assertThat(receive).isNotNull(); + assertThat(receive.getPayload()).isInstanceOf(StompConnectionFailedEvent.class); + + receive = this.stompEvents.receive(10000); + assertThat(receive).isNotNull(); + assertThat(receive.getPayload()).isInstanceOf(StompSessionConnectedEvent.class); + receive = this.stompEvents.receive(10000); assertThat(receive).isNotNull(); assertThat(receive.getPayload()).isInstanceOf(StompReceiptEvent.class);