public void DisableJobParametersTest()
{
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.Id = null;
Assert.Throws<ArgumentNullException>(() => cmdlet.ExecuteCmdlet());
cmdlet.Id = "testJob";
cmdlet.DisableJobOption = DisableJobOption.Terminate;
// Don't go to the service on a Disable CloudJob call
RequestInterceptor interceptor = new RequestInterceptor((baseRequest) =>
{
BatchRequest<CloudJobDisableParameters, CloudJobDisableResponse> request =
(BatchRequest<CloudJobDisableParameters, CloudJobDisableResponse>)baseRequest;
request.ServiceRequestFunc = (cancellationToken) =>
{
CloudJobDisableResponse response = new CloudJobDisableResponse();
Task<CloudJobDisableResponse> task = Task.FromResult(response);
return task;
};
});
cmdlet.AdditionalBehaviors = new List<BatchClientBehavior>() { interceptor };
// Verify no exceptions when required parameter is set
cmdlet.ExecuteCmdlet();
}
public void DisableJobRequestTest()
{
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
DisableJobOption disableOption = DisableJobOption.Terminate;
DisableJobOption requestDisableOption = DisableJobOption.Requeue;
cmdlet.Id = "testJob";
cmdlet.DisableJobOption = disableOption;
// Don't go to the service on an Enable AutoScale call
RequestInterceptor interceptor = new RequestInterceptor((baseRequest) =>
{
BatchRequest <CloudJobDisableParameters, CloudJobDisableResponse> request =
(BatchRequest <CloudJobDisableParameters, CloudJobDisableResponse>)baseRequest;
// Grab the disable option off the outgoing request.
requestDisableOption = request.TypedParameters.DisableJobOption;
request.ServiceRequestFunc = (cancellationToken) =>
{
CloudJobDisableResponse response = new CloudJobDisableResponse();
Task <CloudJobDisableResponse> task = Task.FromResult(response);
return(task);
};
});
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
cmdlet.ExecuteCmdlet();
// Verify that the job disable option was properly set on the outgoing request
Assert.Equal(disableOption, requestDisableOption);
}
public void DisableJobRequestTest()
{
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
DisableJobOption disableOption = DisableJobOption.Terminate;
DisableJobOption requestDisableOption = DisableJobOption.Requeue;
cmdlet.Id = "testJob";
cmdlet.DisableJobOption = disableOption;
// Don't go to the service on an Enable AutoScale call
RequestInterceptor interceptor = new RequestInterceptor((baseRequest) =>
{
BatchRequest<CloudJobDisableParameters, CloudJobDisableResponse> request =
(BatchRequest<CloudJobDisableParameters, CloudJobDisableResponse>)baseRequest;
requestDisableOption = request.TypedParameters.DisableJobOption;
request.ServiceRequestFunc = (cancellationToken) =>
{
CloudJobDisableResponse response = new CloudJobDisableResponse();
Task<CloudJobDisableResponse> task = Task.FromResult(response);
return task;
};
});
cmdlet.AdditionalBehaviors = new List<BatchClientBehavior>() { interceptor };
cmdlet.ExecuteCmdlet();
// Verify that the job disable option was properly set on the outgoing request
Assert.Equal(disableOption, requestDisableOption);
}