@Test
public void testGetAvailableNodes() throws Exception {
NodeId nodeId = createTestNodeId();
dragonZK.registerNodeManager(nodeId);
List<NodeId> nodeIdList = dragonZK.getAvailableNodes();
assertEquals(nodeId, nodeIdList.get(0));
}
@Test
public void testRegisterNodeManager() throws Exception {
NodeId nodeId = createTestNodeId();
dragonZK.registerNodeManager(nodeId);
Stat stat = zkClient.checkExists().forPath("/dragon/nodemanagers/ServerHost:2345");
assertNotNull(stat);
}
@Test
public void testWatchNodeManager() throws Exception {
NodeId nodeId = createTestNodeId();
dragonZK.registerNodeManager(nodeId);
nodeDied = false;
dragonZK.watchNodeManager(
nodeId,
new CuratorWatcher() {
@Override
public void process(WatchedEvent event) throws Exception {
nodeDied = true;
}
});
zkClient.delete().forPath("/dragon/nodemanagers/ServerHost:2345");
Thread.sleep(1000);
assertTrue(nodeDied);
}
