@Test public void testGetShufflePath() throws Exception { ApplicationId appId = createTestAppId(); JobId jobId = JobId.newJobId(appId, 123456); assertEquals( "/dragon/job_458495849584_89894985_123456/shuffle", dragonZK.getShufflePath(jobId)); }
@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); }