use of io.atomix.cluster.messaging.impl.TestMessagingServiceFactory in project atomix by atomix.
the class DefaultClusterServiceTest method testClusterService.
@Test
public void testClusterService() throws Exception {
TestMessagingServiceFactory messagingServiceFactory = new TestMessagingServiceFactory();
ClusterMetadata clusterMetadata = buildClusterMetadata(1, 2, 3);
Node localNode1 = buildNode(1, Node.Type.CORE);
ManagedClusterService clusterService1 = new DefaultClusterService(localNode1, new TestClusterMetadataService(clusterMetadata), messagingServiceFactory.newMessagingService(localNode1.endpoint()).start().join());
Node localNode2 = buildNode(2, Node.Type.CORE);
ManagedClusterService clusterService2 = new DefaultClusterService(localNode2, new TestClusterMetadataService(clusterMetadata), messagingServiceFactory.newMessagingService(localNode2.endpoint()).start().join());
Node localNode3 = buildNode(3, Node.Type.CORE);
ManagedClusterService clusterService3 = new DefaultClusterService(localNode3, new TestClusterMetadataService(clusterMetadata), messagingServiceFactory.newMessagingService(localNode3.endpoint()).start().join());
assertNull(clusterService1.getNode(NodeId.from("1")));
assertNull(clusterService1.getNode(NodeId.from("2")));
assertNull(clusterService1.getNode(NodeId.from("3")));
CompletableFuture<ClusterService>[] futures = new CompletableFuture[3];
futures[0] = clusterService1.start();
futures[1] = clusterService2.start();
futures[2] = clusterService3.start();
CompletableFuture.allOf(futures).join();
Thread.sleep(1000);
assertEquals(3, clusterService1.getNodes().size());
assertEquals(3, clusterService2.getNodes().size());
assertEquals(3, clusterService3.getNodes().size());
assertEquals(Node.Type.CORE, clusterService1.getLocalNode().type());
assertEquals(Node.Type.CORE, clusterService1.getNode(NodeId.from("1")).type());
assertEquals(Node.Type.CORE, clusterService1.getNode(NodeId.from("2")).type());
assertEquals(Node.Type.CORE, clusterService1.getNode(NodeId.from("3")).type());
assertEquals(State.ACTIVE, clusterService1.getLocalNode().getState());
assertEquals(State.ACTIVE, clusterService1.getNode(NodeId.from("1")).getState());
assertEquals(State.ACTIVE, clusterService1.getNode(NodeId.from("2")).getState());
assertEquals(State.ACTIVE, clusterService1.getNode(NodeId.from("3")).getState());
Node dataNode = buildNode(4, Node.Type.DATA);
ManagedClusterService dataClusterService = new DefaultClusterService(dataNode, new TestClusterMetadataService(clusterMetadata), messagingServiceFactory.newMessagingService(dataNode.endpoint()).start().join());
assertEquals(State.INACTIVE, dataClusterService.getLocalNode().getState());
assertNull(dataClusterService.getNode(NodeId.from("1")));
assertNull(dataClusterService.getNode(NodeId.from("2")));
assertNull(dataClusterService.getNode(NodeId.from("3")));
assertNull(dataClusterService.getNode(NodeId.from("4")));
assertNull(dataClusterService.getNode(NodeId.from("5")));
dataClusterService.start().join();
Thread.sleep(1000);
assertEquals(4, clusterService1.getNodes().size());
assertEquals(4, clusterService2.getNodes().size());
assertEquals(4, clusterService3.getNodes().size());
assertEquals(4, dataClusterService.getNodes().size());
Node clientNode = buildNode(5, Node.Type.CLIENT);
ManagedClusterService clientClusterService = new DefaultClusterService(clientNode, new TestClusterMetadataService(clusterMetadata), messagingServiceFactory.newMessagingService(clientNode.endpoint()).start().join());
assertEquals(State.INACTIVE, clientClusterService.getLocalNode().getState());
assertNull(clientClusterService.getNode(NodeId.from("1")));
assertNull(clientClusterService.getNode(NodeId.from("2")));
assertNull(clientClusterService.getNode(NodeId.from("3")));
assertNull(clientClusterService.getNode(NodeId.from("4")));
assertNull(clientClusterService.getNode(NodeId.from("5")));
clientClusterService.start().join();
Thread.sleep(1000);
assertEquals(5, clusterService1.getNodes().size());
assertEquals(5, clusterService2.getNodes().size());
assertEquals(5, clusterService3.getNodes().size());
assertEquals(5, dataClusterService.getNodes().size());
assertEquals(5, clientClusterService.getNodes().size());
assertEquals(Node.Type.CLIENT, clientClusterService.getLocalNode().type());
assertEquals(Node.Type.CORE, clientClusterService.getNode(NodeId.from("1")).type());
assertEquals(Node.Type.CORE, clientClusterService.getNode(NodeId.from("2")).type());
assertEquals(Node.Type.CORE, clientClusterService.getNode(NodeId.from("3")).type());
assertEquals(Node.Type.DATA, clientClusterService.getNode(NodeId.from("4")).type());
assertEquals(Node.Type.CLIENT, clientClusterService.getNode(NodeId.from("5")).type());
assertEquals(State.ACTIVE, clientClusterService.getLocalNode().getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("1")).getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("2")).getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("3")).getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("4")).getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("5")).getState());
Thread.sleep(2500);
clusterService1.stop().join();
Thread.sleep(2500);
assertEquals(5, clusterService2.getNodes().size());
assertEquals(Node.Type.CORE, clusterService2.getNode(NodeId.from("1")).type());
assertEquals(State.INACTIVE, clusterService2.getNode(NodeId.from("1")).getState());
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("2")).getState());
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("3")).getState());
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("4")).getState());
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("5")).getState());
assertEquals(State.INACTIVE, clientClusterService.getNode(NodeId.from("1")).getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("2")).getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("3")).getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("4")).getState());
assertEquals(State.ACTIVE, clientClusterService.getNode(NodeId.from("5")).getState());
dataClusterService.stop().join();
Thread.sleep(2500);
assertEquals(4, clusterService2.getNodes().size());
assertEquals(State.INACTIVE, clusterService2.getNode(NodeId.from("1")).getState());
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("2")).getState());
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("3")).getState());
assertNull(clusterService2.getNode(NodeId.from("4")));
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("5")).getState());
clientClusterService.stop().join();
Thread.sleep(2500);
assertEquals(3, clusterService2.getNodes().size());
assertEquals(State.INACTIVE, clusterService2.getNode(NodeId.from("1")).getState());
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("2")).getState());
assertEquals(State.ACTIVE, clusterService2.getNode(NodeId.from("3")).getState());
assertNull(clusterService2.getNode(NodeId.from("4")));
assertNull(clusterService2.getNode(NodeId.from("5")));
}
use of io.atomix.cluster.messaging.impl.TestMessagingServiceFactory in project atomix by atomix.
the class DefaultClusterMetadataServiceTest method testSingleNodeBootstrap.
@Test
public void testSingleNodeBootstrap() throws Exception {
TestMessagingServiceFactory messagingServiceFactory = new TestMessagingServiceFactory();
ClusterMetadata clusterMetadata = buildClusterMetadata(1);
Node localNode1 = buildNode(1, Node.Type.CORE);
ManagedClusterMetadataService metadataService1 = new DefaultClusterMetadataService(clusterMetadata, messagingServiceFactory.newMessagingService(localNode1.endpoint()).start().join());
metadataService1.start().join();
assertEquals(1, metadataService1.getMetadata().bootstrapNodes().size());
Node localNode2 = buildNode(2, Node.Type.CORE);
ManagedClusterMetadataService metadataService2 = new DefaultClusterMetadataService(clusterMetadata, messagingServiceFactory.newMessagingService(localNode2.endpoint()).start().join());
metadataService2.start().join();
metadataService2.addNode(localNode2);
assertEquals(2, metadataService2.getMetadata().bootstrapNodes().size());
}
use of io.atomix.cluster.messaging.impl.TestMessagingServiceFactory in project atomix by atomix.
the class DefaultClusterMetadataServiceTest method testClusterMetadataService.
@Test
public void testClusterMetadataService() throws Exception {
TestMessagingServiceFactory messagingServiceFactory = new TestMessagingServiceFactory();
ClusterMetadata clusterMetadata = buildClusterMetadata(1, 2, 3);
Node localNode1 = buildNode(1, Node.Type.CORE);
ManagedClusterMetadataService metadataService1 = new DefaultClusterMetadataService(clusterMetadata, messagingServiceFactory.newMessagingService(localNode1.endpoint()).start().join());
Node localNode2 = buildNode(2, Node.Type.CORE);
ManagedClusterMetadataService metadataService2 = new DefaultClusterMetadataService(clusterMetadata, messagingServiceFactory.newMessagingService(localNode2.endpoint()).start().join());
Node localNode3 = buildNode(3, Node.Type.CORE);
ManagedClusterMetadataService metadataService3 = new DefaultClusterMetadataService(clusterMetadata, messagingServiceFactory.newMessagingService(localNode3.endpoint()).start().join());
List<CompletableFuture<ClusterMetadataService>> futures = new ArrayList<>();
futures.add(metadataService1.start());
futures.add(metadataService2.start());
futures.add(metadataService3.start());
Futures.allOf(futures).join();
assertEquals(3, metadataService1.getMetadata().bootstrapNodes().size());
assertEquals(3, metadataService2.getMetadata().bootstrapNodes().size());
assertEquals(3, metadataService3.getMetadata().bootstrapNodes().size());
Node localNode4 = buildNode(4, Node.Type.CORE);
ManagedClusterMetadataService metadataService4 = new DefaultClusterMetadataService(clusterMetadata, messagingServiceFactory.newMessagingService(localNode4.endpoint()).start().join());
metadataService4.start().join();
assertEquals(3, metadataService4.getMetadata().bootstrapNodes().size());
TestClusterMetadataEventListener localEventListener = new TestClusterMetadataEventListener();
metadataService4.addListener(localEventListener);
TestClusterMetadataEventListener remoteEventListener1 = new TestClusterMetadataEventListener();
metadataService1.addListener(remoteEventListener1);
TestClusterMetadataEventListener remoteEventListener2 = new TestClusterMetadataEventListener();
metadataService2.addListener(remoteEventListener2);
TestClusterMetadataEventListener remoteEventListener3 = new TestClusterMetadataEventListener();
metadataService3.addListener(remoteEventListener3);
metadataService4.addNode(localNode4);
assertEquals(4, metadataService4.getMetadata().bootstrapNodes().size());
assertEquals(4, localEventListener.event().subject().bootstrapNodes().size());
assertEquals(4, remoteEventListener1.event().subject().bootstrapNodes().size());
assertEquals(4, metadataService1.getMetadata().bootstrapNodes().size());
assertEquals(4, remoteEventListener2.event().subject().bootstrapNodes().size());
assertEquals(4, metadataService2.getMetadata().bootstrapNodes().size());
assertEquals(4, remoteEventListener3.event().subject().bootstrapNodes().size());
assertEquals(4, metadataService3.getMetadata().bootstrapNodes().size());
Node localNode5 = buildNode(5, Node.Type.CORE);
ManagedClusterMetadataService metadataService5 = new DefaultClusterMetadataService(clusterMetadata, messagingServiceFactory.newMessagingService(localNode5.endpoint()).start().join());
metadataService5.start().join();
assertEquals(4, metadataService5.getMetadata().bootstrapNodes().size());
}
Aggregations