Added test for login with subscriptionName specified only.

Author: Hao Chen

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

@@ -104,6 +104,51 @@ public void SubscriptionIdNotInFirstTenant()
                 null);
         }
 
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void SubscriptionNameNotInFirstTenant()
+        {
+            var tenants = new List<string> { DefaultTenant.ToString(), Guid.NewGuid().ToString() };
+            var subscriptionInSecondTenant= Guid.NewGuid().ToString();
+            var firstList = new List<string> { DefaultSubscription.ToString() };
+            var secondList = new List<string> { Guid.NewGuid().ToString(), subscriptionInSecondTenant };
+            var client = SetupTestEnvironment(tenants, firstList, secondList);
+
+            ((MockTokenAuthenticationFactory)AzureSession.AuthenticationFactory).TokenProvider = (account, environment, tenant) =>
+            new MockAccessToken
+            {
+                UserId = "[email protected]",
+                LoginType = LoginType.OrgId,
+                AccessToken = "bbb",
+                TenantId = DefaultTenant.ToString()
+            };
+
+            var listAsyncResponses = new Queue<Func<SubscriptionListResult>>();
+            listAsyncResponses.Enqueue(() =>
+            {
+                var sub = new Subscription
+                {
+                    Id = DefaultSubscription.ToString(), 
+                    SubscriptionId = DefaultSubscription.ToString(),
+                    DisplayName = DefaultSubscriptionName, 
+                    State = "enabled",
+                };
+                return new SubscriptionListResult 
+                {
+                    Subscriptions = new List<Subscription> { sub }
+                };
+            });
+            MockSubscriptionClientFactory.SetListAsyncResponses(listAsyncResponses);
+
+            var azureRmProfile = client.Login(
+                Context.Account,
+                Context.Environment,
+                null,
+                null,
+                "sub name",
+                null);
+        }
+
         [Fact]
         [Trait(Category.AcceptanceType, Category.CheckIn)]
         public void TokenIdAndAccountIdMismatch()

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

@@ -30,6 +30,8 @@ public class MockSubscriptionClientFactory
         private Queue<List<string>> _subscriptions;
         private HashSet<string> _subscriptionSet;
         private static Queue<Func<GetSubscriptionResult>> _getAsyncQueue;
+        private static Queue<Func<SubscriptionListResult>> _listAsyncQueue;
+
         public MockSubscriptionClientFactory(List<string> tenants, Queue<List<string>> subscriptions)
         {
             _tenants = tenants;
@@ -54,6 +56,10 @@ public static void SetGetAsyncResponses(Queue<Func<GetSubscriptionResult>> respo
         {
             _getAsyncQueue = responses;
         }
+        public static void SetListAsyncResponses(Queue<Func<SubscriptionListResult>> responses)
+        {
+            _listAsyncQueue = responses;
+        }
 
         public SubscriptionClient GetSubscriptionClient()
         {
@@ -101,6 +107,11 @@ public SubscriptionClient GetSubscriptionClient()
                 (s) => s.ListAsync(It.IsAny<CancellationToken>())).Returns(
                     (CancellationToken token) =>
                     {
+                        if (_listAsyncQueue != null && _listAsyncQueue.Any())
+                        {
+                            return Task.FromResult(_listAsyncQueue.Dequeue().Invoke());
+                        }
+                        
                         SubscriptionListResult result = null;
                         if (_subscriptions.Count > 0)
                         {