Dev: Azure Automation cmdlets

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

@@ -157,6 +157,7 @@
     <Compile Include="UnitTests\RegisterAzureAutomationScheduledRunbookTest.cs" />
     <Compile Include="UnitTests\RemoveAzureAutomationAccountTest.cs" />
     <Compile Include="UnitTests\RemoveAzureAutomationCertificateTest.cs" />
+    <Compile Include="UnitTests\RemoveAzureAutomationConnectionTypeTest.cs" />
     <Compile Include="UnitTests\RemoveAzureAutomationConnectionTest.cs" />
     <Compile Include="UnitTests\RemoveAzureAutomationModuleTest.cs" />
     <Compile Include="UnitTests\RemoveAzureAutomationRunbookTest.cs" />

src/ResourceManager/Automation/Commands.Automation.Test/UnitTests/RemoveAzureAutomationConnectionTypeTest.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.ScenarioTest;
+using Microsoft.WindowsAzure.Commands.Test.Utilities.Common;
+using Moq;
+
+namespace Microsoft.Azure.Commands.ResourceManager.Automation.Test.UnitTests
+{
+    [TestClass]
+    public class RemoveAzureAutomationConnectionTypeTest : RMTestBase
+    {
+        private Mock<IAutomationClient> mockAutomationClient;
+
+        private MockCommandRuntime mockCommandRuntime;
+
+        private RemoveAzureAutomationConnectionType cmdlet;
+
+        [TestInitialize]
+        public void SetupTest()
+        {
+            this.mockAutomationClient = new Mock<IAutomationClient>();
+            this.mockCommandRuntime = new MockCommandRuntime();
+            this.cmdlet = new RemoveAzureAutomationConnectionType
+            {
+                AutomationClient = this.mockAutomationClient.Object,
+                CommandRuntime = this.mockCommandRuntime
+            };
+        }
+
+        [TestMethod]
+        public void RemoveAzureAutomationConnectionTypeByNameSuccessfull()
+        {
+            // Setup
+            string resourceGroupName = "resourceGroup";
+            string accountName = "automation";
+            string connectionTypeName = "connectionType";
+
+            this.mockAutomationClient.Setup(f => f.DeleteConnectionType(resourceGroupName, accountName, connectionTypeName));
+
+            // Test
+            this.cmdlet.ResourceGroupName = resourceGroupName;
+            this.cmdlet.AutomationAccountName = accountName;
+            this.cmdlet.Name = connectionTypeName;
+            this.cmdlet.Force = true;
+            this.cmdlet.ExecuteCmdlet();
+
+            // Assert
+            this.mockAutomationClient.Verify(f => f.DeleteConnectionType(resourceGroupName, accountName, connectionTypeName), Times.Once());
+        }
+    }
+}

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

