From ae22fa2a5d10fe97b00c6e56155141e8464747f1 Mon Sep 17 00:00:00 2001
From: jbertram <jbertram@apache.org>
Date: Mon, 7 Dec 2015 10:17:05 -0600
Subject: [PATCH] ARTEMIS-314 test-case to exercise behavior

---
 .../tests/integration/ra/ActiveMQClusteredTest.java  | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/ra/ActiveMQClusteredTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/ra/ActiveMQClusteredTest.java
index 3ffac2ea24..e6b9a0152c 100644
--- a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/ra/ActiveMQClusteredTest.java
+++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/ra/ActiveMQClusteredTest.java
@@ -172,7 +172,16 @@ public class ActiveMQClusteredTest extends ActiveMQRAClusteredTestBase {
    }
 
    @Test
-   public void testRebalance() throws Exception {
+   public void testRebalanceWithInfiniteReconnect() throws Exception {
+      testRebalanceInternal(-1);
+   }
+
+   @Test
+   public void testRebalanceWithNoReconnect() throws Exception {
+      testRebalanceInternal(0);
+   }
+
+   public void testRebalanceInternal(int reconnectAttempts) throws Exception {
       final int CONSUMER_COUNT = 10;
       secondaryJmsServer.createQueue(true, MDBQUEUE, null, true, "/jms/" + MDBQUEUE);
 
@@ -188,6 +197,7 @@ public class ActiveMQClusteredTest extends ActiveMQRAClusteredTestBase {
       spec.setMaxSession(CONSUMER_COUNT);
       spec.setSetupAttempts(5);
       spec.setSetupInterval(200);
+      spec.setReconnectAttempts(reconnectAttempts);
       spec.setHA(true); // if this isn't true then the toplogy listener won't get nodeDown notifications
       spec.setCallTimeout(500L); // if this isn't set then it may take a long time for tearDown to occur on the MDB connection
       qResourceAdapter.setConnectorClassName(INVM_CONNECTOR_FACTORY);