Merge pull request #2931 from krkhan/disable_enc_fix

UpdateVmEncryptionSettings() needs to be called for VolumeType=Data on Linux VMs

Author: cormacpayne

src/ResourceManager/Compute/Commands.Compute/Extension/AzureDiskEncryption/DisableAzureDiskEncryption.cs

@@ -257,17 +257,27 @@ public override void ExecuteCmdlet()
                                                             this.Name,
                                                             parameters).GetAwaiter().GetResult();
 
-                    if (string.IsNullOrWhiteSpace(VolumeType) ||
-                        VolumeType.Equals(AzureDiskEncryptionExtensionContext.VolumeTypeAll, StringComparison.InvariantCultureIgnoreCase) ||
-                        VolumeType.Equals(AzureDiskEncryptionExtensionContext.VolumeTypeOS, StringComparison.InvariantCultureIgnoreCase))
+                    // +---------+---------------+----------------------------+
+                    // | OSType  |  VolumeType   | UpdateVmEncryptionSettings |
+                    // +---------+---------------+----------------------------+
+                    // | Windows | OS            | Yes                        |
+                    // | Windows | Data          | No                         |
+                    // | Windows | Not Specified | Yes                        |
+                    // | Linux   | OS            | N/A                        |
+                    // | Linux   | Data          | Yes                        |
+                    // | Linux   | Not Specified | N/A                        |
+                    // +---------+---------------+----------------------------+
+
+                    if (OperatingSystemTypes.Windows.Equals(currentOSType) &&
+                        VolumeType.Equals(AzureDiskEncryptionExtensionContext.VolumeTypeData, StringComparison.InvariantCultureIgnoreCase))
                     {
-                        var opVm = UpdateVmEncryptionSettings();
-                        var result = Mapper.Map<PSAzureOperationResponse>(opVm);
+                        var result = Mapper.Map<PSAzureOperationResponse>(opExt);
                         WriteObject(result);
                     }
                     else
                     {
-                        var result = Mapper.Map<PSAzureOperationResponse>(opExt);
+                        var opVm = UpdateVmEncryptionSettings();
+                        var result = Mapper.Map<PSAzureOperationResponse>(opVm);
                         WriteObject(result);
                     }
                 }

src/ResourceManager/Compute/Commands.Compute/Models/AzureDiskEncryptionStatusContext.cs

@@ -22,6 +22,7 @@ enum EncryptionStatus
         Encrypted,
         NotEncrypted,
         NotMounted,
+        DecryptionInProgress,
         EncryptionInProgress,
         VMRestartPending,
         Unknown