From 56d26443e22d0704c655fa2535fe48501b8cebff Mon Sep 17 00:00:00 2001
From: Rossen Stoyanchev <rstoyanchev@gopivotal.com>
Date: Tue, 29 Apr 2014 16:36:43 -0400
Subject: [PATCH] Fix one more issue in STOMP broker relay int test

---
 .../stomp/StompBrokerRelayMessageHandlerIntegrationTests.java  | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/spring-messaging/src/test/java/org/springframework/messaging/simp/stomp/StompBrokerRelayMessageHandlerIntegrationTests.java b/spring-messaging/src/test/java/org/springframework/messaging/simp/stomp/StompBrokerRelayMessageHandlerIntegrationTests.java
index 72bc45a6ed..5aeaa2f3c0 100644
--- a/spring-messaging/src/test/java/org/springframework/messaging/simp/stomp/StompBrokerRelayMessageHandlerIntegrationTests.java
+++ b/spring-messaging/src/test/java/org/springframework/messaging/simp/stomp/StompBrokerRelayMessageHandlerIntegrationTests.java
@@ -168,6 +168,8 @@ public class StompBrokerRelayMessageHandlerIntegrationTests {
 		logger.debug("Starting test messageDeliveryExceptionIfSystemSessionForwardFails()");
 
 		stopActiveMqBrokerAndAwait();
+		this.eventPublisher.expectBrokerAvailabilityEvent(false);
+
 		StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.SEND);
 		this.relay.handleMessage(MessageBuilder.createMessage("test".getBytes(), headers.getMessageHeaders()));
 	}
@@ -184,6 +186,7 @@ public class StompBrokerRelayMessageHandlerIntegrationTests {
 
 		MessageExchange error = MessageExchangeBuilder.error(sess1).build();
 		stopActiveMqBrokerAndAwait();
+		this.eventPublisher.expectBrokerAvailabilityEvent(false);
 		this.responseHandler.expectMessages(error);
 	}