This is an automated email from the ASF dual-hosted git repository.

clebertsuconic pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/activemq-artemis.git


The following commit(s) were added to refs/heads/main by this push:
     new 6f779a7908 ARTEMIS-5166 Improving ReconnectTest
6f779a7908 is described below

commit 6f779a790822e872e718228fe81464689a420d1b
Author: Clebert Suconic <[email protected]>
AuthorDate: Tue Nov 19 15:52:03 2024 -0500

    ARTEMIS-5166 Improving ReconnectTest
    
    I'm trying to make it faster and less likely to fail over non issues
---
 .../tests/integration/remoting/ReconnectTest.java  | 88 +++++++++++-----------
 1 file changed, 42 insertions(+), 46 deletions(-)

diff --git 
a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/remoting/ReconnectTest.java
 
b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/remoting/ReconnectTest.java
index 286809231b..81161ac6e6 100644
--- 
a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/remoting/ReconnectTest.java
+++ 
b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/remoting/ReconnectTest.java
@@ -67,7 +67,7 @@ public class ReconnectTest extends ActiveMQTestBase {
    }
 
    public void internalTestReconnect(final boolean isNetty) throws Exception {
-      final int pingPeriod = 1000;
+      final int pingPeriod = 100;
 
       ActiveMQServer server = createServer(false, isNetty);
 
@@ -111,10 +111,7 @@ public class ReconnectTest extends ActiveMQTestBase {
 
          assertTrue(latch.await(5, TimeUnit.SECONDS));
 
-         // Some time to let possible loops to occur
-         Thread.sleep(500);
-
-         assertEquals(1, count.get());
+         Wait.assertEquals(1, count::get, 5000, 100);
 
          locator.close();
       } finally {
@@ -148,42 +145,39 @@ public class ReconnectTest extends ActiveMQTestBase {
       ClientSessionInternal session = null;
 
       try {
-         for (int i = 0; i < 100; i++) {
-            ServerLocator locator = createFactory(isNetty);
-            locator.setClientFailureCheckPeriod(pingPeriod);
-            locator.setRetryInterval(1);
-            locator.setRetryIntervalMultiplier(1d);
-            locator.setReconnectAttempts(-1);
-            locator.setConfirmationWindowSize(-1);
-            ClientSessionFactory factory = createSessionFactory(locator);
+         ServerLocator locator = createFactory(isNetty);
+         locator.setClientFailureCheckPeriod(pingPeriod);
+         locator.setRetryInterval(1);
+         locator.setRetryIntervalMultiplier(1d);
+         locator.setReconnectAttempts(-1);
+         locator.setConfirmationWindowSize(-1);
+         ClientSessionFactory factory = createSessionFactory(locator);
 
-            session = (ClientSessionInternal) factory.createSession();
+         session = (ClientSessionInternal) factory.createSession();
 
-            session.addMetaData("meta1", "meta1");
+         session.addMetaData("meta1", "meta1");
 
-            ServerSession[] sessions = countMetadata(server, "meta1", 1);
-            assertEquals(1, sessions.length);
+         Wait.assertEquals(1, () -> getSessionsWithMetadata(server, 
"meta1").length);
 
-            final AtomicInteger count = new AtomicInteger(0);
+         ServerSession[] sessions = getSessionsWithMetadata(server, "meta1");
 
-            final CountDownLatch latch = new CountDownLatch(1);
+         final CountDownLatch latch = new CountDownLatch(1);
 
-            session.addFailoverListener(eventType -> {
-               if (eventType == FailoverEventType.FAILOVER_COMPLETED) {
-                  latch.countDown();
-               }
-            });
+         session.addFailoverListener(eventType -> {
+            if (eventType == FailoverEventType.FAILOVER_COMPLETED) {
+               latch.countDown();
+            }
+         });
 
-            sessions[0].getRemotingConnection().fail(new 
ActiveMQException("failure!"));
+         sessions[0].getRemotingConnection().fail(new 
ActiveMQException("failure!"));
 
-            assertTrue(latch.await(5, TimeUnit.SECONDS));
+         assertTrue(latch.await(5, TimeUnit.SECONDS));
 
-            sessions = countMetadata(server, "meta1", 1);
+         Wait.assertEquals(1, () -> getSessionsWithMetadata(server, 
"meta1").length);
 
-            assertEquals(1, sessions.length);
+         locator.close();
 
-            locator.close();
-         }
+         Wait.assertEquals(0, () -> getSessionsWithMetadata(server, 
"meta1").length);
       } finally {
          try {
             session.close();
@@ -195,19 +189,12 @@ public class ReconnectTest extends ActiveMQTestBase {
 
    }
 
-   private ServerSession[] countMetadata(ActiveMQServer server, String 
parameter, int expected) throws Exception {
+   private ServerSession[] getSessionsWithMetadata(ActiveMQServer server, 
String parameter) throws Exception {
       List<ServerSession> sessionList = new LinkedList<>();
 
-      for (int i = 0; i < 10 && sessionList.size() != expected; i++) {
-         sessionList.clear();
-         for (ServerSession sess : server.getSessions()) {
-            if (sess.getMetaData(parameter) != null) {
-               sessionList.add(sess);
-            }
-         }
-
-         if (sessionList.size() != expected) {
-            Thread.sleep(100);
+      for (ServerSession sess : server.getSessions()) {
+         if (sess.getMetaData(parameter) != null) {
+            sessionList.add(sess);
          }
       }
 
@@ -353,7 +340,7 @@ public class ReconnectTest extends ActiveMQTestBase {
       final long retryInterval = 50;
       final double retryMultiplier = 1d;
       final int reconnectAttempts = 1;
-      ServerLocator locator = 
createFactory(true).setCallTimeout(2000).setRetryInterval(retryInterval).setRetryIntervalMultiplier(retryMultiplier).setReconnectAttempts(reconnectAttempts).setConfirmationWindowSize(-1);
+      ServerLocator locator = 
createFactory(true).setCallTimeout(200).setRetryInterval(retryInterval).setRetryIntervalMultiplier(retryMultiplier).setReconnectAttempts(reconnectAttempts).setConfirmationWindowSize(-1);
       ClientSessionFactoryInternal sf = (ClientSessionFactoryInternal) 
createSessionFactory(locator);
       final CountDownLatch latch = new CountDownLatch(1);
       sf.addFailoverListener(eventType -> {
@@ -397,8 +384,11 @@ public class ReconnectTest extends ActiveMQTestBase {
       final int reconnectAttempts = 10;
       ServerLocator locator = 
createFactory(true).setCallTimeout(200).setRetryInterval(retryInterval).setRetryIntervalMultiplier(retryMultiplier).setReconnectAttempts(reconnectAttempts).setConfirmationWindowSize(-1);
       ClientSessionFactoryInternal sf = (ClientSessionFactoryInternal) 
createSessionFactory(locator);
+      runAfter(sf::close);
 
       ClientSessionInternal session = 
(ClientSessionInternal)sf.createSession(false, true, true);
+      runAfter(session::close);
+
       SimpleString queueName1 = SimpleString.of("my_queue_one");
       SimpleString addressName1 = SimpleString.of("my_address_one");
 
@@ -412,16 +402,22 @@ public class ReconnectTest extends ActiveMQTestBase {
 
       Wait.assertEquals(1, () -> getConsumerCount(server, session));
 
-      Set<ServerConsumer> serverConsumers = 
server.getSessionByID(session.getName()).getServerConsumers();
-      ServerConsumer serverConsumer = serverConsumers.iterator().next();
-      assertEquals(clientConsumer2.getConsumerContext().getId(), 
serverConsumer.getID());
-
+      Wait.assertTrue(() -> matchConsumers(server, session, clientConsumer2), 
5000, 100);
 
       session.close();
       sf.close();
       server.stop();
    }
 
+   boolean matchConsumers(ActiveMQServer server, ClientSessionInternal 
session, ClientConsumer clientConsumer) {
+      Set<ServerConsumer> serverConsumers = 
server.getSessionByID(session.getName()).getServerConsumers();
+      if (serverConsumers.size() != 1) {
+         return false;
+      }
+      ServerConsumer serverConsumer = serverConsumers.iterator().next();
+      return clientConsumer.getConsumerContext().getId() == 
serverConsumer.getID();
+   }
+
    private int getConsumerCount(ActiveMQServer server, ClientSessionInternal 
session) {
       ServerSession serverSession = server.getSessionByID(session.getName());
       if (serverSession == null) {


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
For further information, visit: https://activemq.apache.org/contact


Reply via email to