From b2cd98e6a581edc0e66d0da73cd8f7428c0f3546 Mon Sep 17 00:00:00 2001
From: Frantisek Hartman <frant.hartm@gmail.com>
Date: Fri, 22 Nov 2024 14:36:55 +0100
Subject: [PATCH] Use 3 nodes for clustered all tests

---
 .../shareddata/ClusteredAsyncMapTest.java     | 41 ++++++++++---------
 .../ClusteredAsynchronousLockTest.java        | 11 +++--
 .../ClusteredSharedCounterTest.java           | 34 ++++++++-------
 3 files changed, 46 insertions(+), 40 deletions(-)

diff --git a/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsyncMapTest.java b/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsyncMapTest.java
index acc70d73af7..d2653f86d08 100644
--- a/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsyncMapTest.java
+++ b/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsyncMapTest.java
@@ -19,20 +19,36 @@
 import io.vertx.core.spi.cluster.ClusterManager;
 import io.vertx.test.fakecluster.FakeClusterManager;
 
+import java.util.concurrent.atomic.AtomicInteger;
+
 /**
  * @author <a href="http://tfox.org">Tim Fox</a>
  */
 public class ClusteredAsyncMapTest extends AsyncMapTest {
 
-  int pos;
+  protected final int numNodes = 3;
+  AtomicInteger pos = new AtomicInteger();
+
+  public void setUp() throws Exception {
+    super.setUp();
+    startNodes(numNodes);
+  }
+
+  @Override
+  protected ClusterManager getClusterManager() {
+    return new FakeClusterManager();
+  }
 
   @Override
   protected Vertx getVertx() {
-    Vertx vertx = vertices[pos];
-    if (++pos == getNumNodes()) {
-      pos = 0;
-    }
-    return vertx;
+    int i = pos.incrementAndGet();
+    i = mod(i, numNodes);
+    return vertices[i];
+  }
+
+  private int mod(int idx, int size) {
+    int i = idx % size;
+    return i < 0 ? i + size : i;
   }
 
   @Test
@@ -59,17 +75,4 @@ public void testGetLocalAsyncMap() {
     await();
   }
 
-  public void setUp() throws Exception {
-    super.setUp();
-    startNodes(getNumNodes());
-  }
-
-  protected int getNumNodes() {
-    return 2;
-  }
-
-  @Override
-  protected ClusterManager getClusterManager() {
-    return new FakeClusterManager();
-  }
 }
diff --git a/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsynchronousLockTest.java b/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsynchronousLockTest.java
index a1db8562ba7..1bba8078414 100644
--- a/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsynchronousLockTest.java
+++ b/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsynchronousLockTest.java
@@ -30,19 +30,18 @@
  */
 public class ClusteredAsynchronousLockTest extends AsynchronousLockTest {
 
-  @Override
-  protected ClusterManager getClusterManager() {
-    return new FakeClusterManager();
-  }
-
   protected final int numNodes = 3;
+  AtomicInteger pos = new AtomicInteger();
 
   public void setUp() throws Exception {
     super.setUp();
     startNodes(numNodes);
   }
 
-  AtomicInteger pos = new AtomicInteger();
+  @Override
+  protected ClusterManager getClusterManager() {
+    return new FakeClusterManager();
+  }
 
   @Override
   protected Vertx getVertx() {
diff --git a/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredSharedCounterTest.java b/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredSharedCounterTest.java
index f7a84a1be09..114208e5496 100644
--- a/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredSharedCounterTest.java
+++ b/vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredSharedCounterTest.java
@@ -19,21 +19,36 @@
 import io.vertx.core.spi.cluster.ClusterManager;
 import io.vertx.test.fakecluster.FakeClusterManager;
 
+import java.util.concurrent.atomic.AtomicInteger;
+
 /**
  * @author <a href="http://tfox.org">Tim Fox</a>
  */
 public class ClusteredSharedCounterTest extends SharedCounterTest {
 
+  protected final int numNodes = 3;
+  AtomicInteger pos = new AtomicInteger();
+
+  public void setUp() throws Exception {
+    super.setUp();
+    startNodes(numNodes);
+  }
+
   @Override
   protected ClusterManager getClusterManager() {
     return new FakeClusterManager();
   }
 
-  protected final int numNodes = 2;
+  @Override
+  protected Vertx getVertx() {
+    int i = pos.incrementAndGet();
+    i = mod(i, numNodes);
+    return vertices[i];
+  }
 
-  public void setUp() throws Exception {
-    super.setUp();
-    startNodes(numNodes);
+  private int mod(int idx, int size) {
+    int i = idx % size;
+    return i < 0 ? i + size : i;
   }
 
   @Test
@@ -57,15 +72,4 @@ public void testGetLocalCounter() {
     await();
   }
 
-  int pos;
-  @Override
-  protected Vertx getVertx() {
-    Vertx vertx = vertices[pos];
-    if (++pos == numNodes) {
-      pos = 0;
-    }
-    return vertx;
-  }
-
-
 }