Fix StompMHWebSocketIntTests for reconnect events

https://build.spring.io/browse/INT-MASTERSPRING40-740

Author: artembilan

spring-integration-stomp/src/test/java/org/springframework/integration/stomp/outbound/StompMessageHandlerWebSocketIntegrationTests.java

@@ -44,6 +44,7 @@
 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 void testStompMessageHandler() throws InterruptedException {
 		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);