public void ListTasksMaxCountTest()
{
// Verify default max count
Assert.Equal(Microsoft.Azure.Commands.Batch.Utils.Constants.DefaultMaxCount, cmdlet.MaxCount);
// Setup cmdlet to list tasks without filters and a max count
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.JobId = "job-1";
cmdlet.Id = null;
cmdlet.Filter = null;
int maxCount = 2;
cmdlet.MaxCount = maxCount;
string[] idsOfConstructedTasks = new[] { "testTask1", "testTask2", "testTask3" };
// Build some CloudTasks instead of querying the service on a List CloudTasks call
RequestInterceptor interceptor = new RequestInterceptor((baseRequest) =>
{
BatchRequest <CloudTaskListParameters, CloudTaskListResponse> request =
(BatchRequest <CloudTaskListParameters, CloudTaskListResponse>)baseRequest;
request.ServiceRequestFunc = (cancellationToken) =>
{
CloudTaskListResponse response = BatchTestHelpers.CreateCloudTaskListResponse(idsOfConstructedTasks);
Task <CloudTaskListResponse> task = Task.FromResult(response);
return(task);
};
});
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
List <PSCloudTask> pipeline = new List <PSCloudTask>();
commandRuntimeMock.Setup(r =>
r.WriteObject(It.IsAny <PSCloudTask>()))
.Callback <object>(t => pipeline.Add((PSCloudTask)t));
cmdlet.ExecuteCmdlet();
// Verify that the max count was respected
Assert.Equal(maxCount, pipeline.Count);
// Verify setting max count <= 0 doesn't return nothing
cmdlet.MaxCount = -5;
pipeline.Clear();
cmdlet.ExecuteCmdlet();
Assert.Equal(idsOfConstructedTasks.Length, pipeline.Count);
}
public void ListBatchTasksWithoutFiltersTest()
{
// Setup cmdlet to list tasks without filters. Use WorkItemName and JobName.
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.JobId = "job-1";
cmdlet.Id = null;
cmdlet.Filter = null;
string[] idsOfConstructedTasks = new[] { "testTask1", "testTask2", "testTask3" };
// Build some CloudTasks instead of querying the service on a List CloudTasks call
RequestInterceptor interceptor = new RequestInterceptor((baseRequest) =>
{
BatchRequest <CloudTaskListParameters, CloudTaskListResponse> request =
(BatchRequest <CloudTaskListParameters, CloudTaskListResponse>)baseRequest;
request.ServiceRequestFunc = (cancellationToken) =>
{
CloudTaskListResponse response = BatchTestHelpers.CreateCloudTaskListResponse(idsOfConstructedTasks);
Task <CloudTaskListResponse> task = Task.FromResult(response);
return(task);
};
});
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
List <PSCloudTask> pipeline = new List <PSCloudTask>();
commandRuntimeMock.Setup(r =>
r.WriteObject(It.IsAny <PSCloudTask>()))
.Callback <object>(t => pipeline.Add((PSCloudTask)t));
cmdlet.ExecuteCmdlet();
// Verify that the cmdlet wrote the constructed tasks to the pipeline
Assert.Equal(3, pipeline.Count);
int taskCount = 0;
foreach (PSCloudTask t in pipeline)
{
Assert.True(idsOfConstructedTasks.Contains(t.Id));
taskCount++;
}
Assert.Equal(idsOfConstructedTasks.Length, taskCount);
}
public void ListBatchTasksByODataFilterTest()
{
// Setup cmdlet to list tasks using an OData filter.
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.JobId = "job-1";
cmdlet.Id = null;
cmdlet.Filter = "startswith(id,'test')";
string[] idsOfConstructedTasks = new[] { "testTask1", "testTask2" };
// Build some CloudTasks instead of querying the service on a List CloudTasks call
CloudTaskListResponse response = BatchTestHelpers.CreateCloudTaskListResponse(idsOfConstructedTasks);
RequestInterceptor interceptor = BatchTestHelpers.CreateNoOpInterceptor <CloudTaskListParameters, CloudTaskListResponse>(response);
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
List <PSCloudTask> pipeline = new List <PSCloudTask>();
commandRuntimeMock.Setup(r =>
r.WriteObject(It.IsAny <PSCloudTask>()))
.Callback <object>(t => pipeline.Add((PSCloudTask)t));
cmdlet.ExecuteCmdlet();
// Verify that the cmdlet wrote the constructed tasks to the pipeline
Assert.Equal(2, pipeline.Count);
int taskCount = 0;
foreach (PSCloudTask t in pipeline)
{
Assert.True(idsOfConstructedTasks.Contains(t.Id));
taskCount++;
}
Assert.Equal(idsOfConstructedTasks.Length, taskCount);
}
public void GetBatchTaskParametersTest()
{
// Setup cmdlet without required parameters
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.JobId = null;
cmdlet.Job = null;
cmdlet.Filter = null;
// Build a CloudTask instead of querying the service on a List CloudTask call
RequestInterceptor interceptor = new RequestInterceptor((baseRequest) =>
{
BatchRequest <CloudTaskListParameters, CloudTaskListResponse> request =
(BatchRequest <CloudTaskListParameters, CloudTaskListResponse>)baseRequest;
request.ServiceRequestFunc = (cancellationToken) =>
{
CloudTaskListResponse response = BatchTestHelpers.CreateCloudTaskListResponse(new string[] { cmdlet.Id });
Task <CloudTaskListResponse> task = Task.FromResult(response);
return(task);
};
});
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
Assert.Throws <ArgumentNullException>(() => cmdlet.ExecuteCmdlet());
cmdlet.JobId = "job-1";
// Verify no exceptions occur
cmdlet.ExecuteCmdlet();
}
