Examples with LeaderLatch org.apache.curator.framework.recipes.leader.LeaderLatch used on opensource projects

Example 16 with LeaderLatch

use of org.apache.curator.framework.recipes.leader.LeaderLatch in project incubator-atlas by apache.

the class ActiveInstanceElectorServiceTest method testElectionIsRejoinedWhenStateUpdateFails.

@Test
public void testElectionIsRejoinedWhenStateUpdateFails() throws Exception {
    when(configuration.containsKey(HAConfiguration.ATLAS_SERVER_HA_ENABLED_KEY)).thenReturn(true);
    when(configuration.getBoolean(HAConfiguration.ATLAS_SERVER_HA_ENABLED_KEY)).thenReturn(true);
    when(configuration.getStringArray(HAConfiguration.ATLAS_SERVER_IDS)).thenReturn(new String[] { "id1" });
    when(configuration.getString(HAConfiguration.ATLAS_SERVER_ADDRESS_PREFIX + "id1")).thenReturn("127.0.0.1:21000");
    when(configuration.getString(HAConfiguration.ATLAS_SERVER_HA_ZK_ROOT_KEY, HAConfiguration.ATLAS_SERVER_ZK_ROOT_DEFAULT)).thenReturn(HAConfiguration.ATLAS_SERVER_ZK_ROOT_DEFAULT);
    LeaderLatch leaderLatch = mock(LeaderLatch.class);
    when(curatorFactory.leaderLatchInstance("id1", HAConfiguration.ATLAS_SERVER_ZK_ROOT_DEFAULT)).thenReturn(leaderLatch);
    doThrow(new AtlasBaseException()).when(activeInstanceState).update("id1");
    ActiveInstanceElectorService activeInstanceElectorService = new ActiveInstanceElectorService(configuration, new HashSet<ActiveStateChangeHandler>(), curatorFactory, activeInstanceState, serviceState);
    activeInstanceElectorService.start();
    activeInstanceElectorService.isLeader();
    InOrder inOrder = inOrder(leaderLatch, curatorFactory);
    inOrder.verify(leaderLatch).close();
    inOrder.verify(curatorFactory).leaderLatchInstance("id1", HAConfiguration.ATLAS_SERVER_ZK_ROOT_DEFAULT);
    inOrder.verify(leaderLatch).addListener(activeInstanceElectorService);
    inOrder.verify(leaderLatch).start();
}