Automation cmdlets for next powershell release

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

@@ -111,9 +111,13 @@
   </ItemGroup>
   <ItemGroup>
     <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="UnitTests\GetAzureAutomationWebhookTest.cs" />
     <Compile Include="UnitTests\NewAzureAutomationAccountTest.cs" />
     <Compile Include="UnitTests\GetAzureAutomationAccountTest.cs" />
+    <Compile Include="UnitTests\NewAzureAutomationWebhookTest.cs" />
     <Compile Include="UnitTests\RemoveAzureAutomationAccountTest.cs" />
+    <Compile Include="UnitTests\RemoveAzureAutomationWebhookTest.cs" />
+    <Compile Include="UnitTests\SetAzureAutomationWebhookTest.cs" />
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\..\..\Common\Commands.Common.Test\Commands.Common.Test.csproj">

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

@@ -50,14 +50,15 @@ public void GetAzureAutomationAllAccountsSuccessfull()
         {
             // Setup
             string resourceGroupName = "resourceGroup";
+            string nextLink = string.Empty;
 
-            this.mockAutomationClient.Setup(f => f.ListAutomationAccounts(resourceGroupName)).Returns((string a) => new List<AutomationAccount>());
+            this.mockAutomationClient.Setup(f => f.ListAutomationAccounts(resourceGroupName, ref nextLink)).Returns((string a, string b) => new List<AutomationAccount>());
 
             // Test
             this.cmdlet.ExecuteCmdlet();
 
             // Assert
-            this.mockAutomationClient.Verify(f => f.ListAutomationAccounts(null), Times.Once());
+            this.mockAutomationClient.Verify(f => f.ListAutomationAccounts(resourceGroupName, ref nextLink), Times.Once());
         }
 
         [TestMethod]

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

@@ -0,0 +1,67 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.Commands.Automation.Cmdlet;
+using Microsoft.Azure.Commands.Automation.Common;
+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;
+
+namespace Microsoft.Azure.Commands.Automation.Test.UnitTests
+{
+    [TestClass]
+    public class GetAzureAutomationWebhookTest : TestBase
+    {
+        private Mock<IAutomationClient> mockAutomationClient;
+
+        private MockCommandRuntime mockCommandRuntime;
+
+        private GetAzureAutomationWebhook cmdlet;
+
+        [TestInitialize]
+        public void SetupTest()
+        {
+            this.mockAutomationClient = new Mock<IAutomationClient>();
+            this.mockCommandRuntime = new MockCommandRuntime();
+            this.cmdlet = new GetAzureAutomationWebhook
+            {
+                AutomationClient = this.mockAutomationClient.Object,
+                CommandRuntime = this.mockCommandRuntime
+            };
+        }
+
+        [TestMethod]
+        public void GetAzureAutomationWebhookByNameSuccessful()
+        {
+            // Setup
+            string resourceGroupName = "resourceGroup";
+            string accountName = "account";
+            string webhookName = "webhookName";
+            this.cmdlet.SetParameterSet("ByName");
+
+            this.mockAutomationClient.Setup(f => f.GetWebhook(resourceGroupName, accountName, webhookName));
+
+            // Test
+            this.cmdlet.ResourceGroupName = resourceGroupName;
+            this.cmdlet.AutomationAccountName = accountName;
+            this.cmdlet.Name = webhookName;
+            this.cmdlet.ExecuteCmdlet();
+
+            // Assert
+            this.mockAutomationClient.Verify(f => f.GetWebhook(resourceGroupName, accountName, webhookName), Times.Once());
+        }
+    }
+}

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

