private void configureMocks() {
firstReduceStatus = new ReduceTaskStatus(attemptID, 0.0f, 1,
TaskStatus.State.RUNNING, "", "RUNNING", "", TaskStatus.Phase.SHUFFLE,
new Counters());
firstReduceStatus.addFetchFailedMap(TaskAttemptID.forName(ATTEMPT1_ID));
secondReduceStatus = new ReduceTaskStatus(attemptID, 0.0f, 1,
TaskStatus.State.RUNNING, "", "RUNNING", "", TaskStatus.Phase.SORT,
new Counters());
secondReduceStatus.addFetchFailedMap(TaskAttemptID.forName(ATTEMPT2_ID));
thirdReduceStatus = new ReduceTaskStatus(attemptID, 0.0f, 1,
TaskStatus.State.RUNNING, "", "RUNNING", "",
TaskStatus.Phase.REDUCE, new Counters());
when(dispatcher.getEventHandler()).thenReturn(ea);
when(appCtx.getEventHandler()).thenReturn(ea);
policy = new CheckpointAMPreemptionPolicy();
policy.init(appCtx);
listener = new MockTaskAttemptListenerImpl(appCtx, secret,
rmHeartbeatHandler, hbHandler, policy);
id = new JVMId("foo", 1, true, 1);
wid = new WrappedJvmID(id.getJobId(), id.isMap, id.getId());
attemptID = new TaskAttemptID("1", 1, TaskType.MAP, 1, 1);
attemptId = TypeConverter.toYarn(attemptID);
}