Export - Import cmdlets

Author: safeermohammed

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

@@ -41,18 +41,18 @@ public class ExportAzureAutomationRunbook : AzureAutomationBaseCmdlet
         /// <summary>
         /// Gets or sets the runbook version type
         /// </summary>
-        [Parameter(Mandatory = false, HelpMessage = "Returns the draft or the published runbook version only. If not set, returns published.")]
+        [Parameter(Mandatory = false, HelpMessage = "Specifies what version of the runbook, draft or published, should be returned.")]
         [ValidateSet(Constants.Published, Constants.Draft, IgnoreCase = true)]
         public string Slot { get; set; }
 
         /// <summary>
         /// Gets or sets the output folder for the configuration script.
         /// </summary>
-        [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The folder where runbook definition should be placed.")]
+        [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The folder where the runbook should be placed.")]
         public string OutputFolder { get; set; }
 
         /// <summary>
-        /// Gets or sets switch parameter to confirm overwriting of existing configuration script.
+        /// Gets or sets switch parameter to confirm overwriting of existing local runbook file.
         /// </summary>
         [Parameter(Mandatory = false, HelpMessage = "Forces an overwrite of an existing local file with the same name.")]
         public SwitchParameter Force { get; set; }

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

@@ -53,8 +53,9 @@ public class ImportAzureAutomationRunbook : AzureAutomationBaseCmdlet
         /// <summary>
         /// Gets or sets the runbook version type
         /// </summary>
-        [Parameter(Mandatory = false, HelpMessage = "Runbook definition type.")]
+        [Parameter(Mandatory = true, HelpMessage = "Runbook definition type.")]
         [ValidateSet(Constants.RunbookType.Graph, Constants.RunbookType.PowerShell, Constants.RunbookType.PowerShellWorkflow, IgnoreCase = true)]
+        [ValidateNotNullOrEmpty]
         public string Type { get; set; }
 
         /// <summary>

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

@@ -52,8 +52,9 @@ public class NewAzureAutomationRunbook : AzureAutomationBaseCmdlet
         /// <summary>
         /// Gets or sets the runbook version type
         /// </summary>
-        [Parameter(Mandatory = false, HelpMessage = "Runbook definition type.")]
+        [Parameter(Mandatory = true, HelpMessage = "Runbook definition type.")]
         [ValidateSet(Constants.RunbookType.Graph, Constants.RunbookType.PowerShell, Constants.RunbookType.PowerShellWorkflow, IgnoreCase = true)]
+        [ValidateNotNullOrEmpty]
         public string Type { get; set; }
 
         /// <summary>

src/ResourceManager/Automation/Commands.Automation/Common/AutomationClient.cs

@@ -1715,8 +1715,6 @@ private DirectoryInfo WriteRunbookToFile(string outputFolder, string runbookName
 
             var fileExtension = (0 == string.Compare(runbookType, Constants.RunbookType.Graph, StringComparison.OrdinalIgnoreCase)) ? Constants.SupportedFileExtensions.Graph : Constants.SupportedFileExtensions.PowerShellScript;
 
-            fileExtension = "." + fileExtension;
-
             var outputFilePath = outputFolderFullPath + "\\" + runbookName + fileExtension;
 
             // file exists and overwrite Not specified