executorService.awaitTermination(10, TimeUnit.SECONDS);
String bspRoot = "/bsp";
ZooKeeperSyncClientImpl peerClient = (ZooKeeperSyncClientImpl) SyncServiceFactory
.getPeerSyncClient(configuration);
ZKSyncBSPMasterClient masterClient = (ZKSyncBSPMasterClient) SyncServiceFactory
.getMasterSyncClient(configuration);
masterClient.init(configuration);
Thread.sleep(100);
Log.info("Created master and client sync clients");
assertTrue(masterClient.hasKey(bspRoot));
Log.info("BSP root exists");
BSPJobID jobID = new BSPJobID("test1", 1);
masterClient.registerJob(jobID.toString());
TaskID taskId1 = new TaskID(jobID, 1);
TaskID taskId2 = new TaskID(jobID, 2);
TaskAttemptID task1 = new TaskAttemptID(taskId1, 1);
TaskAttemptID task2 = new TaskAttemptID(taskId2, 1);
int zkPort = BSPNetUtils.getFreePort(21815);
configuration.setInt(Constants.PEER_PORT, zkPort);
peerClient.init(configuration, jobID, task1);
peerClient.registerTask(jobID, "hamanode1", 5000L, task1);
peerClient.registerTask(jobID, "hamanode2", 5000L, task2);
peerClient.storeInformation(
peerClient.constructKey(jobID, "info", "level2"), new IntWritable(5),
true, null);
String[] names = peerClient.getAllPeerNames(task1.getJobID());
Log.info("Found child count = " + names.length);
assertEquals(2, names.length);