add Unit Test

Author: robplank

src/ResourceManager/Automation/Commands.Automation.Test/Commands.Automation.Test.csproj

@@ -102,7 +102,9 @@
       <SpecificVersion>False</SpecificVersion>
       <HintPath>..\..\..\packages\Microsoft.Bcl.Async.1.0.168\lib\net40\Microsoft.Threading.Tasks.Extensions.Desktop.dll</HintPath>
     </Reference>
-    <Reference Include="Microsoft.VisualStudio.QualityTools.UnitTestFramework, Version=10.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL" />
+    <Reference Include="Microsoft.VisualStudio.QualityTools.UnitTestFramework, Version=10.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL">
+      <Private>False</Private>
+    </Reference>
     <Reference Include="Microsoft.WindowsAzure.Management, Version=4.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
       <SpecificVersion>False</SpecificVersion>
       <HintPath>..\..\..\packages\Microsoft.WindowsAzure.Management.4.1.1\lib\net40\Microsoft.WindowsAzure.Management.dll</HintPath>
@@ -189,6 +191,7 @@
     <Compile Include="UnitTests\StartAzureAutomationRunbookTest.cs" />
     <Compile Include="UnitTests\StopAzureAutomationJobTest.cs" />
     <Compile Include="UnitTests\SuspendAzureAutomationJobTest.cs" />
+    <Compile Include="UnitTests\GetAzureAutomationHybridWorkerGroupTest.cs" />
     <Compile Include="UnitTests\UnregisterAzureAutomationScheduledRunbookTest.cs" />
   </ItemGroup>
   <ItemGroup>

src/ResourceManager/Automation/Commands.Automation.Test/UnitTests/GetAzureAutomationHybridWorkerGroupTest.cs

@@ -0,0 +1,75 @@
+using Microsoft.Azure.Commands.Automation.Cmdlet;
+using Microsoft.Azure.Commands.Automation.Common;
+using Microsoft.Azure.Commands.Automation.Model;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Microsoft.WindowsAzure.Commands.Common.Test.Mocks;
+using Microsoft.WindowsAzure.Commands.Test.Utilities.Common;
+using Microsoft.WindowsAzure.Commands.Utilities.Common;
+using Moq;
+using System.Collections.Generic;
+
+namespace Microsoft.Azure.Commands.ResourceManager.Automation.Test.UnitTests
+{
+    [TestClass]
+    public class GetAzureAutomationHybridWorkerGroupTest : RMTestBase
+    {
+        private Mock<IAutomationClient> mockAutomationClient;
+
+        private MockCommandRuntime mockCommandRuntime;
+
+        private GetAzureAutomationHybridWorkerGroup cmdlet;
+
+        [TestInitialize]
+        public void SetupTest()
+        {
+            this.mockAutomationClient = new Mock<IAutomationClient>();
+            this.mockCommandRuntime = new MockCommandRuntime();
+            this.cmdlet = new GetAzureAutomationHybridWorkerGroup
+            {
+                AutomationClient = this.mockAutomationClient.Object,
+                CommandRuntime = this.mockCommandRuntime
+            };
+        }
+
+        [TestMethod]
+        public void GetAzureAutomationHybridWorkerGroupByNameSuccessfull()
+        {
+            //Setup
+            string resourceGroupName = "resourceGroup";
+            string accountName = "automation";
+            string hybridRunbookWorkerGroupName = "hybridRunbookWorkerGroup";
+
+            this.mockAutomationClient.Setup(f => f.GetHybridRunbookWorkerGroup(resourceGroupName, accountName, hybridRunbookWorkerGroupName));
+
+            // Test
+            this.cmdlet.ResourceGroupName = resourceGroupName;
+            this.cmdlet.AutomationAccountName = accountName;
+            this.cmdlet.Name = hybridRunbookWorkerGroupName;
+            this.cmdlet.SetParameterSet("ByName");
+            this.cmdlet.ExecuteCmdlet();
+
+            // Assert
+            this.mockAutomationClient.Verify(f => f.GetHybridRunbookWorkerGroup(resourceGroupName, accountName, hybridRunbookWorkerGroupName), Times.Once());
+        }
+
+        [TestMethod]
+        public void GetAzureAutomationHybridWorkerGroupByAllSuccessfull()
+        {
+            // Setup
+            string resourceGroupName = "resourceGroup";
+            string accountName = "automation";
+            string nextLink = string.Empty;
+
+            this.mockAutomationClient.Setup(f => f.ListHybridRunbookWorkerGroups(resourceGroupName, accountName, ref nextLink)).Returns((string a, string b, string c) => new List<HybridRunbookWorkerGroup>()); ;
+
+            // Test
+            this.cmdlet.ResourceGroupName = resourceGroupName;
+            this.cmdlet.AutomationAccountName = accountName;
+            this.cmdlet.SetParameterSet("ByAll");
+            this.cmdlet.ExecuteCmdlet();
+
+            //Assert
+            this.mockAutomationClient.Verify(f => f.ListHybridRunbookWorkerGroups(resourceGroupName, accountName, ref nextLink), Times.Once());
+        }
+    }
+}