@Test
public void testGetShuffleNodeByTaskId() throws Exception {
JobId jobId = JobId.newJobId(createTestAppId(), 123456);
TaskId taskId = TaskId.newTaskId(jobId, 34565, TaskType.MAP);
DragonZooKeeper.NodeData node = new DragonZooKeeper.NodeData();
node.nodeId = createTestNodeId();
node.taskId = taskId;
if (zkClient.checkExists().forPath("/dragon/job_458495849584_89894985_123456/shuffle")
== null) {
dragonZK.createShufflePath(jobId);
}
PathChildrenCache shuffleNodeCache =
new PathChildrenCache(zkClient, dragonZK.getShufflePath(jobId), true);
dragonZK.setShuffleNodeCache(shuffleNodeCache);
dragonZK.createShuffleNode(jobId, newArrayList(node));
Thread.sleep(1000);
NodeId nodeId = dragonZK.getShuffleNodeByTaskId(jobId, taskId);
assertNotNull(nodeId);
assertEquals(node.nodeId, nodeId);
}
