Enum for Plan

Author: safeermohammed

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

@@ -17,8 +17,9 @@
 using System.Management.Automation;
 using System.Security.Permissions;
 using Microsoft.Azure.Commands.Automation.Common;
-using Microsoft.Azure.Commands.Automation.Model;
+using Microsoft.Azure.Management.Automation.Models;
 using Microsoft.WindowsAzure.Commands.Utilities.Common;
+using AutomationAccount = Microsoft.Azure.Commands.Automation.Model.AutomationAccount;
 
 namespace Microsoft.Azure.Commands.Automation.Cmdlet
 {
@@ -75,7 +76,7 @@ public IAutomationClient AutomationClient
         /// Gets or sets the plan.
         /// </summary>
         [Parameter( Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The plan of the automation account")]
-        [ValidateSet("Free", "Basic", IgnoreCase = true)]
+        [ValidateSet(SkuNameEnum.Free, SkuNameEnum.Basic, IgnoreCase = true)]
         public string Plan { get; set; }
 
         /// <summary>

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

@@ -17,8 +17,9 @@
 using System.Management.Automation;
 using System.Security.Permissions;
 using Microsoft.Azure.Commands.Automation.Common;
-using Microsoft.Azure.Commands.Automation.Model;
+using Microsoft.Azure.Management.Automation.Models;
 using Microsoft.WindowsAzure.Commands.Utilities.Common;
+using AutomationAccount = Microsoft.Azure.Commands.Automation.Model.AutomationAccount;
 
 namespace Microsoft.Azure.Commands.Automation.Cmdlet
 {
@@ -69,7 +70,7 @@ public IAutomationClient AutomationClient
         /// Gets or sets the plan.
         /// </summary>
         [Parameter( Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The plan of the automation account")]
-        [ValidateSet("Free", "Basic", IgnoreCase = true)]
+        [ValidateSet(SkuNameEnum.Free, SkuNameEnum.Basic, IgnoreCase = true)]
         public string Plan { get; set; }
 
         /// <summary>