Fix StompMHWebSocketIntTests for reconnect events

https://build.spring.io/browse/INT-MASTERSPRING40-740
This commit is contained in:
Artem Bilan
2019-06-17 13:02:24 -04:00
parent a77dac5347
commit 153a748166

View File

@@ -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);