@@ -0,0 +1,75 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.Commands.Automation.Cmdlet;
+using Microsoft.Azure.Commands.Automation.Common;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Microsoft.WindowsAzure.Commands.Common.Test.Mocks;
+using Microsoft.WindowsAzure.Commands.Test.Utilities.Common;
+using Moq;
+using System;
+
+namespace Microsoft.Azure.Commands.Automation.Test.UnitTests
+{
+    [TestClass]
+    public class NewAzureAutomationWebhookTest : TestBase
+    {
+        private Mock<IAutomationClient> mockAutomationClient;
+
+        private MockCommandRuntime mockCommandRuntime;
+
+        private NewAzureAutomationWebhook cmdlet;
+
+        [TestInitialize]
+        public void SetupTest()
+        {
+            this.mockAutomationClient = new Mock<IAutomationClient>();
+            this.mockCommandRuntime = new MockCommandRuntime();
+            this.cmdlet = new NewAzureAutomationWebhook
+                              {
+                                  AutomationClient = this.mockAutomationClient.Object,
+                                  CommandRuntime = this.mockCommandRuntime
+                              };
+        }
+
+        [TestMethod]
+        public void NewAzureAutomationWebhookByNameSuccessful()
+        {
+            // Setup
+            string resourceGroupName = "resourceGroup";
+            string accountName = "account";
+            string name = "webhookName";
+            string runbookName = "runbookName";
+            DateTimeOffset expiryTime = DateTimeOffset.Now.AddDays(1);
+
+            this.mockAutomationClient.Setup(
+                f => f.CreateWebhook(resourceGroupName, accountName, name, runbookName, true, expiryTime, null));
+
+            // Test
+            this.cmdlet.ResourceGroupName = resourceGroupName;
+            this.cmdlet.AutomationAccountName = accountName;
+            this.cmdlet.Name = name;
+            this.cmdlet.RunbookName = runbookName;
+            this.cmdlet.ExpiryTime = expiryTime;
+            this.cmdlet.IsEnabled = true;
+            this.cmdlet.Parameters = null;
+            this.cmdlet.ExecuteCmdlet();
+
+            // Assert
+            this.mockAutomationClient.Verify(
+                f => f.CreateWebhook(resourceGroupName, accountName, name, runbookName, true, expiryTime, null),
+                Times.Once());
+        }
+    }
+}

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

@@ -0,0 +1,67 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.Commands.Automation.Cmdlet;
+using Microsoft.Azure.Commands.Automation.Common;
+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;
+
+namespace Microsoft.Azure.Commands.Automation.Test.UnitTests
+{
+    [TestClass]
+    public class RemoveAzureAutomationWebhookTest : TestBase
+    {
+        private Mock<IAutomationClient> mockAutomationClient;
+
+        private MockCommandRuntime mockCommandRuntime;
+
+        private RemoveAzureAutomationWebhook cmdlet;
+
+        [TestInitialize]
+        public void SetupTest()
+        {
+            this.mockAutomationClient = new Mock<IAutomationClient>();
+            this.mockCommandRuntime = new MockCommandRuntime();
+            this.cmdlet = new RemoveAzureAutomationWebhook
+            {
+                AutomationClient = this.mockAutomationClient.Object,
+                CommandRuntime = this.mockCommandRuntime
+            };
+        }
+
+        [TestMethod]
+        public void RemoveAzureAutomationWebhookByNameSuccessful()
+        {
+            // Setup
+            string resourceGroupName = "resourceGroup";
+            string accountName = "account";
+            string webhookName = "webhookName";
+            this.cmdlet.SetParameterSet("ByName");
+
+            this.mockAutomationClient.Setup(f => f.DeleteWebhook(resourceGroupName, accountName, webhookName));
+
+            // Test
+            this.cmdlet.ResourceGroupName = resourceGroupName;
+            this.cmdlet.AutomationAccountName = accountName;
+            this.cmdlet.Name = webhookName;
+            this.cmdlet.ExecuteCmdlet();
+
+            // Assert
+            this.mockAutomationClient.Verify(f => f.DeleteWebhook(resourceGroupName, accountName, webhookName), Times.Once());
+        }
+    }
+}

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

