Skip to content

Commit d2c57ed

Browse files
authored
Use 3 nodes for clustered all tests (#5406)
1 parent 3a5d1b7 commit d2c57ed

File tree

3 files changed

+46
-40
lines changed

3 files changed

+46
-40
lines changed

vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsyncMapTest.java

Lines changed: 22 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -19,20 +19,36 @@
1919
import io.vertx.core.spi.cluster.ClusterManager;
2020
import io.vertx.test.fakecluster.FakeClusterManager;
2121

22+
import java.util.concurrent.atomic.AtomicInteger;
23+
2224
/**
2325
* @author <a href="http://tfox.org">Tim Fox</a>
2426
*/
2527
public class ClusteredAsyncMapTest extends AsyncMapTest {
2628

27-
int pos;
29+
protected final int numNodes = 3;
30+
AtomicInteger pos = new AtomicInteger();
31+
32+
public void setUp() throws Exception {
33+
super.setUp();
34+
startNodes(numNodes);
35+
}
36+
37+
@Override
38+
protected ClusterManager getClusterManager() {
39+
return new FakeClusterManager();
40+
}
2841

2942
@Override
3043
protected Vertx getVertx() {
31-
Vertx vertx = vertices[pos];
32-
if (++pos == getNumNodes()) {
33-
pos = 0;
34-
}
35-
return vertx;
44+
int i = pos.incrementAndGet();
45+
i = mod(i, numNodes);
46+
return vertices[i];
47+
}
48+
49+
private int mod(int idx, int size) {
50+
int i = idx % size;
51+
return i < 0 ? i + size : i;
3652
}
3753

3854
@Test
@@ -59,17 +75,4 @@ public void testGetLocalAsyncMap() {
5975
await();
6076
}
6177

62-
public void setUp() throws Exception {
63-
super.setUp();
64-
startNodes(getNumNodes());
65-
}
66-
67-
protected int getNumNodes() {
68-
return 2;
69-
}
70-
71-
@Override
72-
protected ClusterManager getClusterManager() {
73-
return new FakeClusterManager();
74-
}
7578
}

vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredAsynchronousLockTest.java

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -30,19 +30,18 @@
3030
*/
3131
public class ClusteredAsynchronousLockTest extends AsynchronousLockTest {
3232

33-
@Override
34-
protected ClusterManager getClusterManager() {
35-
return new FakeClusterManager();
36-
}
37-
3833
protected final int numNodes = 3;
34+
AtomicInteger pos = new AtomicInteger();
3935

4036
public void setUp() throws Exception {
4137
super.setUp();
4238
startNodes(numNodes);
4339
}
4440

45-
AtomicInteger pos = new AtomicInteger();
41+
@Override
42+
protected ClusterManager getClusterManager() {
43+
return new FakeClusterManager();
44+
}
4645

4746
@Override
4847
protected Vertx getVertx() {

vertx-core/src/test/java/io/vertx/tests/shareddata/ClusteredSharedCounterTest.java

Lines changed: 19 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -19,21 +19,36 @@
1919
import io.vertx.core.spi.cluster.ClusterManager;
2020
import io.vertx.test.fakecluster.FakeClusterManager;
2121

22+
import java.util.concurrent.atomic.AtomicInteger;
23+
2224
/**
2325
* @author <a href="http://tfox.org">Tim Fox</a>
2426
*/
2527
public class ClusteredSharedCounterTest extends SharedCounterTest {
2628

29+
protected final int numNodes = 3;
30+
AtomicInteger pos = new AtomicInteger();
31+
32+
public void setUp() throws Exception {
33+
super.setUp();
34+
startNodes(numNodes);
35+
}
36+
2737
@Override
2838
protected ClusterManager getClusterManager() {
2939
return new FakeClusterManager();
3040
}
3141

32-
protected final int numNodes = 2;
42+
@Override
43+
protected Vertx getVertx() {
44+
int i = pos.incrementAndGet();
45+
i = mod(i, numNodes);
46+
return vertices[i];
47+
}
3348

34-
public void setUp() throws Exception {
35-
super.setUp();
36-
startNodes(numNodes);
49+
private int mod(int idx, int size) {
50+
int i = idx % size;
51+
return i < 0 ? i + size : i;
3752
}
3853

3954
@Test
@@ -57,15 +72,4 @@ public void testGetLocalCounter() {
5772
await();
5873
}
5974

60-
int pos;
61-
@Override
62-
protected Vertx getVertx() {
63-
Vertx vertx = vertices[pos];
64-
if (++pos == numNodes) {
65-
pos = 0;
66-
}
67-
return vertx;
68-
}
69-
70-
7175
}

0 commit comments

Comments
 (0)