File tree Expand file tree Collapse file tree 3 files changed +46
-40
lines changed
vertx-core/src/test/java/io/vertx/tests/shareddata Expand file tree Collapse file tree 3 files changed +46
-40
lines changed Original file line number Diff line number Diff line change 1919import io .vertx .core .spi .cluster .ClusterManager ;
2020import 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 */
2527public 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}
Original file line number Diff line number Diff line change 3030 */
3131public 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 () {
Original file line number Diff line number Diff line change 1919import io .vertx .core .spi .cluster .ClusterManager ;
2020import 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 */
2527public 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}
You can’t perform that action at this time.
0 commit comments