use of com.hazelcast.core.ClientService in project neo4j by neo4j.
the class HazelcastClientTest method shouldRegisterReadReplicaInTopology.
@Test
public void shouldRegisterReadReplicaInTopology() throws Throwable {
// given
com.hazelcast.core.Cluster cluster = mock(Cluster.class);
Set<Member> members = asSet(makeMember(1));
when(cluster.getMembers()).thenReturn(members);
Endpoint endpoint = mock(Endpoint.class);
when(endpoint.getUuid()).thenReturn("12345");
Client client = mock(Client.class);
final String clientId = "12345";
when(client.getUuid()).thenReturn(clientId);
ClientService clientService = mock(ClientService.class);
when(clientService.getConnectedClients()).thenReturn(asSet(client));
HazelcastMap hazelcastMap = new HazelcastMap();
HazelcastMultiMap hazelcastMultiMap = new HazelcastMultiMap();
HazelcastInstance hazelcastInstance = mock(HazelcastInstance.class);
when(hazelcastInstance.getAtomicReference(anyString())).thenReturn(mock(IAtomicReference.class));
when(hazelcastInstance.getMap(anyString())).thenReturn(hazelcastMap);
when(hazelcastInstance.getMultiMap(anyString())).thenReturn(hazelcastMultiMap);
when(hazelcastInstance.getLocalEndpoint()).thenReturn(endpoint);
when(hazelcastInstance.getExecutorService(anyString())).thenReturn(new StubExecutorService());
when(hazelcastInstance.getCluster()).thenReturn(cluster);
when(hazelcastInstance.getClientService()).thenReturn(clientService);
HazelcastConnector connector = mock(HazelcastConnector.class);
when(connector.connectToHazelcast()).thenReturn(hazelcastInstance);
OnDemandJobScheduler jobScheduler = new OnDemandJobScheduler();
HazelcastClient hazelcastClient = new HazelcastClient(connector, jobScheduler, NullLogProvider.getInstance(), config(), myself);
// when
hazelcastClient.start();
jobScheduler.runJob();
// then
assertEquals(1, hazelcastMap.size());
}
use of com.hazelcast.core.ClientService in project neo4j by neo4j.
the class HazelcastClientTest method shouldRemoveReadReplicasOnGracefulShutdown.
@Test
public void shouldRemoveReadReplicasOnGracefulShutdown() throws Throwable {
// given
com.hazelcast.core.Cluster cluster = mock(Cluster.class);
Set<Member> members = asSet(makeMember(1));
when(cluster.getMembers()).thenReturn(members);
Endpoint endpoint = mock(Endpoint.class);
when(endpoint.getUuid()).thenReturn("12345");
Client client = mock(Client.class);
final String clientId = "12345";
when(client.getUuid()).thenReturn(clientId);
ClientService clientService = mock(ClientService.class);
when(clientService.getConnectedClients()).thenReturn(asSet(client));
HazelcastMap hazelcastMap = new HazelcastMap();
HazelcastInstance hazelcastInstance = mock(HazelcastInstance.class);
when(hazelcastInstance.getAtomicReference(anyString())).thenReturn(mock(IAtomicReference.class));
when(hazelcastInstance.getMap(anyString())).thenReturn(hazelcastMap);
when(hazelcastInstance.getMultiMap(anyString())).thenReturn(new HazelcastMultiMap());
when(hazelcastInstance.getLocalEndpoint()).thenReturn(endpoint);
when(hazelcastInstance.getExecutorService(anyString())).thenReturn(new StubExecutorService());
when(hazelcastInstance.getCluster()).thenReturn(cluster);
when(hazelcastInstance.getClientService()).thenReturn(clientService);
HazelcastConnector connector = mock(HazelcastConnector.class);
when(connector.connectToHazelcast()).thenReturn(hazelcastInstance);
OnDemandJobScheduler jobScheduler = new OnDemandJobScheduler();
HazelcastClient hazelcastClient = new HazelcastClient(connector, jobScheduler, NullLogProvider.getInstance(), config(), myself);
hazelcastClient.start();
jobScheduler.runJob();
// when
hazelcastClient.stop();
// then
assertEquals(0, hazelcastMap.size());
}
Aggregations