diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup.Test/Commands.AzureBackup.Test.csproj b/src/ResourceManager/AzureBackup/Commands.AzureBackup.Test/Commands.AzureBackup.Test.csproj index 3e5033a2462f..e6de3e29ac97 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup.Test/Commands.AzureBackup.Test.csproj +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup.Test/Commands.AzureBackup.Test.csproj @@ -40,7 +40,8 @@ True - ..\..\..\packages\Microsoft.Azure.Management.BackupServices.1.0.4-preview\lib\net40\Microsoft.Azure.Management.BackupServicesManagement.dll + False + ..\..\..\packages\Microsoft.Azure.Management.BackupServices.1.0.5-preview\lib\net40\Microsoft.Azure.Management.BackupServicesManagement.dll False diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup.Test/packages.config b/src/ResourceManager/AzureBackup/Commands.AzureBackup.Test/packages.config index 24eeae4af6aa..d0d223c6e304 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup.Test/packages.config +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup.Test/packages.config @@ -4,7 +4,7 @@ - + diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/EnableAzureRMBackupContainerReregistration.cs b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/EnableAzureRMBackupContainerReregistration.cs index 7918dc778980..24b170af1d89 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/EnableAzureRMBackupContainerReregistration.cs +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/EnableAzureRMBackupContainerReregistration.cs @@ -46,6 +46,8 @@ protected override void ProcessRecord() { case AzureBackupContainerType.Windows: case AzureBackupContainerType.SCDPM: + case AzureBackupContainerType.AzureBackupServer: + case AzureBackupContainerType.Other: AzureBackupClient.EnableMachineContainerReregistration(Container.ResourceGroupName, Container.ResourceName, Container.Id); break; default: diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/GetAzureRMBackupContainer.cs b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/GetAzureRMBackupContainer.cs index 35c2d8e30b15..1bb691991d18 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/GetAzureRMBackupContainer.cs +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/GetAzureRMBackupContainer.cs @@ -60,6 +60,8 @@ protected override void ProcessRecord() { case AzureBackupContainerType.Windows: case AzureBackupContainerType.SCDPM: + case AzureBackupContainerType.AzureBackupServer: + case AzureBackupContainerType.Other: containers.AddRange(GetMachineContainers(Vault.ResourceGroupName, Vault.Name)); break; case AzureBackupContainerType.AzureVM: diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/UnregisterAzureRMBackupContainer.cs b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/UnregisterAzureRMBackupContainer.cs index fd3bf7d509f4..e4a09401574f 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/UnregisterAzureRMBackupContainer.cs +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Cmdlets/Container/UnregisterAzureRMBackupContainer.cs @@ -47,6 +47,8 @@ protected override void ProcessRecord() { case AzureBackupContainerType.Windows: case AzureBackupContainerType.SCDPM: + case AzureBackupContainerType.AzureBackupServer: + case AzureBackupContainerType.Other: DeleteServer(); break; case AzureBackupContainerType.AzureVM: diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Commands.AzureBackup.csproj b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Commands.AzureBackup.csproj index c6a6460dfc96..aec29b762da4 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Commands.AzureBackup.csproj +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Commands.AzureBackup.csproj @@ -57,7 +57,8 @@ ..\..\..\packages\Microsoft.Azure.Common.2.1.0\lib\net45\Microsoft.Azure.Common.NetFramework.dll - ..\..\..\packages\Microsoft.Azure.Management.BackupServices.1.0.4-preview\lib\net40\Microsoft.Azure.Management.BackupServicesManagement.dll + False + ..\..\..\packages\Microsoft.Azure.Management.BackupServices.1.0.5-preview\lib\net40\Microsoft.Azure.Management.BackupServicesManagement.dll False diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Helpers/ContainerHelpers.cs b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Helpers/ContainerHelpers.cs index 67d011686ac0..b0fa09138368 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Helpers/ContainerHelpers.cs +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Helpers/ContainerHelpers.cs @@ -38,33 +38,37 @@ internal class ContainerHelpers { private static readonly Regex ResourceGroupRegex = new Regex(@"/subscriptions/(?.+)/resourceGroups/(?.+)/providers/(?.+)/BackupVault/(?.+)/containers/(?.+)", RegexOptions.Compiled); - internal static AzureBackupContainerType GetContainerType(string customerType) + internal static AzureBackupContainerType GetContainerType(string customerTypeString) { - CustomerType type = (CustomerType)Enum.Parse(typeof(CustomerType), customerType); - AzureBackupContainerType containerType = 0; + CustomerType customerType = CustomerType.Invalid; - switch (type) + if (Enum.TryParse(customerTypeString, out customerType)) { - case CustomerType.DPM: - containerType = AzureBackupContainerType.SCDPM; - break; - case CustomerType.InMage: - break; - case CustomerType.Invalid: - break; - case CustomerType.ManagedContainer: - break; - case CustomerType.OBS: - containerType = AzureBackupContainerType.Windows; - break; - case CustomerType.SBS: - containerType = AzureBackupContainerType.Windows; - break; - case CustomerType.SqlPaaS: - break; - default: - break; + switch (customerType) + { + case CustomerType.DPM: + containerType = AzureBackupContainerType.SCDPM; + break; + case CustomerType.OBS: + containerType = AzureBackupContainerType.Windows; + break; + case CustomerType.SBS: + containerType = AzureBackupContainerType.Windows; + break; + case CustomerType.DPMVenus: + containerType = AzureBackupContainerType.AzureBackupServer; + break; + case CustomerType.Invalid: + break; + default: + containerType = AzureBackupContainerType.Other; + break; + } + } + else if (!string.IsNullOrEmpty(customerTypeString)) + { + containerType = AzureBackupContainerType.Other; } return containerType; diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Models/AzureBackupEnums.cs b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Models/AzureBackupEnums.cs index a4c6156f6d8c..a5e33721c26c 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup/Models/AzureBackupEnums.cs +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup/Models/AzureBackupEnums.cs @@ -25,6 +25,8 @@ public enum AzureBackupContainerType Windows = 1, SCDPM, AzureVM, + AzureBackupServer, + Other, } public enum DataSourceType diff --git a/src/ResourceManager/AzureBackup/Commands.AzureBackup/packages.config b/src/ResourceManager/AzureBackup/Commands.AzureBackup/packages.config index ec875e2b5092..eec44ee711d4 100644 --- a/src/ResourceManager/AzureBackup/Commands.AzureBackup/packages.config +++ b/src/ResourceManager/AzureBackup/Commands.AzureBackup/packages.config @@ -4,7 +4,7 @@ - +