@@ -0,0 +1,78 @@
+// ----------------------------------------------------------------------------------
+//
+// 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 System;
+using System.Collections;
+using System.Collections.Generic;
+using System.Management.Automation;
+using System.Security.Permissions;
+using Microsoft.Azure.Commands.Automation.Common;
+using Microsoft.Azure.Commands.Automation.Model;
+using Microsoft.WindowsAzure.Commands.Utilities.Common;
+
+namespace Microsoft.Azure.Commands.Automation.Cmdlet
+{
+    /// <summary>
+    /// Imports dsc node configuration script
+    /// </summary>
+    [Cmdlet(VerbsData.Import, "AzureRmAutomationDscNodeConfiguration")]
+    [OutputType(typeof(NodeConfiguration))]
+    public class ImportAzureAutomationDscNodeConfiguration : AzureAutomationBaseCmdlet
+    {
+        /// <summary>
+        /// True to overwrite the existing configuration; false otherwise.
+        /// </summary>        
+        private bool overwriteExistingConfiguration;
+
+        /// <summary>
+        /// Gets or sets the source path.
+        /// </summary>
+        [Parameter(Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "Path to the node configuration .mof to import.")]        
+        [ValidateNotNullOrEmpty]
+        public string Path { get; set; }
+
+        /// <summary>
+        /// Gets or sets the configuration name for the node configuration.
+        /// </summary>
+        [Parameter(Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "The name of the DSC Configuration to import the Node Configuration under. All Node Configurations in Azure Automation must exist under a Configuration. The name of the Configuration will become the namespace of the imported Node Configuration, in the form of 'ConfigurationName.MofFileName'")]        
+        public string ConfigurationName { get; set; }
+
+
+        /// <summary>
+        /// Gets or sets switch parameter to confirm overwriting of existing configurations.
+        /// </summary>
+        [Parameter(Mandatory = false, HelpMessage = "Forces the command to overwrite an existing Node Configuration.")]
+        public SwitchParameter Force
+        {
+            get { return this.overwriteExistingConfiguration; }
+            set { this.overwriteExistingConfiguration = value; }
+        }
+
+        /// <summary>
+        /// Execute this cmdlet.
+        /// </summary>
+        [PermissionSet(SecurityAction.Demand, Name = "FullTrust")]
+        protected override void ProcessRecord()
+        {
+            var nodeConfiguration = this.AutomationClient.CreateNodeConfiguration(
+                    this.ResourceGroupName,
+                    this.AutomationAccountName,
+                    this.Path,
+                    this.ConfigurationName,
+                    this.Force);
+
+            this.WriteObject(nodeConfiguration);
+        }
+    }
+}

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

@@ -43,6 +43,13 @@ public class ImportAzureAutomationRunbook : AzureAutomationBaseCmdlet
         [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The runbook description.")]
         public string Description { get; set; }
 
+        /// <summary>
+        /// Gets or sets the runbook name
+        /// </summary>
+        [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The name of the runbook to import, if different from the file name. Not supported for PowerShell Workflow runbooks.")]
+        [Alias("RunbookName")]
+        public string Name { get; set; }
+
         /// <summary>
         /// Gets or sets the runbook tags.
         /// </summary>
@@ -98,7 +105,8 @@ protected override void AutomationProcessRecord()
                     this.LogProgress, 
                     this.LogVerbose,
                     this.Published.IsPresent,
-                    this.Force.IsPresent);
+                    this.Force.IsPresent,
+                    this.Name);
 
             this.WriteObject(runbook);
         }

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

@@ -16,7 +16,7 @@
 using Microsoft.Azure.Commands.Automation.Properties;
 using Microsoft.WindowsAzure.Commands.Common;
 using System;
-using System.Collections.Generic;
+using System.Collections;
 using System.Management.Automation;
 using System.Security.Permissions;
 
@@ -66,7 +66,7 @@ public class NewAzureAutomationWebhook : AzureAutomationBaseCmdlet
         /// </summary>
         [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true,
             HelpMessage = "The Runbook parameters name/value.")]
-        public IDictionary<string, string> Parameters { get; set; }
+        public IDictionary Parameters { get; set; }
 
         [Parameter(Mandatory = false, HelpMessage = "Skip warning message about one-time viewable webhook URL")]
         public SwitchParameter Force { get; set; }
@@ -91,7 +91,7 @@ protected override void AutomationProcessRecord()
                         this.RunbookName,
                         this.IsEnabled,
                         this.ExpiryTime,
-                        this.Parameters.ToHashtable())));
+                        this.Parameters)));
 
         }
     }

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

