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