From 06b06b1c8ff09c5859bd0f6d17e764e8aa310022 Mon Sep 17 00:00:00 2001
From: Gary Russell <grussell@gopivotal.com>
Date: Mon, 26 Aug 2013 21:59:20 -0400
Subject: [PATCH] INT-3103 Fix Connection Timeout Tests

Test cases could occasionally fail because an ErrorMessage is
now returned to the listener.
---
 .../ip/tcp/connection/ConnectionTimeoutTests.java | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/ConnectionTimeoutTests.java b/spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/ConnectionTimeoutTests.java
index e401884e87..5e9ad3ab87 100644
--- a/spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/ConnectionTimeoutTests.java
+++ b/spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/ConnectionTimeoutTests.java
@@ -33,6 +33,7 @@ import org.springframework.context.ApplicationEvent;
 import org.springframework.context.ApplicationEventPublisher;
 import org.springframework.integration.Message;
 import org.springframework.integration.ip.util.TestingUtilities;
+import org.springframework.integration.message.ErrorMessage;
 import org.springframework.integration.support.MessageBuilder;
 import org.springframework.integration.test.util.SocketUtils;
 import org.springframework.integration.test.util.TestUtils;
@@ -121,8 +122,10 @@ public class ConnectionTimeoutTests {
 		client.registerListener(new TcpListener() {
 			@Override
 			public boolean onMessage(Message<?> message) {
-				reply.set(message);
-				replyLatch.countDown();
+				if (!(message instanceof ErrorMessage)) {
+					reply.set(message);
+					replyLatch.countDown();
+				}
 				return false;
 			}
 		});
@@ -157,7 +160,9 @@ public class ConnectionTimeoutTests {
 		client.registerListener(new TcpListener() {
 			@Override
 			public boolean onMessage(Message<?> message) {
-				reply.set(message);
+				if (!(message instanceof ErrorMessage)) {
+					reply.set(message);
+				}
 				return false;
 			}
 		});
@@ -195,7 +200,9 @@ public class ConnectionTimeoutTests {
 		client.registerListener(new TcpListener() {
 			@Override
 			public boolean onMessage(Message<?> message) {
-				reply.set(message);
+				if (!(message instanceof ErrorMessage)) {
+					reply.set(message);
+				}
 				return false;
 			}
 		});