@@ -0,0 +1,73 @@
+// ----------------------------------------------------------------------------------
+//
+// 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 System;
+using System.Linq;
+using System.Management.Automation;
+using System.Security.Permissions;
+using Microsoft.Azure.Commands.Automation.Common;
+using Microsoft.Azure.Commands.Automation.Properties;
+
+namespace Microsoft.Azure.Commands.Automation.Cmdlet
+{
+    /// <summary>
+    /// Removes a Connection type for automation.
+    /// </summary>
+    [Cmdlet(VerbsCommon.Remove, "AzureRmAutomationConnectionType", DefaultParameterSetName = AutomationCmdletParameterSets.ByName)]
+    public class RemoveAzureAutomationConnectionType : AzureAutomationBaseCmdlet
+    {
+        /// <summary>
+        /// Gets or sets the connection name.
+        /// </summary>
+        [Parameter(ParameterSetName = AutomationCmdletParameterSets.ByName, Position = 2, Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "The connection type name.")]
+        [ValidateNotNullOrEmpty]
+        public string Name { get; set; }
+
+        [Parameter(ParameterSetName = AutomationCmdletParameterSets.ByName, Position = 3, HelpMessage = "Confirm the removal of the connection type")]
+        public SwitchParameter Force { get; set; }
+
+        /// <summary>
+        /// Execute this cmdlet.
+        /// </summary>
+        [PermissionSet(SecurityAction.Demand, Name = "FullTrust")]
+        protected override void AutomationProcessRecord()
+        {
+            var nextLink = string.Empty;
+            var removeMessageWarning = Resources.RemovingAzureAutomationResourceWarning;
+
+            // check if any connections exists that use this connection type
+            do
+            {
+                var ret = this.AutomationClient.ListConnections(this.ResourceGroupName, this.AutomationAccountName, ref nextLink);
+
+                if (ret.ToList().Any(connection => 0 ==
+                                                   string.Compare(connection.ConnectionTypeName, this.Name,
+                                                       StringComparison.CurrentCultureIgnoreCase)))
+                {
+                    removeMessageWarning = Resources.RemoveConnectionTypeThatHasConnectionWarning;
+                    break;
+                }
+
+            } while (!string.IsNullOrEmpty(nextLink));
+
+
+            ConfirmAction(
+                       Force.IsPresent,
+                       string.Format(removeMessageWarning, "ConnectionType"),
+                       string.Format(Resources.RemoveAzureAutomationResourceDescription, "ConnectionType"),
+                       Name,
+                       () => this.AutomationClient.DeleteConnectionType(this.ResourceGroupName, this.AutomationAccountName, Name));
+        }
+    }
+}

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

@@ -12,9 +12,9 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using System.Collections;
 using Microsoft.Azure.Commands.Automation.Model;
 using Microsoft.WindowsAzure.Commands.Common;
-using System.Collections.Generic;
 using System.Management.Automation;
 using System.Security.Permissions;
 
@@ -48,7 +48,7 @@ public class SetAzureAutomationWebhook : AzureAutomationBaseCmdlet
         /// </summary>
         [Parameter(Position = 4, Mandatory = false, ValueFromPipelineByPropertyName = true,
             HelpMessage = "The Runbook parameters name/value.")]
-        public IDictionary<string, string> Parameters { get; set; }
+        public IDictionary Parameters { get; set; }
 
         /// <summary>
         /// Execute this cmdlet.
@@ -60,7 +60,7 @@ protected override void AutomationProcessRecord()
                 this.ResourceGroupName,
                 this.AutomationAccountName,
                 this.Name,
-                this.Parameters.ToHashtable(),
+                this.Parameters,
                 this.IsEnabled);
             this.WriteObject(updatedWebhook);
         }

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

@@ -12,12 +12,15 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using System;
 using System.Collections;
 using System.Collections.Generic;
+using System.Globalization;
 using System.Management.Automation;
 using System.Security.Permissions;
 using Microsoft.Azure.Commands.Automation.Common;
 using Microsoft.Azure.Commands.Automation.Model;
+using Microsoft.Azure.Commands.Automation.Properties;
 using Microsoft.WindowsAzure.Commands.Utilities.Common;
 
 namespace Microsoft.Azure.Commands.Automation.Cmdlet
@@ -42,6 +45,12 @@ public class StartAzureAutomationDscCompilationJob : AzureAutomationBaseCmdlet
         [Parameter(Mandatory = false, HelpMessage = "The compilation job parameters.")]
         public IDictionary Parameters { get; set; }
 
+        /// <summary>
+        /// Gets or sets the configuration data.
+        /// </summary>
+        [Parameter(Mandatory = false, HelpMessage = "The compilation job configuration data.")]
+        public IDictionary ConfigurationData { get; set; }
+
         /// <summary>
         /// Execute this cmdlet.
         /// </summary>
