@Test
public void testDiagnostics_TAUpdate(){
TezTaskID taskId = getNewTaskID();
scheduleTaskAttempt(taskId);
mockTask.handle(new TaskEventTAUpdate(mockTask.getLastAttempt().getID(), TaskEventType.T_ATTEMPT_KILLED));
assertEquals(1, mockTask.getDiagnostics().size());
assertEquals("TaskAttempt 0 killed", mockTask.getDiagnostics().get(0));
launchTaskAttempt(mockTask.getLastAttempt().getID());
mockTask.getLastAttempt().handle(new TaskAttemptEventDiagnosticsUpdate(mockTask.getLastAttempt().getID(), "diagnostics of test"));
mockTask.handle(new TaskEventTAUpdate(mockTask.getLastAttempt().getID(), TaskEventType.T_ATTEMPT_FAILED));
assertEquals(2, mockTask.getDiagnostics().size());
assertEquals("TaskAttempt 1 failed, info=[diagnostics of test]", mockTask.getDiagnostics().get(1));
}