Fix for #2387, #2388 fix subscription and tenant validate sets when no subscriptions or tenants selected

Author: markcowl

src/ResourceManager/Profile/Commands.Profile.Test/ContextCmdletTests.cs

@@ -12,6 +12,11 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using System;
+using System.CodeDom;
+using System.Collections.Generic;
+using System.Linq;
+using System.Linq.Expressions;
 using Microsoft.Azure.Commands.Common.Authentication;
 using Microsoft.Azure.Commands.Common.Authentication.Models;
 using Microsoft.Azure.Commands.Profile;
@@ -25,14 +30,18 @@
 using System.Management.Automation;
 using Xunit;
 using Xunit.Abstractions;
+using Xunit.Sdk;
 
 namespace Microsoft.Azure.Commands.ResourceManager.Profile.Test
 {
     public class ContextCmdletTests : RMTestBase
     {
         private MemoryDataStore dataStore;
         private MockCommandRuntime commandRuntimeMock;
-
+        const string guid1 = "a0cc8bd7-2c6a-47e9-a4c4-3f6ed136e240";
+        const string guid2 = "eab635c0-a35a-4f70-9e46-e5351c7b5c8b";
+        const string guid3 = "52f66548-2550-417b-941e-9d6e04f3ac8d";
+        const string guid4 = "40e67ee2-1a1a-4517-9253-ab6f93c5710f";
         public ContextCmdletTests(ITestOutputHelper output)
         {
             XunitTracingInterceptor.AddToContext(new XunitTracingInterceptor(output));
@@ -78,8 +87,15 @@ public void SelectAzureContextWithNoSubscriptionAndTenant()
             var existingTenants = account.GetProperty(AzureAccount.Property.Tenants);
             var allowedTenants = existingTenants == null ? tenantToSet : existingTenants + "," + tenantToSet;
             account.SetProperty(AzureAccount.Property.Tenants, allowedTenants);
+            account.SetProperty(AzureAccount.Property.Subscriptions, new string[0]);
 
-            ((RuntimeDefinedParameterDictionary)cmdlt.GetDynamicParameters())["TenantId"].Value = tenantToSet;
+            var paramDictionary =
+                ((RuntimeDefinedParameterDictionary)cmdlt.GetDynamicParameters());
+            var tenantParam = paramDictionary["TenantId"];
+            Assert.True(tenantParam.Attributes.Any(a => a is ValidateSetAttribute 
+              && ((ValidateSetAttribute)a).ValidValues.Any(v => string.Equals(v, tenantToSet, StringComparison.OrdinalIgnoreCase))));
+            Assert.False(paramDictionary["SubscriptionId"].Attributes.Any(a => a is ValidateSetAttribute));
+            tenantParam.Value = tenantToSet;
 
             // Act
             cmdlt.InvokeBeginProcessing();
@@ -94,6 +110,50 @@ public void SelectAzureContextWithNoSubscriptionAndTenant()
             Assert.NotEqual(tenantToSet, context.Tenant.TenantId);
         }
 
+        [Theory]
+        [InlineData(null, null)]
+        [InlineData(new string[0], new string[0])]
+        [InlineData(new string[] { guid1}, new string[] {guid2})]
+        [InlineData(new string[] { guid1, guid2 }, new string[] { guid3, guid4 })]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void SetsDynamicParametersForContext(string[] subscriptions, string[] tenants)
+        {
+            var cmdlt = new SetAzureRMContextCommand();
+
+            // Setup
+            cmdlt.CommandRuntime = commandRuntimeMock;
+
+            // Make sure that the tenant ID we are attempting to set is
+            // valid for the account
+            var account = AzureRmProfileProvider.Instance.Profile.Context.Account;
+            account.SetProperty(AzureAccount.Property.Tenants, tenants);
+            account.SetProperty(AzureAccount.Property.Subscriptions, subscriptions);
+
+            var paramDictionary =
+                ((RuntimeDefinedParameterDictionary)cmdlt.GetDynamicParameters());
+            var subscriptionParams = paramDictionary["SubscriptionId"];
+            VerifyValidationAttribute(subscriptionParams, subscriptions);
+            var tenantParams = paramDictionary["TenantId"];
+            VerifyValidationAttribute(tenantParams, tenants);
+        }
+
+        private void VerifyValidationAttribute(RuntimeDefinedParameter parameter, string[] expectedValues)
+        {
+            if (expectedValues != null && expectedValues.Length > 0)
+            {
+                var validateAttribute = parameter.Attributes.First(a => a is ValidateSetAttribute) as ValidateSetAttribute;
+                Assert.NotNull(validateAttribute);
+                foreach (var expectedValue in expectedValues)
+                {
+                    Assert.Contains(expectedValue, validateAttribute.ValidValues, StringComparer.OrdinalIgnoreCase);
+                }
+            }
+            else
+            {
+                Assert.False(parameter.Attributes.Any(a => a is ValidateSetAttribute));
+            }
+        }
+
         [Fact]
         [Trait(Category.AcceptanceType, Category.CheckIn)]
         public void SelectAzureContextWithNoSubscriptionAndNoTenant()
@@ -113,5 +173,7 @@ public void SelectAzureContextWithNoSubscriptionAndNoTenant()
             var context = (PSAzureContext)commandRuntimeMock.OutputPipeline[0];
             Assert.NotNull(context);
         }
+
+
     }
 }

