public void ListTaskFilesMaxCountTest()
{
// Verify default max count
Assert.Equal(Microsoft.Azure.Commands.Batch.Utils.Constants.DefaultMaxCount, cmdlet.MaxCount);
// Setup cmdlet to list Task files and a max count.
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.WorkItemName = "workItem";
cmdlet.JobName = "job-0000000001";
cmdlet.TaskName = "task";
cmdlet.Name = null;
cmdlet.Filter = null;
int maxCount = 2;
cmdlet.MaxCount = maxCount;
string[] namesOfConstructedTaskFiles = new[] { "stdout.txt", "stderr.txt", "wd" };
// Build some Task files instead of querying the service on a ListTaskFiles call
YieldInjectionInterceptor interceptor = new YieldInjectionInterceptor((opContext, request) =>
{
if (request is ListTaskFilesRequest)
{
ListTaskFilesResponse response = BatchTestHelpers.CreateListTaskFilesResponse(namesOfConstructedTaskFiles);
Task <object> task = Task <object> .Factory.StartNew(() => { return(response); });
return(task);
}
return(null);
});
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
List <PSTaskFile> pipeline = new List <PSTaskFile>();
commandRuntimeMock.Setup(r =>
r.WriteObject(It.IsAny <PSTaskFile>()))
.Callback <object>(t => pipeline.Add((PSTaskFile)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(namesOfConstructedTaskFiles.Length, pipeline.Count);
}
public void ListBatchTaskFilesWithoutFiltersTest()
{
// Setup cmdlet to list Task files without filters.
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.WorkItemName = "workItem";
cmdlet.JobName = "job-0000000001";
cmdlet.TaskName = "task";
cmdlet.Name = null;
cmdlet.Filter = null;
string[] namesOfConstructedTaskFiles = new[] { "stdout.txt", "stderr.txt", "wd" };
// Build some Task files instead of querying the service on a ListTaskFiles call
YieldInjectionInterceptor interceptor = new YieldInjectionInterceptor((opContext, request) =>
{
if (request is ListTaskFilesRequest)
{
ListTaskFilesResponse response = BatchTestHelpers.CreateListTaskFilesResponse(namesOfConstructedTaskFiles);
Task <object> task = Task <object> .Factory.StartNew(() => { return(response); });
return(task);
}
return(null);
});
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
List <PSTaskFile> pipeline = new List <PSTaskFile>();
commandRuntimeMock.Setup(r =>
r.WriteObject(It.IsAny <PSTaskFile>()))
.Callback <object>(t => pipeline.Add((PSTaskFile)t));
cmdlet.ExecuteCmdlet();
// Verify that the cmdlet wrote the constructed Task files to the pipeline
Assert.Equal(3, pipeline.Count);
int taskCount = 0;
foreach (PSTaskFile f in pipeline)
{
Assert.True(namesOfConstructedTaskFiles.Contains(f.Name));
taskCount++;
}
Assert.Equal(namesOfConstructedTaskFiles.Length, taskCount);
}
public void GetBatchTaskFileParametersTest()
{
// Setup cmdlet without required parameters
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.WorkItemName = null;
cmdlet.JobName = null;
cmdlet.TaskName = null;
cmdlet.Name = null;
cmdlet.Task = null;
cmdlet.Filter = null;
// Build some Task files instead of querying the service on a ListTaskFiles call
YieldInjectionInterceptor interceptor = new YieldInjectionInterceptor((opContext, request) =>
{
if (request is ListTaskFilesRequest)
{
ListTaskFilesResponse response = BatchTestHelpers.CreateListTaskFilesResponse(new string[] { "stdout.txt" });
Task <object> task = Task <object> .Factory.StartNew(() => { return(response); });
return(task);
}
return(null);
});
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
Assert.Throws <ArgumentNullException>(() => cmdlet.ExecuteCmdlet());
cmdlet.WorkItemName = "workItem";
cmdlet.JobName = "job-0000000001";
cmdlet.TaskName = "task";
// Verify no exceptions occur
cmdlet.ExecuteCmdlet();
cmdlet.WorkItemName = null;
cmdlet.JobName = null;
cmdlet.TaskName = null;
cmdlet.Task = new PSCloudTask("task", "cmd /c dir /s");
// Verify that we don't get an argument exception. We should get an InvalidOperationException though since the task is unbound
Assert.Throws <InvalidOperationException>(() => cmdlet.ExecuteCmdlet());
}
/// <summary>
/// Builds a ListTaskFilesResponse object
/// </summary>
public static ListTaskFilesResponse CreateListTaskFilesResponse(IEnumerable <string> fileNames)
{
ListTaskFilesResponse response = new ListTaskFilesResponse();
SetProperty(response, "StatusCode", HttpStatusCode.OK);
List <Azure.Batch.Protocol.Entities.File> files = new List <Azure.Batch.Protocol.Entities.File>();
foreach (string name in fileNames)
{
Azure.Batch.Protocol.Entities.File file = new Azure.Batch.Protocol.Entities.File();
SetProperty(file, "Name", name);
files.Add(file);
}
SetProperty(response, "Files", files);
return(response);
}
/// <summary>
/// Builds a ListTaskFilesResponse object
/// </summary>
public static ListTaskFilesResponse CreateListTaskFilesResponse(IEnumerable<string> fileNames)
{
ListTaskFilesResponse response = new ListTaskFilesResponse();
SetProperty(response, "StatusCode", HttpStatusCode.OK);
List<Azure.Batch.Protocol.Entities.File> files = new List<Azure.Batch.Protocol.Entities.File>();
foreach (string name in fileNames)
{
Azure.Batch.Protocol.Entities.File file = new Azure.Batch.Protocol.Entities.File();
SetProperty(file, "Name", name);
files.Add(file);
}
SetProperty(response, "Files", files);
return response;
}
