Batch cmdlets

src/ResourceManager/Batch/Commands.Batch.Test/Accounts/GetBatchAccountCommandTests.cs

@@ -18,6 +18,7 @@
 using System.Collections.Generic;
 using System.Management.Automation;
 using Xunit;
+using BatchClient = Microsoft.Azure.Commands.Batch.Models.BatchClient;
 
 namespace Microsoft.Azure.Commands.Batch.Test.Accounts
 {

src/ResourceManager/Batch/Commands.Batch.Test/Accounts/GetBatchAccountKeysCommandTests.cs

@@ -18,6 +18,7 @@
 using System.Collections.Generic;
 using System.Management.Automation;
 using Xunit;
+using BatchClient = Microsoft.Azure.Commands.Batch.Models.BatchClient;
 
 namespace Microsoft.Azure.Commands.Batch.Test.Accounts
 {

src/ResourceManager/Batch/Commands.Batch.Test/Accounts/NewBatchAccountCommandTests.cs

@@ -19,6 +19,7 @@
 using System.Collections.Generic;
 using System.Management.Automation;
 using Xunit;
+using BatchClient = Microsoft.Azure.Commands.Batch.Models.BatchClient;
 
 namespace Microsoft.Azure.Commands.Batch.Test.Accounts
 {

src/ResourceManager/Batch/Commands.Batch.Test/Accounts/RegenBatchAccountKeyCommandTests.cs

@@ -18,6 +18,7 @@
 using System.Collections.Generic;
 using System.Management.Automation;
 using Xunit;
+using BatchClient = Microsoft.Azure.Commands.Batch.Models.BatchClient;
 
 namespace Microsoft.Azure.Commands.Batch.Test.Accounts
 {

src/ResourceManager/Batch/Commands.Batch.Test/Accounts/RemoveBatchAccountCommandTests.cs

@@ -17,6 +17,7 @@
 using Moq;
 using System.Management.Automation;
 using Xunit;
+using BatchClient = Microsoft.Azure.Commands.Batch.Models.BatchClient;
 
 namespace Microsoft.Azure.Commands.Batch.Test.Accounts
 {

src/ResourceManager/Batch/Commands.Batch.Test/Accounts/SetBatchAccountCommandTests.cs

@@ -19,6 +19,7 @@
 using System.Collections.Generic;
 using System.Management.Automation;
 using Xunit;
+using BatchClient = Microsoft.Azure.Commands.Batch.Models.BatchClient;
 
 namespace Microsoft.Azure.Commands.Batch.Test.Accounts
 {

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

@@ -12,20 +12,46 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using System.IO;
+using System.Linq;
+using System.Net;
+using System.Net.Http;
+using System.Threading.Tasks;
+using Microsoft.Azure.Batch;
+using Microsoft.Azure.Batch.Common;
+using Microsoft.Azure.Batch.Protocol;
+using Microsoft.Azure.Batch.Protocol.Entities;
+using Microsoft.Azure.Commands.Batch.Models;
+using Microsoft.Azure.Commands.Batch.Test.ScenarioTests;
+using Microsoft.Azure.Management.Batch;
 using Microsoft.Azure.Management.Batch.Models;
+using System;
 using System.Collections;
 using System.Collections.Generic;
+using System.Reflection;
+using Microsoft.Azure.Management.Resources;
+using Microsoft.Azure.Management.Resources.Models;
+using Microsoft.Azure.Test.HttpRecorder;
+using Microsoft.WindowsAzure.Commands.ServiceManagement.Model;
+using Microsoft.WindowsAzure.Storage.Shared.Protocol;
 using Xunit;
+using JobExecutionEnvironment = Microsoft.Azure.Batch.Protocol.Entities.JobExecutionEnvironment;
 
 namespace Microsoft.Azure.Commands.Batch.Test
 {
+    /// <summary>
+    /// Helper methods for the Batch cmdlet tests
+    /// </summary>
     public static class BatchTestHelpers
     {
+        /// <summary>
+        /// Builds an AccountResource object using the specified parameters
+        /// </summary>
         public static AccountResource CreateAccountResource(string accountName, string resourceGroupName, Hashtable[] tags = null)
         {
             string tenantUrlEnding = "batch-test.windows-int.net";
             string endpoint = string.Format("{0}.{1}", accountName, tenantUrlEnding);
-            string subscription = "00000000-0000-0000-0000-000000000000";
+            string subscription = Guid.Empty.ToString();
             string resourceGroup = resourceGroupName;
 
             AccountResource resource = new AccountResource()
@@ -39,9 +65,27 @@ public static AccountResource CreateAccountResource(string accountName, string r
             {
                 resource.Tags = Microsoft.Azure.Commands.Batch.Helpers.CreateTagDictionary(tags, true);
             }
+
             return resource;
         }
 
+        /// <summary>
+        /// Builds a BatchAccountContext object with the keys set for testing
+        /// </summary>
+        public static BatchAccountContext CreateBatchContextWithKeys()
+        {
+            AccountResource resource = CreateAccountResource("account", "resourceGroup");
+            BatchAccountContext context = BatchAccountContext.ConvertAccountResourceToNewAccountContext(resource);
+            string dummyKey = "0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000";
+            SetProperty(context, "PrimaryAccountKey", dummyKey);
+            SetProperty(context, "SecondaryAccountKey", dummyKey);
+
+            return context;
+        }
+
+        /// <summary>
+        /// Verifies that two BatchAccountContext objects are equal
+        /// </summary>
         public static void AssertBatchAccountContextsAreEqual(BatchAccountContext context1, BatchAccountContext context2)
         {
             if (context1 == null)
@@ -67,5 +111,190 @@ public static void AssertBatchAccountContextsAreEqual(BatchAccountContext contex
             Assert.Equal<string>(context1.TagsTable, context2.TagsTable);
             Assert.Equal<string>(context1.TaskTenantUrl, context2.TaskTenantUrl);
         }
+
+        /// <summary>
+        /// Builds a PSCloudWorkItem for testing
+        /// </summary>
+        public static PSCloudWorkItem CreatePSCloudWorkItem()
+        {
+            BatchAccountContext context = CreateBatchContextWithKeys();
+            using (IWorkItemManager wiManager = context.BatchOMClient.OpenWorkItemManager())
+            {
+                ICloudWorkItem workItem = wiManager.CreateWorkItem("testWorkItem");
+                return new PSCloudWorkItem(workItem);
+            }
+        }
+
+        /// <summary>
+        /// Builds a GetPoolResponse object
+        /// </summary>
+        public static GetPoolResponse CreateGetPoolResponse(string poolName)
+        {
+            GetPoolResponse response = new GetPoolResponse();
+            SetProperty(response, "StatusCode", HttpStatusCode.OK);
+
+            Pool pool = new Pool();
+            SetProperty(pool, "Name", poolName);
+
+            SetProperty(response, "Pool", pool);
+
+            return response;
+        }
+
+        /// <summary>
+        /// Builds a ListPoolsResponse object
+        /// </summary>
+        public static ListPoolsResponse CreateListPoolsResponse(IEnumerable<string> poolNames)
+        {
+            ListPoolsResponse response = new ListPoolsResponse();
+            SetProperty(response, "StatusCode", HttpStatusCode.OK);
+
+            List<Pool> pools = new List<Pool>();
+
+            foreach (string name in poolNames)
+            {
+                Pool pool = new Pool();
+                SetProperty(pool, "Name", name);
+                pools.Add(pool);
+            }
+
+            SetProperty(response, "Pools", pools);
+
+            return response;
+        }
+
+        /// <summary>
+        /// Builds a GetWorkItemResponse object
+        /// </summary>
+        public static GetWorkItemResponse CreateGetWorkItemResponse(string workItemName)
+        {
+            GetWorkItemResponse response = new GetWorkItemResponse();
+            SetProperty(response, "StatusCode", HttpStatusCode.OK);
+
+            JobExecutionEnvironment jee = new JobExecutionEnvironment();
+
+            WorkItem workItem = new WorkItem(workItemName, jee);
+            SetProperty(response, "WorkItem", workItem);
+
+            return response;
+        }
+
+        /// <summary>
+        /// Builds a ListWorkItemsResponse object
+        /// </summary>
+        public static ListWorkItemsResponse CreateListWorkItemsResponse(IEnumerable<string> workItemNames)
+        {
+            ListWorkItemsResponse response = new ListWorkItemsResponse();
+            SetProperty(response, "StatusCode", HttpStatusCode.OK);
+
+            List<WorkItem> workItems = new List<WorkItem>();
+            JobExecutionEnvironment jee = new JobExecutionEnvironment();
+
+            foreach (string name in workItemNames)
+            {
+                workItems.Add(new WorkItem(name, jee));
+            }
+
+            SetProperty(response, "WorkItems", workItems);
+
+            return response;
+        }
+
+        /// <summary>
+        /// Builds a GetJobResponse object
+        /// </summary>
+        public static GetJobResponse CreateGetJobResponse(string jobName)
+        {
+            GetJobResponse response = new GetJobResponse();
+            SetProperty(response, "StatusCode", HttpStatusCode.OK);
+
+            Job job = new Job();
+            SetProperty(job, "Name", jobName);
+
+            SetProperty(response, "Job", job);
+
+            return response;
+        }
+
+        /// <summary>
+        /// Builds a ListJobsResponse object
+        /// </summary>
+        public static ListJobsResponse CreateListJobsResponse(IEnumerable<string> jobNames)
+        {
+            ListJobsResponse response = new ListJobsResponse();
+            SetProperty(response, "StatusCode", HttpStatusCode.OK);
+
+            List<Job> jobs = new List<Job>();
+
+            foreach (string name in jobNames)
+            {
+                Job job = new Job();
+                SetProperty(job, "Name", name);
+                jobs.Add(job);
+            }
+
+            SetProperty(response, "Jobs", jobs);
+
+            return response;
+        }
+
+        /// <summary>
+        /// Builds a GetTaskResponse object
+        /// </summary>
+        public static GetTaskResponse CreateGetTaskResponse(string taskName)
+        {
+            GetTaskResponse response = new GetTaskResponse();
+            SetProperty(response, "StatusCode", HttpStatusCode.OK);
+
+            Azure.Batch.Protocol.Entities.Task task = new Azure.Batch.Protocol.Entities.Task();
+            SetProperty(task, "Name", taskName);
+
+            SetProperty(response, "Task", task);
+
+            return response;
+        }
+
+        /// <summary>
+        /// Builds a ListTasksResponse object
+        /// </summary>
+        public static ListTasksResponse CreateListTasksResponse(IEnumerable<string> taskNames)
+        {
+            ListTasksResponse response = new ListTasksResponse();
+            SetProperty(response, "StatusCode", HttpStatusCode.OK);
+
+            List<Azure.Batch.Protocol.Entities.Task> tasks = new List<Azure.Batch.Protocol.Entities.Task>();
+
+            foreach (string name in taskNames)
+            {
+                Azure.Batch.Protocol.Entities.Task task = new Azure.Batch.Protocol.Entities.Task();
+                SetProperty(task, "Name", name);
+                tasks.Add(task);
+            }
+
+            SetProperty(response, "Tasks", tasks);
+
+            return response;
+        }
+
+
+        /// <summary>
+        /// Uses Reflection to set a property value on an object. Can be used to bypass restricted set accessors.
+        /// </summary>
+        internal static void SetProperty(object obj, string propertyName, object propertyValue)
+        {
+            Type t = obj.GetType();
+            PropertyInfo propInfo = t.GetProperty(propertyName);
+            propInfo.SetValue(obj, propertyValue);
+        }
+
+        /// <summary>
+        /// Uses Reflection to set a property value on an object. 
+        /// </summary>
+        internal static void SetField(object obj, string fieldName, object fieldValue)
+        {
+            Type t = obj.GetType();
+            FieldInfo fieldInfo = t.GetField(fieldName, BindingFlags.NonPublic | BindingFlags.Instance);
+            fieldInfo.SetValue(obj, fieldValue);
+        }
     }
 }