src/ResourceManager/Profile/Commands.Profile/Context/SetAzureRMContext.cs

@@ -119,7 +119,6 @@ private RuntimeDefinedParameterDictionary CreateDynamicParameterDictionary()
                     HelpMessage = "Subscription",
                     ValueFromPipelineByPropertyName = true
                 },
-                new ValidateSetAttribute(AzureRmProfileProvider.Instance.Profile.Context.Account.GetPropertyAsArray(AzureAccount.Property.Subscriptions)),
             };
 
             var tenantIdAttributes = new Collection<Attribute>
@@ -139,9 +138,34 @@ private RuntimeDefinedParameterDictionary CreateDynamicParameterDictionary()
                     ValueFromPipelineByPropertyName = true
                 },
                 new AliasAttribute("Domain"),
-                new ValidateSetAttribute(AzureRmProfileProvider.Instance.Profile.Context.Account.GetPropertyAsArray(AzureAccount.Property.Tenants)),
             };
 
+            if (AzureRmProfileProvider.Instance != null
+                && AzureRmProfileProvider.Instance.Profile != null
+                && AzureRmProfileProvider.Instance.Profile.Context != null
+                && AzureRmProfileProvider.Instance.Profile.Context.Account != null)
+            {
+                var account = AzureRmProfileProvider.Instance.Profile.Context.Account;
+                if (account.IsPropertySet(AzureAccount.Property.Subscriptions))
+                {
+                    var subscriptions = account.GetPropertyAsArray(AzureAccount.Property.Subscriptions);
+                    if (subscriptions != null && subscriptions.Length > 0)
+                    {
+                        subscriptionIdAttributes.Add(
+                            new ValidateSetAttribute(subscriptions));
+                    }
+                }
+                if (account.IsPropertySet(AzureAccount.Property.Tenants))
+                {
+                    var tenants = account.GetPropertyAsArray(AzureAccount.Property.Tenants);
+                    if (tenants != null && tenants.Length > 0)
+                    {
+                        tenantIdAttributes.Add(
+                            new ValidateSetAttribute(tenants));
+                    }
+                }
+            }
+
             _tenantId = new RuntimeDefinedParameter("TenantId", typeof(string), tenantIdAttributes);
             _subscriptionId = new RuntimeDefinedParameter("SubscriptionId", typeof(string), subscriptionIdAttributes);