@@ -50,7 +59,15 @@ protected override void AutomationProcessRecord()
         {
             CompilationJob job = null;
 
-            job = this.AutomationClient.StartCompilationJob(this.ResourceGroupName, this.AutomationAccountName, this.ConfigurationName, this.Parameters);
+            if (this.Parameters != null && this.Parameters.Contains("ConfigurationData"))
+            {
+                throw new ArgumentException(
+                                          string.Format(
+                                              CultureInfo.CurrentCulture,
+                                              Resources.ConfigurationDataShouldNotBeInJobParameters, "-ConfigurationData"));
+            }
+
+            job = this.AutomationClient.StartCompilationJob(this.ResourceGroupName, this.AutomationAccountName, this.ConfigurationName, this.Parameters, this.ConfigurationData);
 
             this.WriteObject(job);
         }

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

@@ -63,6 +63,10 @@
       <HintPath>..\..\..\packages\Microsoft.Azure.Common.Authentication.1.3.1-preview\lib\net45\Microsoft.Azure.Common.Authentication.dll</HintPath>
       <Private>True</Private>
     </Reference>
+    <Reference Include="Microsoft.Azure.Management.Automation, Version=0.9.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
+      <SpecificVersion>False</SpecificVersion>
+      <HintPath>..\..\..\packages\Microsoft.Azure.Management.Automation.0.50.2-prerelease\lib\portable-net45+wp8+wpa81+win\Microsoft.Azure.Management.Automation.dll</HintPath>
+    </Reference>
     <Reference Include="Microsoft.Azure.ResourceManager, Version=2.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
       <SpecificVersion>False</SpecificVersion>
       <HintPath>..\..\..\packages\Microsoft.Azure.Management.Resources.2.18.7-preview\lib\net40\Microsoft.Azure.ResourceManager.dll</HintPath>
@@ -79,10 +83,6 @@
       <SpecificVersion>False</SpecificVersion>
       <HintPath>..\..\..\packages\Microsoft.Azure.Common.2.1.0\lib\net45\Microsoft.Azure.Common.NetFramework.dll</HintPath>
     </Reference>
-    <Reference Include="Microsoft.Azure.Management.Automation, Version=0.9.0.0, Culture=neutral, processorArchitecture=MSIL">
-      <SpecificVersion>False</SpecificVersion>
-      <HintPath>..\..\..\packages\Microsoft.Azure.Management.Automation.0.50.1-prerelease\lib\net40\Microsoft.Azure.Management.Automation.dll</HintPath>
-    </Reference>
     <Reference Include="Microsoft.Threading.Tasks, Version=1.0.12.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL">
       <SpecificVersion>False</SpecificVersion>
       <HintPath>..\..\..\packages\Microsoft.Bcl.Async.1.0.168\lib\net40\Microsoft.Threading.Tasks.dll</HintPath>
@@ -127,6 +127,7 @@
   </ItemGroup>
   <ItemGroup>
     <Compile Include="Cmdlet\AzureAutomationBaseCmdlet.cs" />
+    <Compile Include="Cmdlet\ImportAzureAutomationDscNodeConfiguration.cs" />
     <Compile Include="Cmdlet\ExportAzureAutomationDscConfiguration.cs" />
     <Compile Include="Cmdlet\ExportAzureAutomationDscNodeReportContent.cs" />
     <Compile Include="Cmdlet\GetAzureAutomationCertificate.cs" />
@@ -159,6 +160,7 @@
     <Compile Include="Cmdlet\RegisterAzureAutomationDscNode.cs" />
     <Compile Include="Cmdlet\RegisterAzureAutomationScheduledRunbook.cs" />
     <Compile Include="Cmdlet\RemoveAzureAutomationCertificate.cs" />
+    <Compile Include="Cmdlet\RemoveAzureAutomationConnectionType.cs" />
     <Compile Include="Cmdlet\RemoveAzureAutomationConnection.cs" />
     <Compile Include="Cmdlet\RemoveAzureAutomationCredential.cs" />
     <Compile Include="Cmdlet\RemoveAzureAutomationModule.cs" />