public virtual void TestTaskStartTimes()
{
TaskId taskId = Org.Mockito.Mockito.Mock <TaskId>();
JobHistoryParser.TaskInfo taskInfo = Org.Mockito.Mockito.Mock <JobHistoryParser.TaskInfo
>();
IDictionary <TaskAttemptID, JobHistoryParser.TaskAttemptInfo> taskAttempts = new SortedDictionary
<TaskAttemptID, JobHistoryParser.TaskAttemptInfo>();
TaskAttemptID id = new TaskAttemptID("0", 0, TaskType.Map, 0, 0);
JobHistoryParser.TaskAttemptInfo info = Org.Mockito.Mockito.Mock <JobHistoryParser.TaskAttemptInfo
>();
Org.Mockito.Mockito.When(info.GetAttemptId()).ThenReturn(id);
Org.Mockito.Mockito.When(info.GetStartTime()).ThenReturn(10l);
taskAttempts[id] = info;
id = new TaskAttemptID("1", 0, TaskType.Map, 1, 1);
info = Org.Mockito.Mockito.Mock <JobHistoryParser.TaskAttemptInfo>();
Org.Mockito.Mockito.When(info.GetAttemptId()).ThenReturn(id);
Org.Mockito.Mockito.When(info.GetStartTime()).ThenReturn(20l);
taskAttempts[id] = info;
Org.Mockito.Mockito.When(taskInfo.GetAllTaskAttempts()).ThenReturn(taskAttempts);
CompletedTask task = new CompletedTask(taskId, taskInfo);
TaskReport report = task.GetReport();
// Make sure the startTime returned by report is the lesser of the
// attempy launch times
NUnit.Framework.Assert.IsTrue(report.GetStartTime() == 10);
}
internal CompletedTaskAttempt(TaskId taskId, JobHistoryParser.TaskAttemptInfo attemptInfo
)
{
this.attemptInfo = attemptInfo;
this.attemptId = TypeConverter.ToYarn(attemptInfo.GetAttemptId());
if (attemptInfo.GetTaskStatus() != null)
{
this.state = TaskAttemptState.ValueOf(attemptInfo.GetTaskStatus());
}
else
{
this.state = TaskAttemptState.Killed;
localDiagMessage = "Attmpt state missing from History : marked as KILLED";
diagnostics.AddItem(localDiagMessage);
}
if (attemptInfo.GetError() != null)
{
diagnostics.AddItem(attemptInfo.GetError());
}
}
/// <summary>test some methods of CompletedTaskAttempt</summary>
public virtual void TestCompletedTaskAttempt()
{
JobHistoryParser.TaskAttemptInfo attemptInfo = Org.Mockito.Mockito.Mock <JobHistoryParser.TaskAttemptInfo
>();
Org.Mockito.Mockito.When(attemptInfo.GetRackname()).ThenReturn("Rackname");
Org.Mockito.Mockito.When(attemptInfo.GetShuffleFinishTime()).ThenReturn(11L);
Org.Mockito.Mockito.When(attemptInfo.GetSortFinishTime()).ThenReturn(12L);
Org.Mockito.Mockito.When(attemptInfo.GetShufflePort()).ThenReturn(10);
JobID jobId = new JobID("12345", 0);
TaskID taskId = new TaskID(jobId, TaskType.Reduce, 0);
TaskAttemptID taskAttemptId = new TaskAttemptID(taskId, 0);
Org.Mockito.Mockito.When(attemptInfo.GetAttemptId()).ThenReturn(taskAttemptId);
CompletedTaskAttempt taskAttemt = new CompletedTaskAttempt(null, attemptInfo);
NUnit.Framework.Assert.AreEqual("Rackname", taskAttemt.GetNodeRackName());
NUnit.Framework.Assert.AreEqual(Phase.Cleanup, taskAttemt.GetPhase());
NUnit.Framework.Assert.IsTrue(taskAttemt.IsFinished());
NUnit.Framework.Assert.AreEqual(11L, taskAttemt.GetShuffleFinishTime());
NUnit.Framework.Assert.AreEqual(12L, taskAttemt.GetSortFinishTime());
NUnit.Framework.Assert.AreEqual(10, taskAttemt.GetShufflePort());
}
