Merge pull request #177 from m-moris/fix175

ogail · ogail · commit 94189d95c2ff · 2015-03-01T14:20:54.000-08:00
Fix #175 Issue
diff --git a/src/ServiceManagement/Compute/Commands.ServiceManagement/IaaS/DiskRepository/SetAzureVMImageDataDiskConfig.cs b/src/ServiceManagement/Compute/Commands.ServiceManagement/IaaS/DiskRepository/SetAzureVMImageDataDiskConfig.cs
@@ -99,7 +99,9 @@ protected override void ProcessRecord()
             }
 
             var diskConfig = DiskConfig.DataDiskConfigurations.FirstOrDefault(
-                d => string.Equals(d.Name, this.DataDiskName, StringComparison.OrdinalIgnoreCase));
+                d => this.ParameterSetName == UpdateAzureVMImageParamSet
+                    ? string.Equals(d.Name, this.DataDiskName, StringComparison.OrdinalIgnoreCase)
+                    : d.MediaLink == this.MediaLink);
 
             if (diskConfig == null)
             {

Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,9 @@ protected override void ProcessRecord()`
`99`	`99`	`}`
`100`	`100`
`101`	`101`	`var diskConfig = DiskConfig.DataDiskConfigurations.FirstOrDefault(`
`102`		`- d => string.Equals(d.Name, this.DataDiskName, StringComparison.OrdinalIgnoreCase));`
	`102`	`+ d => this.ParameterSetName == UpdateAzureVMImageParamSet`
	`103`	`+ ? string.Equals(d.Name, this.DataDiskName, StringComparison.OrdinalIgnoreCase)`
	`104`	`+ : d.MediaLink == this.MediaLink);`
`103`	`105`
`104`	`106`	`if (diskConfig == null)`
`105`	`107`	`{`