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

@@ -49,6 +49,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;
@@ -145,6 +146,14 @@ public void testStompMessageHandler() throws InterruptedException {
 		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));