update test for add-azurevmimage

Author: huangpf

src/ServiceManagement/Compute/Commands.ServiceManagement.Test/FunctionalTests/FunctionalTestCommonVhd.cs

@@ -148,10 +148,14 @@ public void AzureVMImageTest()
 
             string oldLabel = "old label";
             string newLabel = "new label";
+            string vmSize = "Small";
+            var iconUri = "http://www.bing.com";
+            var smallIconUri = "http://www.bing.com";
+            bool showInGui = true;
 
             try
             {
-                OSImageContext result = vmPowershellCmdlets.AddAzureVMImage(newImageName, mediaLocation, OS.Windows, oldLabel);
+                OSImageContext result = vmPowershellCmdlets.AddAzureVMImage(newImageName, mediaLocation, OS.Windows, oldLabel, vmSize, iconUri, smallIconUri, showInGui);
 
                 OSImageContext resultReturned = vmPowershellCmdlets.GetAzureVMImage(newImageName)[0];
                 Assert.IsTrue(!string.IsNullOrEmpty(resultReturned.IOType));

src/ServiceManagement/Compute/Commands.ServiceManagement.Test/FunctionalTests/IaasCmdletInfo/AddAzureVMImageCmdletInfo.cs

@@ -43,6 +43,25 @@ public AddAzureVMImageCmdletInfo(string imageName, string mediaLocation, OS os,
             }
         }
 
+        public AddAzureVMImageCmdletInfo(string imageName, string mediaLocation, OS os, string label, string recommendedSize, string iconUri, string smallIconUri, bool showInGui)
+            : this(imageName, mediaLocation, os, label, recommendedSize)
+        {
+            if (!string.IsNullOrEmpty(iconUri))
+            {
+                cmdletParams.Add(new CmdletParam("IconUri", iconUri));
+            }
+
+            if (!string.IsNullOrEmpty(iconUri))
+            {
+                cmdletParams.Add(new CmdletParam("SmallIconUri", smallIconUri));
+            }
+
+            if (showInGui)
+            {
+                cmdletParams.Add(new CmdletParam("ShowInGui"));
+            }
+        }
+
         public AddAzureVMImageCmdletInfo(
             string imageName,
             string mediaLocation,

src/ServiceManagement/Compute/Commands.ServiceManagement.Test/FunctionalTests/ServiceManagementCmdletTestHelper.cs

@@ -1324,11 +1324,11 @@ public ManagementOperationContext UpdateAzureVM(string vmName, string serviceNam
 
         #region AzureVMImage
 
-        public SM.OSImageContext AddAzureVMImage(string imageName, string mediaLocation, OS os, string label = null, string recommendedSize = null)
+        public SM.OSImageContext AddAzureVMImage(string imageName, string mediaLocation, OS os, string label = null, string recommendedSize = null, string iconUri = null, string smallIconUri = null, bool showInGui = false)
         {
             SM.OSImageContext result = new SM.OSImageContext();
             Utilities.RetryActionUntilSuccess(
-                () => result = RunPSCmdletAndReturnFirst<SM.OSImageContext>(new AddAzureVMImageCmdletInfo(imageName, mediaLocation, os, label, recommendedSize)),
+                () => result = RunPSCmdletAndReturnFirst<SM.OSImageContext>(new AddAzureVMImageCmdletInfo(imageName, mediaLocation, os, label, recommendedSize, iconUri, smallIconUri, showInGui)),
                 "409", 3, 60);
             return result;
         }