GH-3199: Test for fail back with Long.MAX_VALUE

Resolves #3199

Author: garyrussell

spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/FailoverClientConnectionFactoryTests.java

@@ -18,6 +18,7 @@
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
+import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.doAnswer;
 import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.doThrow;
@@ -40,7 +41,6 @@
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicReference;
 
-import org.junit.Rule;
 import org.junit.Test;
 import org.mockito.InOrder;
 import org.mockito.Mockito;
@@ -56,7 +56,6 @@
 import org.springframework.integration.ip.tcp.TcpInboundGateway;
 import org.springframework.integration.ip.tcp.TcpOutboundGateway;
 import org.springframework.integration.ip.util.TestingUtilities;
-import org.springframework.integration.test.rule.Log4j2LevelAdjuster;
 import org.springframework.integration.test.util.TestUtils;
 import org.springframework.integration.util.SimplePool;
 import org.springframework.messaging.Message;
@@ -86,12 +85,6 @@ public void publishEvent(Object event) {
 
 	};
 
-	@Rule
-	public Log4j2LevelAdjuster adjuster =
-			Log4j2LevelAdjuster.trace()
-					.classes(SimplePool.class)
-					.categories("org.springframework.integration.ip.tcp");
-
 	@Test
 	public void testFailoverGood() throws Exception {
 		AbstractClientConnectionFactory factory1 = mock(AbstractClientConnectionFactory.class);
@@ -117,15 +110,20 @@ public void testFailoverGood() throws Exception {
 
 	@Test
 	public void testRefreshShared() throws Exception {
-		testRefreshShared(false);
+		testRefreshShared(false, 10_000);
 	}
 
 	@Test
 	public void testRefreshSharedCloseOnRefresh() throws Exception {
-		testRefreshShared(true);
+		testRefreshShared(true, 10_000);
+	}
+
+	@Test
+	public void testRefreshSharedInfinite() throws Exception {
+		testRefreshShared(false, Long.MAX_VALUE);
 	}
 
-	private void testRefreshShared(boolean closeOnRefresh) throws Exception {
+	private void testRefreshShared(boolean closeOnRefresh, long interval) throws Exception {
 		AbstractClientConnectionFactory factory1 = mock(AbstractClientConnectionFactory.class);
 		AbstractClientConnectionFactory factory2 = mock(AbstractClientConnectionFactory.class);
 		List<AbstractClientConnectionFactory> factories = new ArrayList<AbstractClientConnectionFactory>();
@@ -150,21 +148,29 @@ private void testRefreshShared(boolean closeOnRefresh) throws Exception {
 		failoverFactory.start();
 		TcpConnectionSupport connection = failoverFactory.getConnection();
 		assertThat(TestUtils.getPropertyValue(failoverFactory, "theConnection")).isNotNull();
-		failoverFactory.setRefreshSharedInterval(10_000);
+		failoverFactory.setRefreshSharedInterval(interval);
+		InOrder inOrder = inOrder(factory1, factory2, conn1, conn2);
+		inOrder.verify(factory1).getConnection();
+		inOrder.verify(factory2).getConnection();
+		inOrder.verify(conn1).registerListener(any());
+		inOrder.verify(conn1).isOpen();
 		assertThat(failoverFactory.getConnection()).isSameAs(connection);
+		inOrder.verifyNoMoreInteractions();
 		failoverFactory.setRefreshSharedInterval(-1);
 		assertThat(failoverFactory.getConnection()).isNotSameAs(connection);
-		InOrder inOrder = inOrder(factory1, factory2, conn1);
-		inOrder.verify(factory1).getConnection();
-		inOrder.verify(factory2).getConnection();
 		inOrder.verify(factory1).getConnection();
 		inOrder.verify(factory2).getConnection();
 		if (closeOnRefresh) {
+			inOrder.verify(conn2).registerListener(any());
+			inOrder.verify(conn2).isOpen();
 			inOrder.verify(conn1).close();
 		}
 		else {
+			inOrder.verify(conn1).registerListener(any());
+			inOrder.verify(conn1).isOpen();
 			inOrder.verify(conn1, never()).close();
 		}
+		inOrder.verifyNoMoreInteractions();
 	}
 
 	@Test(expected = UncheckedIOException.class)