From 8b2637dfbd5337eb90d64d38948933432809de8e Mon Sep 17 00:00:00 2001
From: Gary Russell <grussell@gopivotal.com>
Date: Wed, 5 Mar 2014 10:21:25 -0500
Subject: [PATCH] INT-3305 Fix SFTP Test Case

JIRA: https://jira.springsource.org/browse/INT-3305
---
 .../integration/sftp/session/INT3305Tests.java        | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/spring-integration-sftp/src/test/java/org/springframework/integration/sftp/session/INT3305Tests.java b/spring-integration-sftp/src/test/java/org/springframework/integration/sftp/session/INT3305Tests.java
index 0e74d86c97..7f7649aaaa 100644
--- a/spring-integration-sftp/src/test/java/org/springframework/integration/sftp/session/INT3305Tests.java
+++ b/spring-integration-sftp/src/test/java/org/springframework/integration/sftp/session/INT3305Tests.java
@@ -21,10 +21,7 @@ import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertThat;
 import static org.junit.Assert.fail;
 
-import java.util.List;
-
 import org.apache.sshd.SshServer;
-import org.apache.sshd.common.session.AbstractSession;
 import org.apache.sshd.server.PasswordAuthenticator;
 import org.apache.sshd.server.keyprovider.SimpleGeneratorHostKeyProvider;
 import org.apache.sshd.server.session.ServerSession;
@@ -72,9 +69,13 @@ public class INT3305Tests {
 				assertThat(e.getCause(), instanceOf(IllegalStateException.class));
 				assertThat(e.getCause().getMessage(), equalTo("failed to connect"));
 			}
-			List<AbstractSession> sessions = server.getActiveSessions();
 
-			assertEquals(0, sessions.size());
+			int n = 0;
+			while (n++ < 100 && server.getActiveSessions().size() > 0) {
+				Thread.sleep(100);
+			}
+
+			assertEquals(0, server.getActiveSessions().size());
 		}
 		finally {
 			server.stop(true);