public void ListPoolsMaxCountTest() { // Verify default max count Assert.Equal(Microsoft.Azure.Commands.Batch.Utils.Constants.DefaultMaxCount, cmdlet.MaxCount); // Setup cmdlet to list pools without filters and a max count BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys(); cmdlet.BatchContext = context; cmdlet.Id = null; cmdlet.Filter = null; int maxCount = 2; cmdlet.MaxCount = maxCount; string[] idsOfConstructedPools = new[] { "pool1", "pool2", "pool3" }; // Build some CloudPools instead of querying the service on a List CloudPools call RequestInterceptor interceptor = new RequestInterceptor((baseRequest) => { BatchRequest <CloudPoolListParameters, CloudPoolListResponse> request = (BatchRequest <CloudPoolListParameters, CloudPoolListResponse>)baseRequest; request.ServiceRequestFunc = (cancellationToken) => { CloudPoolListResponse response = BatchTestHelpers.CreateCloudPoolListResponse(idsOfConstructedPools); Task <CloudPoolListResponse> 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 <PSCloudPool> pipeline = new List <PSCloudPool>(); commandRuntimeMock.Setup(r => r.WriteObject(It.IsAny <PSCloudPool>())) .Callback <object>(p => pipeline.Add((PSCloudPool)p)); 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(idsOfConstructedPools.Length, pipeline.Count); }
public void ListBatchPoolByODataFilterTest() { // Setup cmdlet to list pools using an OData filter BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys(); cmdlet.BatchContext = context; cmdlet.Id = null; cmdlet.Filter = "startswith(id,'test')"; string[] idsOfConstructedPools = new[] { "test1", "test2" }; // Build some CloudPools instead of querying the service on a List CloudPools call RequestInterceptor interceptor = new RequestInterceptor((baseRequest) => { BatchRequest <CloudPoolListParameters, CloudPoolListResponse> request = (BatchRequest <CloudPoolListParameters, CloudPoolListResponse>)baseRequest; request.ServiceRequestFunc = (cancellationToken) => { CloudPoolListResponse response = BatchTestHelpers.CreateCloudPoolListResponse(idsOfConstructedPools); Task <CloudPoolListResponse> 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 <PSCloudPool> pipeline = new List <PSCloudPool>(); commandRuntimeMock.Setup(r => r.WriteObject(It.IsAny <PSCloudPool>())) .Callback <object>(p => pipeline.Add((PSCloudPool)p)); cmdlet.ExecuteCmdlet(); // Verify that the cmdlet wrote the constructed pools to the pipeline Assert.Equal(2, pipeline.Count); int poolCount = 0; foreach (PSCloudPool p in pipeline) { Assert.True(idsOfConstructedPools.Contains(p.Id)); poolCount++; } Assert.Equal(idsOfConstructedPools.Length, poolCount); }
public void ListBatchPoolWithoutFiltersTest() { // Setup cmdlet to list pools without filters BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys(); cmdlet.BatchContext = context; cmdlet.Id = null; cmdlet.Filter = null; string[] idsOfConstructedPools = new[] { "pool1", "pool2", "pool3" }; // Build some CloudPools instead of querying the service on a List CloudPools call CloudPoolListResponse response = BatchTestHelpers.CreateCloudPoolListResponse(idsOfConstructedPools); RequestInterceptor interceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor <CloudPoolListParameters, CloudPoolListResponse>(response); cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>() { interceptor }; // Setup the cmdlet to write pipeline output to a list that can be examined later List <PSCloudPool> pipeline = new List <PSCloudPool>(); commandRuntimeMock.Setup(r => r.WriteObject(It.IsAny <PSCloudPool>())) .Callback <object>(p => pipeline.Add((PSCloudPool)p)); cmdlet.ExecuteCmdlet(); // Verify that the cmdlet wrote the constructed pools to the pipeline Assert.Equal(3, pipeline.Count); int poolCount = 0; foreach (PSCloudPool p in pipeline) { Assert.True(idsOfConstructedPools.Contains(p.Id)); poolCount++; } Assert.Equal(idsOfConstructedPools.Length, poolCount); }
public void ListPoolsMaxCountTest() { // Verify default max count Assert.Equal(Microsoft.Azure.Commands.Batch.Utils.Constants.DefaultMaxCount, cmdlet.MaxCount); // Setup cmdlet to list pools without filters and a max count BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys(); cmdlet.BatchContext = context; cmdlet.Id = null; cmdlet.Filter = null; int maxCount = 2; cmdlet.MaxCount = maxCount; string[] idsOfConstructedPools = new[] { "pool1", "pool2", "pool3" }; // Build some CloudPools instead of querying the service on a List CloudPools call AzureOperationResponse <IPage <ProxyModels.CloudPool>, ProxyModels.PoolListHeaders> response = BatchTestHelpers.CreateCloudPoolListResponse(idsOfConstructedPools); RequestInterceptor interceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor < ProxyModels.PoolListOptions, AzureOperationResponse <IPage <ProxyModels.CloudPool>, ProxyModels.PoolListHeaders> >(response); cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>() { interceptor }; // Setup the cmdlet to write pipeline output to a list that can be examined later List <PSCloudPool> pipeline = new List <PSCloudPool>(); commandRuntimeMock.Setup(r => r.WriteObject(It.IsAny <PSCloudPool>())) .Callback <object>(p => pipeline.Add((PSCloudPool)p)); 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(idsOfConstructedPools.Length, pipeline.Count); }