public void TestSparkJobInfo()
{
const int jobId = 65536;
int[] stageIds = new[] { 100, 102, 104 };
const string status = "RUNNING";
// arrange
Mock <IStatusTrackerProxy> statusTrackerProxy = new Mock <IStatusTrackerProxy>();
var expectedJobInfo = new SparkJobInfo(jobId, stageIds, status);
statusTrackerProxy.Setup(m => m.GetJobInfo(It.IsAny <int>())).Returns(expectedJobInfo);
var tracker = new StatusTracker(statusTrackerProxy.Object);
// act
SparkJobInfo jobInfo = tracker.GetJobInfo(jobId);
// assert
Assert.IsNotNull(jobInfo);
Assert.AreEqual(jobId, jobInfo.JobId);
Assert.AreEqual(stageIds, jobInfo.StageIds);
Assert.AreEqual(status, jobInfo.Status);
}
public void TestSparkJobInfo(
[Values(JobExecutionStatus.Failed, JobExecutionStatus.Running, JobExecutionStatus.Succeeded, JobExecutionStatus.Unknown)] JobExecutionStatus status
)
{
const int jobId = 65536;
int[] stageIds = new[] { 100, 102, 104 };
// arrange
Mock <IStatusTrackerProxy> statusTrackerProxy = new Mock <IStatusTrackerProxy>();
var expectedJobInfo = new SparkJobInfo(jobId, stageIds, status);
statusTrackerProxy.Setup(m => m.GetJobInfo(It.IsAny <int>())).Returns(expectedJobInfo);
var tracker = new StatusTracker(statusTrackerProxy.Object);
// act
SparkJobInfo jobInfo = tracker.GetJobInfo(jobId);
// assert
Assert.IsNotNull(jobInfo);
Assert.AreEqual(jobId, jobInfo.JobId);
Assert.AreEqual(stageIds, jobInfo.StageIds);
Assert.AreEqual(status, jobInfo.Status);
}