@@ -0,0 +1,70 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.Commands.Automation.Cmdlet;
+using Microsoft.Azure.Commands.Automation.Common;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Microsoft.WindowsAzure.Commands.Common.Test.Mocks;
+using Microsoft.WindowsAzure.Commands.Test.Utilities.Common;
+using Moq;
+
+namespace Microsoft.Azure.Commands.Automation.Test.UnitTests
+{
+    [TestClass]
+    public class SetAzureAutomationWebhookTest : TestBase
+    {
+        private Mock<IAutomationClient> mockAutomationClient;
+
+        private MockCommandRuntime mockCommandRuntime;
+
+        private SetAzureAutomationWebhook cmdlet;
+
+        [TestInitialize]
+        public void SetupTest()
+        {
+            this.mockAutomationClient = new Mock<IAutomationClient>();
+            this.mockCommandRuntime = new MockCommandRuntime();
+            this.cmdlet = new SetAzureAutomationWebhook
+            {
+                AutomationClient = this.mockAutomationClient.Object,
+                CommandRuntime = this.mockCommandRuntime
+            };
+        }
+
+        [TestMethod]
+        public void SetAzureAutomationWebhookToDisabledSuccessful()
+        {
+            // Setup
+            string resourceGroupName = "resourceGroup";
+            string accountName = "account";
+            string name = "webhookName";
+
+            this.mockAutomationClient.Setup(
+                f => f.UpdateWebhook(resourceGroupName, accountName, name, null, false));
+
+            // Test
+            this.cmdlet.ResourceGroupName = resourceGroupName;
+            this.cmdlet.AutomationAccountName = accountName;
+            this.cmdlet.Name = name;
+            this.cmdlet.IsEnabled = false;
+            this.cmdlet.Parameters = null;
+            this.cmdlet.ExecuteCmdlet();
+
+            // Assert
+            this.mockAutomationClient.Verify(
+                f => f.UpdateWebhook(resourceGroupName, accountName, name, null, false),
+                Times.Once());
+        }
+    }
+}

src/ResourceManager/Automation/Commands.Automation/Cmdlet/GetAzureAutomationAccount.cs

@@ -52,14 +52,14 @@ public IAutomationClient AutomationClient
         /// <summary>
         /// Gets or sets the automation account name.
         /// </summary>
-        [Parameter(Position = 0, Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "The resource group name.")]
-        [ValidateNotNullOrEmpty]
+        [Parameter(ParameterSetName = AutomationCmdletParameterSets.ByAll, Position = 0, Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The resource group name.")]
+        [Parameter(ParameterSetName = AutomationCmdletParameterSets.ByAutomationAccountName, Position = 0, Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "The resource group name.")]
         public string ResourceGroupName { get; set; }
 
         /// <summary>
         /// Gets or sets the automation account name.
         /// </summary>
-        [Parameter(ParameterSetName = AutomationCmdletParameterSets.ByAutomationAccountName, Position = 1, Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The automation account name.")]
+        [Parameter(ParameterSetName = AutomationCmdletParameterSets.ByAutomationAccountName, Position = 1, Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "The automation account name.")]
         [Alias("AutomationAccountName")]
         [ValidateNotNullOrEmpty]
         public string Name { get; set; }
@@ -77,13 +77,19 @@ public override void ExecuteCmdlet()
                 { 
                    this.AutomationClient.GetAutomationAccount(this.ResourceGroupName, this.Name)
                 };
+                this.WriteObject(ret, true);
             }
             else
             {
-                ret = this.AutomationClient.ListAutomationAccounts(this.ResourceGroupName);
-            }
+                string nextLink = string.Empty;
 
-            this.WriteObject(ret, true);
+                do
+                {
+                    ret = this.AutomationClient.ListAutomationAccounts(this.ResourceGroupName, ref nextLink);
+                    this.WriteObject(ret, true);
+
+                } while (!string.IsNullOrEmpty(nextLink));
+            }
         }
     }
 }

src/ResourceManager/Automation/Commands.Automation/Cmdlet/GetAzureAutomationModule.cs

@@ -48,13 +48,19 @@ protected override void AutomationExecuteCmdlet()
                 { 
                    this.AutomationClient.GetModule(this.ResourceGroupName, this.AutomationAccountName, this.Name)
                 };
+                this.GenerateCmdletOutput(ret);
             }
             else
             {
-                ret = this.AutomationClient.ListModules(this.ResourceGroupName, this.AutomationAccountName);
-            }
+                string nextLink = string.Empty;
+
+                do
+                {
+                    ret = this.AutomationClient.ListModules(this.ResourceGroupName, this.AutomationAccountName, ref nextLink);
+                    this.GenerateCmdletOutput(ret);
 
-            this.GenerateCmdletOutput(ret);
+                } while (!string.IsNullOrEmpty(nextLink));
+            }
         }
     }
 }