MAPREDUCE-6675. TestJobImpl.testUnusableNode failed (haibochen via rkanter)

This commit is contained in:
Robert Kanter 2016-05-04 22:42:21 -07:00
parent ecce3b7d53
commit 9d3fcdfbb3

View File

@ -531,7 +531,7 @@ public void testUnusableNodeTransition() throws Exception {
Configuration conf = new Configuration(); Configuration conf = new Configuration();
conf.set(MRJobConfig.MR_AM_STAGING_DIR, stagingDir); conf.set(MRJobConfig.MR_AM_STAGING_DIR, stagingDir);
conf.setInt(MRJobConfig.NUM_REDUCES, 1); conf.setInt(MRJobConfig.NUM_REDUCES, 1);
AsyncDispatcher dispatcher = new AsyncDispatcher(); DrainDispatcher dispatcher = new DrainDispatcher();
dispatcher.init(conf); dispatcher.init(conf);
dispatcher.start(); dispatcher.start();
CyclicBarrier syncBarrier = new CyclicBarrier(2); CyclicBarrier syncBarrier = new CyclicBarrier(2);
@ -608,6 +608,7 @@ public void handle(TaskAttemptEvent event) {
NodeReport secondMapperNodeReport = nodeReports.get(1); NodeReport secondMapperNodeReport = nodeReports.get(1);
job.handle(new JobUpdatedNodesEvent(job.getID(), job.handle(new JobUpdatedNodesEvent(job.getID(),
Collections.singletonList(firstMapperNodeReport))); Collections.singletonList(firstMapperNodeReport)));
dispatcher.await();
// complete the reducer // complete the reducer
for (TaskId taskId: job.tasks.keySet()) { for (TaskId taskId: job.tasks.keySet()) {
if (taskId.getTaskType() == TaskType.REDUCE) { if (taskId.getTaskType() == TaskType.REDUCE) {