String jtId = "test";
TaskAttemptID attemptID = new TaskAttemptID(jtId, 1, TaskType.MAP, 0, 0);
Task task = new MapTask(null, attemptID, 0, null, 5);
mapLauncher.addToTaskQueue(new LaunchTaskAction(task));
// verify that task is added to runningTasks
TaskInProgress killTip = tt.runningTasks.get(attemptID);
assertNotNull(killTip);
// wait for a while for launcher to pick up the task
// this loop waits atmost for 30 seconds
for (int i = 0; i < 300; i++) {
if (mapLauncher.getNumWaitingTasksToLaunch() == 0) {
break;
}
UtilsForTests.waitFor(100);
}
assertEquals("Launcher didnt pick up the task " + attemptID + "to launch",
0, mapLauncher.getNumWaitingTasksToLaunch());
// Now, that launcher has picked up the task, it waits until all five slots
// are available. i.e. it waits for-ever
// lets kill the task so that map launcher comes out
tt.processKillTaskAction(new KillTaskAction(attemptID));
assertEquals(TaskStatus.State.KILLED, killTip.getRunState());
// launch another attempt which requires only one slot
TaskAttemptID runningAttemptID = new TaskAttemptID(jtId, 1, TaskType.MAP,
0, expectedLaunchAttemptId);
mapLauncher.addToTaskQueue(new LaunchTaskAction(new MapTask(null,
runningAttemptID, 0, null, 1)));
TaskInProgress runningTip = tt.runningTasks.get(runningAttemptID);
assertNotNull(runningTip);
// wait for a while for the task to be launched
// this loop waits at most for 30 seconds
for (int i = 0; i < 300; i++) {
if (runningTip.getRunState().equals(TaskStatus.State.RUNNING)) {
break;
}
UtilsForTests.waitFor(100);
}
// verify that the task went to running
assertEquals(TaskStatus.State.RUNNING, runningTip.getRunState());
}