update fix

huangpf · huangpf · commit 68487ec2d2bb · 2015-01-16T13:36:51.000-08:00
diff --git a/src/ServiceManagement/Compute/Commands.ServiceManagement.Test/FunctionalTests/ServiceManagementCmdletTestHelper.cs b/src/ServiceManagement/Compute/Commands.ServiceManagement.Test/FunctionalTests/ServiceManagementCmdletTestHelper.cs
@@ -1414,7 +1414,7 @@ public void SaveAzureVMImage(string serviceName, string vmName, string newImageN
 
         public Collection<SM.VMImageContext> GetAzureVMImageReturningVMImages(string imageName = null)
         {
-            Collection<SM.OSImageContext> images = GetAzureVMImage();
+            Collection<SM.OSImageContext> images = GetAzureVMImage(imageName);
             Collection<SM.VMImageContext> vmImages = new Collection<SM.VMImageContext>();
             foreach (SM.OSImageContext image in images)
             {
diff --git a/src/ServiceManagement/Compute/Commands.ServiceManagement.Test/FunctionalTests/VMTemplateTests.cs b/src/ServiceManagement/Compute/Commands.ServiceManagement.Test/FunctionalTests/VMTemplateTests.cs
@@ -760,7 +760,7 @@ public void VerifyVMImage(string vmImageName, OS ImageFamily, string imageLabel,
         {
             var vmImages = vmPowershellCmdlets.GetAzureVMImageReturningVMImages(vmImageName);
             Assert.IsTrue(vmImages.Count >= 1);
-            var vmImageInfo = vmImages.First(t => t.ImageName == vmImageName);
+            var vmImageInfo = vmImages[0];
             Utilities.PrintContext(vmImageInfo);
             Utilities.PrintContext(vmImageInfo.OSDiskConfiguration);
             foreach (var disk in vmImageInfo.DataDiskConfigurations)

Original file line number	Diff line number	Diff line change
`@@ -1414,7 +1414,7 @@ public void SaveAzureVMImage(string serviceName, string vmName, string newImageN`
`1414`	`1414`
`1415`	`1415`	`public Collection<SM.VMImageContext> GetAzureVMImageReturningVMImages(string imageName = null)`
`1416`	`1416`	`{`
`1417`		`- Collection<SM.OSImageContext> images = GetAzureVMImage();`
	`1417`	`+ Collection<SM.OSImageContext> images = GetAzureVMImage(imageName);`
`1418`	`1418`	`Collection<SM.VMImageContext> vmImages = new Collection<SM.VMImageContext>();`
`1419`	`1419`	`foreach (SM.OSImageContext image in images)`
`1420`	`1420`	`{`
Original file line number	Diff line number	Diff line change
`@@ -760,7 +760,7 @@ public void VerifyVMImage(string vmImageName, OS ImageFamily, string imageLabel,`
`760`	`760`	`{`
`761`	`761`	`var vmImages = vmPowershellCmdlets.GetAzureVMImageReturningVMImages(vmImageName);`
`762`	`762`	`Assert.IsTrue(vmImages.Count >= 1);`
`763`		`- var vmImageInfo = vmImages.First(t => t.ImageName == vmImageName);`
	`763`	`+ var vmImageInfo = vmImages[0];`
`764`	`764`	`Utilities.PrintContext(vmImageInfo);`
`765`	`765`	`Utilities.PrintContext(vmImageInfo.OSDiskConfiguration);`
`766`	`766`	`foreach (var disk in vmImageInfo.DataDiskConfigurations)`