Examples with InstanceSpec org.apache.curator.test.InstanceSpec used on opensource projects

Example 1 with InstanceSpec

use of org.apache.curator.test.InstanceSpec in project helios by spotify.

the class ZooKeeperTestingClusterManager method createPeers.

private List<InstanceSpec> createPeers(final int numPeers) {
    final ImmutableList.Builder<InstanceSpec> peers = ImmutableList.builder();
    for (int i = 0; i < numPeers; i++) {
        final int port = temporaryPorts.localPort("zk-client" + i);
        final int electionPort = temporaryPorts.localPort("zk-elect" + i);
        final int quorumPort = temporaryPorts.localPort("zk-quorum" + i);
        final Path peerDir = peerDir(i);
        try {
            Files.createDirectory(peerDir);
        } catch (IOException e) {
            throw Throwables.propagate(e);
        }
        final InstanceSpec spec = new InstanceSpec(peerDir.toFile(), port, electionPort, quorumPort, true, i);
        peers.add(spec);
    }
    return peers.build();
}

Example 2 with InstanceSpec

use of org.apache.curator.test.InstanceSpec in project ignite by apache.

the class ZookeeperIpFinderTest method testFourNodesKillRestartZookeeper.

/**
     * @throws Exception If failed.
     */
public void testFourNodesKillRestartZookeeper() throws Exception {
    allowDuplicateRegistrations = false;
    // start 4 nodes
    System.setProperty(TcpDiscoveryZookeeperIpFinder.PROP_ZK_CONNECTION_STRING, zkCluster.getConnectString());
    startGrids(4);
    // wait until all grids are started
    waitForRemoteNodes(grid(0), 3);
    // each node will only register itself
    assertEquals(4, zkCurator.getChildren().forPath(SERVICES_IGNITE_ZK_PATH).size());
    // remember ZK server configuration and stop the cluster
    Collection<InstanceSpec> instances = zkCluster.getInstances();
    zkCluster.stop();
    Thread.sleep(1000);
    // start the cluster with the previous configuration
    zkCluster = new TestingCluster(instances);
    zkCluster.start();
    // block the client until connected
    zkCurator.blockUntilConnected();
    // check that the nodes have registered again
    assertEquals(4, zkCurator.getChildren().forPath(SERVICES_IGNITE_ZK_PATH).size());
    // stop all grids
    stopAllGrids();
    assertTrue(GridTestUtils.waitForCondition(new GridAbsPredicate() {

        @Override
        public boolean apply() {
            try {
                return 0 == zkCurator.getChildren().forPath(SERVICES_IGNITE_ZK_PATH).size();
            } catch (Exception e) {
                U.error(log, "Failed to wait for zk condition", e);
                return false;
            }
        }
    }, 20000));
}

Example 3 with InstanceSpec

use of org.apache.curator.test.InstanceSpec in project hadoop by apache.

the class TestLeaderElectorService method testZKClusterDown.

// 1. rm1 active
// 2. restart zk cluster
// 3. rm1 will first relinquish leadership and re-acquire leadership
@Test
public void testZKClusterDown() throws Exception {
    rm1 = startRM("rm1", HAServiceState.ACTIVE);
    // stop zk cluster
    zkCluster.stop();
    waitFor(rm1, HAServiceState.STANDBY);
    Collection<InstanceSpec> instanceSpecs = zkCluster.getInstances();
    zkCluster = new TestingCluster(instanceSpecs);
    zkCluster.start();
    // rm becomes active again
    waitFor(rm1, HAServiceState.ACTIVE);
}

Example 4 with InstanceSpec

use of org.apache.curator.test.InstanceSpec in project hadoop by apache.

the class TestLeaderElectorService method testKillZKInstance.

// 1. rm1 active
// 2. rm2 standby
// 3. kill the current connected zk instance
// 4. either rm1 or rm2 will become active.
@Test
public void testKillZKInstance() throws Exception {
    rm1 = startRM("rm1", HAServiceState.ACTIVE);
    rm2 = startRM("rm2", HAServiceState.STANDBY);
    CuratorBasedElectorService service = (CuratorBasedElectorService) rm1.getRMContext().getLeaderElectorService();
    ZooKeeper zkClient = service.getCuratorClient().getZookeeperClient().getZooKeeper();
    InstanceSpec connectionInstance = zkCluster.findConnectionInstance(zkClient);
    zkCluster.killServer(connectionInstance);
    // wait for rm1 or rm2 to be active by randomness
    GenericTestUtils.waitFor(new Supplier<Boolean>() {

        @Override
        public Boolean get() {
            try {
                HAServiceState rm1State = rm1.getAdminService().getServiceStatus().getState();
                HAServiceState rm2State = rm2.getAdminService().getServiceStatus().getState();
                return (rm1State.equals(HAServiceState.ACTIVE) && rm2State.equals(HAServiceState.STANDBY)) || (rm1State.equals(HAServiceState.STANDBY) && rm2State.equals(HAServiceState.ACTIVE));
            } catch (IOException e) {
            }
            return false;
        }
    }, 2000, 15000);
}

Example 5 with InstanceSpec

use of org.apache.curator.test.InstanceSpec in project zipkin by openzipkin.

the class SampleRateUpdateGuardTest method shouldStillBeOneLeaderAfterZKFailure.

@Test
public void shouldStillBeOneLeaderAfterZKFailure() throws Exception {
    SampleRateUpdateGuard guard1 = guard(new ZooKeeperCollectorSampler.Builder());
    SampleRateUpdateGuard guard2 = guard(new ZooKeeperCollectorSampler.Builder());
    waitForALeader(guard1, guard2);
    SampleRateUpdateGuard leader = guard1.latch.hasLeadership() ? guard1 : guard2;
    InstanceSpec instance = zookeeper.cluster.findConnectionInstance(leader.client.getZookeeperClient().getZooKeeper());
    zookeeper.cluster.killServer(instance);
    waitForALeader(guard1, guard2);
    assertThat(guard1.latch.hasLeadership() ^ guard2.latch.hasLeadership()).withFailMessage("After a ZK node failure, one guard, but not both, should be the leader").isTrue();
}