Added iaas cmdlets for Azure Batch module

src/ResourceManager/AzureBatch/Commands.Batch.Test/Accounts/GetBatchNodeAgentSkusCommandTests.cs

@@ -0,0 +1,114 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+using Microsoft.Azure.Batch;
+using Microsoft.Azure.Batch.Protocol;
+using Microsoft.Rest.Azure;
+using Microsoft.WindowsAzure.Commands.ScenarioTest;
+using Moq;
+using System.Collections.Generic;
+using System.Linq;
+using System.Management.Automation;
+using System.Threading.Tasks;
+using Microsoft.Azure.Commands.Batch.Models;
+using Xunit;
+using BatchClient = Microsoft.Azure.Commands.Batch.Models.BatchClient;
+using ProxyModels = Microsoft.Azure.Batch.Protocol.Models;
+
+namespace Microsoft.Azure.Commands.Batch.Test.Accounts
+{
+    public class GetBatchNodeAgentSkusCommandTests : WindowsAzure.Commands.Test.Utilities.Common.RMTestBase
+    {
+        private GetBatchAccountNodeAgentSkuCommand cmdlet;
+        private Mock<BatchClient> batchClientMock;
+        private Mock<ICommandRuntime> commandRuntimeMock;
+
+        public GetBatchNodeAgentSkusCommandTests()
+        {
+            batchClientMock = new Mock<BatchClient>();
+            commandRuntimeMock = new Mock<ICommandRuntime>();
+            cmdlet = new GetBatchAccountNodeAgentSkuCommand()
+            {
+                CommandRuntime = commandRuntimeMock.Object,
+                BatchClient = batchClientMock.Object,
+            };
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void ListBatchNodeAgentSkusParametersTest()
+        {
+            BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
+            cmdlet.BatchContext = context;
+
+            string[] idsOfNodeAgentSkus = new[] { "batch.node.centos 7", "batch.node.debian 8", "batch.node.opensuse 13.2" };
+
+            // Don't go to the service on an Get NodeAgentSkus call
+            AzureOperationResponse<IPage<ProxyModels.NodeAgentSku>, ProxyModels.AccountListNodeAgentSkusHeaders> response =
+                BatchTestHelpers.CreateNodeAgentSkuResponse(idsOfNodeAgentSkus);
+
+            RequestInterceptor interceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor<
+                ProxyModels.AccountListNodeAgentSkusOptions,
+                AzureOperationResponse<IPage<ProxyModels.NodeAgentSku>, ProxyModels.AccountListNodeAgentSkusHeaders>>(responseToUse: response);
+
+            cmdlet.AdditionalBehaviors = new List<BatchClientBehavior>() { interceptor };
+
+            // Setup the cmdlet to write pipeline output to a list that can be examined later
+            List<PSNodeAgentSku> pipeline = new List<PSNodeAgentSku>();
+            commandRuntimeMock.Setup(r =>
+                r.WriteObject(It.IsAny<PSNodeAgentSku>()))
+                .Callback<object>(p => pipeline.Add((PSNodeAgentSku)p));
+
+            // Verify no exceptions when required parameter is set
+            cmdlet.ExecuteCmdlet();
+
+            Assert.Equal(3, pipeline.Count);
+            int nodeAgentCount = 0;
+            foreach (PSNodeAgentSku p in pipeline)
+            {
+                Assert.True(idsOfNodeAgentSkus.Contains(p.Id));
+                nodeAgentCount++;
+            }
+            Assert.Equal(idsOfNodeAgentSkus.Length, nodeAgentCount);
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void ListBatchNodeAgentSkusWithFilterTest()
+        {
+            BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
+            cmdlet.BatchContext = context;
+            cmdlet.Filter = "osType eq 'linux'";
+
+            string requestFilter = null;
+
+            // Don't go to the service on an Get NodeAgentSkus call
+            AzureOperationResponse<IPage<ProxyModels.NodeAgentSku>, ProxyModels.AccountListNodeAgentSkusHeaders> getResponse = BatchTestHelpers.CreateGenericAzureOperationListResponse<ProxyModels.NodeAgentSku, ProxyModels.AccountListNodeAgentSkusHeaders>();
+            RequestInterceptor requestInterceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor<ProxyModels.AccountListNodeAgentSkusOptions, AzureOperationResponse<IPage<ProxyModels.NodeAgentSku>, ProxyModels.AccountListNodeAgentSkusHeaders>>(responseToUse: getResponse);
+            ResponseInterceptor responseInterceptor = new ResponseInterceptor((response, request) =>
+            {
+                ProxyModels.AccountListNodeAgentSkusOptions listNodeAgentSkusOptions = (ProxyModels.AccountListNodeAgentSkusOptions) request.Options;
+                requestFilter = listNodeAgentSkusOptions.Filter;
+
+                return Task.FromResult(response);
+            });
+            cmdlet.AdditionalBehaviors = new List<BatchClientBehavior>() { requestInterceptor, responseInterceptor };
+
+            // Verify no exceptions when required parameter is set
+            cmdlet.ExecuteCmdlet();
+
+            Assert.Equal(cmdlet.Filter, requestFilter);
+        }
+    }
+}

src/ResourceManager/AzureBatch/Commands.Batch.Test/BatchTestHelpers.cs

@@ -21,6 +21,7 @@
 using System.Collections;
 using System.Collections.Generic;
 using System.IO;
+using System.Linq;
 using System.Net;
 using System.Net.Http;
 using System.Reflection;
@@ -157,7 +158,7 @@ public static void AssertBatchAccountContextsAreEqual(BatchAccountContext contex
         public static RequestInterceptor CreateFakeServiceResponseInterceptor<TBody, TOptions, TResponse>(TResponse responseToUse = default(TResponse),
             Action<BatchRequest<TBody, TOptions, TResponse>> requestAction = null)
             where TOptions : ProxyModels.IOptions, new()
-            where TResponse : IAzureOperationResponse, new()
+            where TResponse : class, IAzureOperationResponse, new()
         {
             RequestInterceptor interceptor = new RequestInterceptor((baseRequest) =>
             {
@@ -170,15 +171,12 @@ public static void AssertBatchAccountContextsAreEqual(BatchAccountContext contex
 
                 request.ServiceRequestFunc = (cancellationToken) =>
                 {
-                    if (responseToUse == null)
+                    TResponse response = responseToUse ?? new TResponse()
                     {
-                        responseToUse = new TResponse()
-                        {
-                            Response = new HttpResponseMessage()
-                        };
-                    }
+                        Response = new HttpResponseMessage()
+                    };
 
-                    Task<TResponse> task = Task.FromResult(responseToUse);
+                    Task<TResponse> task = Task.FromResult(response);
                     return task;
                 };
             });
@@ -192,7 +190,7 @@ public static void AssertBatchAccountContextsAreEqual(BatchAccountContext contex
         /// <typeparam name="THeader">The type of header for the response</typeparam>
         /// <returns></returns>
         public static AzureOperationResponse<TBody, THeader> CreateGenericAzureOperationResponse<TBody, THeader>()
-            where TBody : class, new () 
+            where TBody : class, new ()
             where THeader : class, new ()
         {
             var response = new AzureOperationResponse<TBody, THeader>()
@@ -221,7 +219,7 @@ public static AzureOperationResponse<IPage<TBody>, THeader> CreateGenericAzureOp
             };
 
             return response;
-        } 
+        }
 
         /// <summary>
         /// Creates a RequestInterceptor that does not contact the Batch Service on a Get NodeFile or a Get NodeFile Properties call.
@@ -319,7 +317,7 @@ public static RequestInterceptor CreateFakeGetFileAndPropertiesFromComputeNodeRe
             cert.Thumbprint = thumbprint;
 
             response.Body = cert;
-            
+
             return response;
         }
 
@@ -330,7 +328,7 @@ public static RequestInterceptor CreateFakeGetFileAndPropertiesFromComputeNodeRe
         {
             var response = new AzureOperationResponse<IPage<ProxyModels.Certificate>, ProxyModels.CertificateListHeaders>();
             response.Response = new HttpResponseMessage(HttpStatusCode.OK);
-            
+
             List<ProxyModels.Certificate> certs = new List<ProxyModels.Certificate>();
 
             foreach (string t in certThumbprints)
@@ -383,6 +381,23 @@ public static RequestInterceptor CreateFakeGetFileAndPropertiesFromComputeNodeRe
             return response;
         }
 
+        /// <summary>
+        /// Builds a GetRemoteLoginSettingsResponse object
+        /// </summary>
+        public static AzureOperationResponse<ProxyModels.ComputeNodeGetRemoteLoginSettingsResult, ProxyModels.ComputeNodeGetRemoteLoginSettingsHeaders> CreateRemoteLoginSettingsGetResponse(string ipAddress)
+        {
+            var settings = new ProxyModels.ComputeNodeGetRemoteLoginSettingsResult();
+            settings.RemoteLoginIPAddress = ipAddress;
+
+            var response = new AzureOperationResponse<ProxyModels.ComputeNodeGetRemoteLoginSettingsResult, ProxyModels.ComputeNodeGetRemoteLoginSettingsHeaders>()
+            {
+                Response = new HttpResponseMessage(HttpStatusCode.OK),
+                Body = settings
+            };
+
+            return response;
+        }
+
         /// <summary>
         /// Builds a ComputeNodeGetResponse object
         /// </summary>
@@ -492,7 +507,7 @@ public static RequestInterceptor CreateFakeGetFileAndPropertiesFromComputeNodeRe
                 job.Id = id;
                 jobs.Add(job);
             }
-            
+
             response.Body = new MockPagedEnumerable<ProxyModels.CloudJob>(jobs);
 
             return response;
@@ -519,7 +534,7 @@ public static RequestInterceptor CreateFakeGetFileAndPropertiesFromComputeNodeRe
         /// </summary>
         public static AzureOperationResponse<IPage<ProxyModels.CloudTask>, ProxyModels.TaskListHeaders> CreateCloudTaskListResponse(IEnumerable<string> taskIds)
         {
-            var response = new AzureOperationResponse<IPage<ProxyModels.CloudTask>, ProxyModels.TaskListHeaders>(); 
+            var response = new AzureOperationResponse<IPage<ProxyModels.CloudTask>, ProxyModels.TaskListHeaders>();
             response.Response = new HttpResponseMessage(HttpStatusCode.OK);
 
             List<ProxyModels.CloudTask> tasks = new List<ProxyModels.CloudTask>();
@@ -563,6 +578,23 @@ public static RequestInterceptor CreateFakeGetFileAndPropertiesFromComputeNodeRe
             return response;
         }
 
+        /// <summary>
+        /// Builds a NodeAgentSKUResponse object
+        /// </summary>
+        public static AzureOperationResponse<IPage<ProxyModels.NodeAgentSku>, ProxyModels.AccountListNodeAgentSkusHeaders> CreateNodeAgentSkuResponse(IEnumerable<string> skuIds)
+        {
+            IEnumerable<ProxyModels.NodeAgentSku> nodeAgents =
+                skuIds.Select(id => new ProxyModels.NodeAgentSku() { Id = id });
+
+            var response = new AzureOperationResponse<IPage<ProxyModels.NodeAgentSku>, ProxyModels.AccountListNodeAgentSkusHeaders>()
+            {
+                Response = new HttpResponseMessage(HttpStatusCode.OK),
+                Body = new MockPagedEnumerable<ProxyModels.NodeAgentSku>(nodeAgents)
+            };
+
+            return response;
+        }
+
         /// <summary>
         /// Builds a NodeFileGetPropertiesResponse object
         /// </summary>
@@ -638,7 +670,7 @@ public static RequestInterceptor CreateFakeGetFileAndPropertiesFromComputeNodeRe
             response.Response = new HttpResponseMessage(HttpStatusCode.OK);
 
             List<ProxyModels.NodeFile> files = new List<ProxyModels.NodeFile>();
-            
+
             foreach (string name in fileNames)
             {
                 ProxyModels.NodeFile file = new ProxyModels.NodeFile();
@@ -689,7 +721,7 @@ public static CloudJobSchedule CreateFakeBoundJobSchedule(BatchAccountContext co
                 request.ServiceRequestFunc = (cancellationToken) =>
                 {
                     var response = new AzureOperationResponse<ProxyModels.CloudJobSchedule, ProxyModels.JobScheduleGetHeaders>();
-                    
+
                     response.Body = new ProxyModels.CloudJobSchedule(id: jobScheduleId, schedule: new ProxyModels.Schedule(), jobSpecification: new ProxyModels.JobSpecification());
 
                     Task<AzureOperationResponse<ProxyModels.CloudJobSchedule, ProxyModels.JobScheduleGetHeaders>> task = Task.FromResult(response);
@@ -759,7 +791,7 @@ internal static void SetProperty(object obj, string propertyName, object propert
         }
 
         /// <summary>
-        /// Uses Reflection to set a property value on an object. 
+        /// Uses Reflection to set a property value on an object.
         /// </summary>
         internal static void SetField(object obj, string fieldName, object fieldValue)
         {

src/ResourceManager/AzureBatch/Commands.Batch.Test/Commands.Batch.Test.csproj

@@ -44,8 +44,8 @@
     <Reference Include="Hyak.Common">
       <HintPath>..\..\..\packages\Hyak.Common.1.0.3\lib\portable-net403+win+wpa81\Hyak.Common.dll</HintPath>
     </Reference>
-    <Reference Include="Microsoft.Azure.Batch, Version=4.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
-      <HintPath>..\..\..\packages\Azure.Batch.4.0.0\lib\net45\Microsoft.Azure.Batch.dll</HintPath>
+    <Reference Include="Microsoft.Azure.Batch, Version=4.0.1.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
+      <HintPath>..\..\..\packages\Azure.Batch.4.0.1\lib\net45\Microsoft.Azure.Batch.dll</HintPath>
       <Private>True</Private>
     </Reference>
     <Reference Include="Microsoft.Azure.Common">
@@ -181,6 +181,7 @@
   <ItemGroup>
     <Compile Include="Accounts\GetBatchAccountCommandTests.cs" />
     <Compile Include="Accounts\GetBatchAccountKeysCommandTests.cs" />
+    <Compile Include="Accounts\GetBatchNodeAgentSkusCommandTests.cs" />
     <Compile Include="Accounts\NewBatchAccountCommandTests.cs" />
     <Compile Include="Accounts\RegenBatchAccountKeyCommandTests.cs" />
     <Compile Include="Accounts\RemoveBatchAccountCommandTests.cs" />
@@ -192,6 +193,7 @@
     <Compile Include="Certificates\StopBatchCertificateDeletionCommandTests.cs" />
     <Compile Include="ComputeNodes\DisableBatchComputeNodeSchedulingCommandTests.cs" />
     <Compile Include="ComputeNodes\EnableBatchComputeNodeSchedulingCommandTests.cs" />
+    <Compile Include="ComputeNodes\GetBatchComputeNodeLoginSettingsCommandTests.cs" />
     <Compile Include="ComputeNodes\RemoveBatchComputeNodeCommandTests.cs" />
     <Compile Include="ComputeNodes\ResetBatchComputeNodeCommandTests.cs" />
     <Compile Include="ComputeNodes\RestartBatchComputeNodeCommandTests.cs" />
@@ -361,6 +363,12 @@
     <None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.ComputeNodeTests\TestGetComputeNodeById.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.ComputeNodeTests\TestGetComputeNodeRemoteLoginSettingsById.json">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
+    <None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.ComputeNodeTests\TestGetComputeNodeRemoteLoginSettingsPipeline.json">
+    <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.ComputeNodeTests\TestListAllComputeNodes.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
@@ -586,6 +594,9 @@
     <None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.PoolTests\TestListAllPools.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.BatchAccountTests\TestListNodeAgentSkus.json">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.PoolTests\TestListPoolsByFilter.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>