public JobGetResponse MockJobCompletion(string jobId, JobCompletionType type)
{
var jobResponse = new JobGetResponse
{
JobDetail = new JobDetailRootJsonObject {
Id = jobId, Status = new Status(), Userargs = new Userargs()
}
};
switch (type)
{
case JobCompletionType.Success:
break;
case JobCompletionType.Fail:
throw new CloudException("Some Failure");
case JobCompletionType.TimeOut:
throw new TimeoutException("Some Failure");
default:
break;
}
return(jobResponse);
}
private void WaitForJobCompletion_Internal(JobCompletionType completionType, string expectedMessageFormat)
{
var cmdlet = GetWaitAzureHDInsightJobCommandDefinition(completionType);
bool success = false;
try
{
cmdlet.ExecuteCmdlet();
}
catch (CloudException ex)
{
var expectedExceptionMessage = string.Format(CultureInfo.InvariantCulture, expectedMessageFormat, cmdlet.JobId, cmdlet.TimeoutInSeconds);
if (ex.Message.Equals(expectedExceptionMessage))
{
success = true;
}
}
if (!success)
{
throw new Exception("Operation didn't fail");
}
}
public WaitAzureHDInsightJobCommand GetWaitAzureHDInsightJobCommandDefinition(JobCompletionType jobCompetionType)
{
// Update HDInsight Management properties for Job.
SetupManagementClientForJobTests();
// Setup Job Management mocks
const string jobId = "jobid_1984120_001";
var cmdlet = new WaitAzureHDInsightJobCommand
{
CommandRuntime = commandRuntimeMock.Object,
HDInsightJobClient = hdinsightJobManagementMock.Object,
HDInsightManagementClient = hdinsightManagementMock.Object,
HttpCredential = new PSCredential("httpuser", string.Format("Password1!").ConvertToSecureString()),
ClusterName = ClusterName,
JobId = jobId,
TimeoutInSeconds = 10
};
JobGetResponse jobDetails = null;
TimeSpan? jobDuration = TimeSpan.FromSeconds(10);
hdinsightJobManagementMock.Setup(c => c.WaitForJobCompletion(It.IsAny <string>(), It.IsAny <TimeSpan?>(), null))
.Callback((string id, TimeSpan? duration, TimeSpan? interval) => { jobDetails = MockJobCompletion(id, jobCompetionType); })
.Returns(() => jobDetails);
return(cmdlet);
}
public JobGetResponse MockJobCompletion(string jobId, JobCompletionType type)
{
var jobResponse = new JobGetResponse
{
JobDetail = new JobDetailRootJsonObject { Id = jobId, Status = new Status(), Userargs = new Userargs() }
};
switch (type)
{
case JobCompletionType.Success:
break;
case JobCompletionType.Fail:
throw new CloudException("Some Failure");
case JobCompletionType.TimeOut:
throw new TimeoutException("Some Failure");
default:
break;
}
return jobResponse;
}
public WaitAzureHDInsightJobCommand GetWaitAzureHDInsightJobCommandDefinition(JobCompletionType jobCompetionType)
{
// Update HDInsight Management properties for Job.
SetupManagementClientForJobTests();
// Setup Job Management mocks
const string jobId = "jobid_1984120_001";
var cmdlet = new WaitAzureHDInsightJobCommand
{
CommandRuntime = commandRuntimeMock.Object,
HDInsightJobClient = hdinsightJobManagementMock.Object,
HDInsightManagementClient = hdinsightManagementMock.Object,
HttpCredential = new PSCredential("httpuser", string.Format("Password1!").ConvertToSecureString()),
ClusterName = ClusterName,
JobId = jobId,
TimeoutInSeconds = 10
};
JobGetResponse jobDetails = null;
TimeSpan? jobDuration = TimeSpan.FromSeconds(10);
hdinsightJobManagementMock.Setup(c => c.WaitForJobCompletion(It.IsAny<string>(), It.IsAny<TimeSpan?>(), null))
.Callback((string id, TimeSpan? duration, TimeSpan? interval) => { jobDetails = MockJobCompletion(id, jobCompetionType); })
.Returns(() => jobDetails);
return cmdlet;
}
public void WaitForJobCompletion_Internal(JobCompletionType completionType, string expectedMessageFormat)
{
var cmdlet = GetWaitAzureHDInsightJobCommandDefinition(completionType);
bool success = false;
try
{
cmdlet.ExecuteCmdlet();
}
catch (CloudException ex)
{
var expectedExceptionMessage = string.Format(CultureInfo.InvariantCulture, expectedMessageFormat, cmdlet.JobId, cmdlet.TimeoutInSeconds);
if (ex.Message.Equals(expectedExceptionMessage))
{
success = true;
}
}
if (!success)
{
throw new Exception("Operation didn't fail");
}
}