Add acr environment extention to fix get/add/set azenvironment (#231)

* add container registry to azure environment extention

* fix

Author: VeryEarly

src/Authentication.Abstractions/Extensions/AzureEnvironmentExtensions.cs

@@ -151,6 +151,9 @@ public static bool TryGetEndpointString(this IAzureEnvironment environment, stri
                     case AzureEnvironment.Endpoint.BatchEndpointResourceId:
                         propertyValue = environment.BatchEndpointResourceId;
                         break;
+                    case AzureEnvironment.Endpoint.ContainerRegistryEndpointSuffix:
+                        propertyValue = environment.ContainerRegistryEndpointSuffix;
+                        break;
                     default:
                         // get property from the extended properties of the environment
                         propertyValue = environment.GetProperty(endpointName);
@@ -265,6 +268,9 @@ public static void SetEndpoint(this IAzureEnvironment environment, string endpoi
                     case AzureEnvironment.Endpoint.ServiceManagement:
                         environment.ServiceManagementUrl = propertyValue;
                         break;
+                    case AzureEnvironment.Endpoint.ContainerRegistryEndpointSuffix:
+                        environment.ContainerRegistryEndpointSuffix = propertyValue;
+                        break;
                     case AzureEnvironment.ExtendedEndpoint.OperationalInsightsEndpointResourceId:
                         environment.SetProperty(AzureEnvironment.ExtendedEndpoint.OperationalInsightsEndpointResourceId, propertyValue);
                         break;
@@ -342,6 +348,9 @@ public static string GetTokenAudience(this IAzureEnvironment environment, string
                 case AzureEnvironment.ExtendedEndpoint.ManagedHsmServiceEndpointResourceId:
                     resource = AzureEnvironment.ExtendedEndpoint.ManagedHsmServiceEndpointResourceId;
                     break;
+                case AzureEnvironment.Endpoint.ContainerRegistryEndpointSuffix:
+                    resource = AzureEnvironment.Endpoint.ContainerRegistryEndpointSuffix;
+                    break;
                 default:
                     resource = AzureEnvironment.Endpoint.ActiveDirectoryServiceEndpointResourceId;
                     break;
@@ -524,6 +533,10 @@ public static void CopyFrom(this IAzureEnvironment environment, IAzureEnvironmen
                 {
                     environment.BatchEndpointResourceId = other.BatchEndpointResourceId;
                 }
+                if (other.IsEndpointSet(AzureEnvironment.Endpoint.ContainerRegistryEndpointSuffix))
+                {
+                    environment.ContainerRegistryEndpointSuffix = other.ContainerRegistryEndpointSuffix;
+                }
 
                 environment.VersionProfiles.Clear();
                 foreach (var profile in other.VersionProfiles)
@@ -615,6 +628,10 @@ public static void Update(this IAzureEnvironment environment, IAzureEnvironment
                 {
                     environment.DataLakeEndpointResourceId = other.DataLakeEndpointResourceId;
                 }
+                if (other.IsEndpointSet(AzureEnvironment.Endpoint.ContainerRegistryEndpointSuffix))
+                {
+                    environment.ContainerRegistryEndpointSuffix = other.ContainerRegistryEndpointSuffix;
+                }
 
                 foreach (var profile in other.VersionProfiles)
                 {