protected override void ExecuteCmdletImpl()
{
GetTaskCountsOptions options = new GetTaskCountsOptions(this.BatchContext, this.JobId, this.Job, this.AdditionalBehaviors);
PSTaskSlotCounts taskSlotCount = BatchClient.GetTaskSlotCounts(options);
WriteObject(taskSlotCount);
}
public void GetBatchTaskSlotCountTest()
{
// Setup cmdlet to get task slot count by job id
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.JobId = "job-1";
const int requiredSlots = 2;
const int active = 3;
const int running = 5;
const int succeeded = 2;
const int failed = 1;
// Build a TaskCountsResult instead of querying the service
AzureOperationResponse <ProxyModels.TaskCountsResult, ProxyModels.JobGetTaskCountsHeaders> response =
BatchTestHelpers.CreateTaskCountsGetResponse(requiredSlots, active, running, succeeded, failed);
RequestInterceptor interceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor <
ProxyModels.JobGetTaskCountsOptions,
AzureOperationResponse <ProxyModels.TaskCountsResult, ProxyModels.JobGetTaskCountsHeaders> >(response);
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
PSTaskSlotCounts slotCount = null;
commandRuntimeMock
.Setup(r => r.WriteObject(It.IsAny <PSTaskSlotCounts>()))
.Callback <object>(p => {
slotCount = (PSTaskSlotCounts)p;
});
cmdlet.ExecuteCmdlet();
Assert.Equal(6, slotCount.Active);
Assert.Equal(10, slotCount.Running);
Assert.Equal(6, slotCount.Completed);
Assert.Equal(4, slotCount.Succeeded);
Assert.Equal(2, slotCount.Failed);
}
