Updated Mock for subscriptionClient and added tests for RMProfile.

Author: Hao Chen

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

@@ -27,6 +27,8 @@
 using Microsoft.Azure.Commands.Common.Authentication.Models;
 using Microsoft.Azure.Commands.Profile;
 using Microsoft.Azure.Commands.Profile.Models;
+using Microsoft.Azure.Subscriptions.Models;
+using Hyak.Common;
 
 namespace Microsoft.Azure.Commands.ResourceManager.Common.Test
 {
@@ -67,6 +69,41 @@ private static RMProfileClient SetupTestEnvironment(List<string> tenants, params
             return new RMProfileClient(profile);
         }
 
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void SubscriptionIdNotInFirstTenant()
+        {
+            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 getAsyncResponses = new Queue<Func<GetSubscriptionResult>>();
+            getAsyncResponses.Enqueue(() =>
+            {
+                throw new CloudException("InvalidAuthenticationTokenTenant: The access token is from the wrong issuer");
+            });
+            MockSubscriptionClientFactory.SetGetAsyncResponses(getAsyncResponses);
+            
+            var azureRmProfile = client.Login(
+                Context.Account,
+                Context.Environment,
+                null,
+                subscriptionInSecondTenant,
+                null,
+                null);
+        }
+
         [Fact]
         [Trait(Category.AcceptanceType, Category.CheckIn)]
         public void TokenIdAndAccountIdMismatch()
@@ -78,7 +115,6 @@ public void TokenIdAndAccountIdMismatch()
             var thirdList = new List<string> { DefaultSubscription.ToString(), secondsubscriptionInTheFirstTenant };
             var fourthList = new List<string> { DefaultSubscription.ToString(), secondsubscriptionInTheFirstTenant };
             var client = SetupTestEnvironment(tenants, firstList, secondList, thirdList, fourthList);
-
             var tokens = new Queue<MockAccessToken>();
             tokens.Enqueue(new MockAccessToken
             {

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

@@ -29,6 +29,7 @@ public class MockSubscriptionClientFactory
         private IList<string> _tenants;
         private Queue<List<string>> _subscriptions;
         private HashSet<string> _subscriptionSet;
+        private static Queue<Func<GetSubscriptionResult>> _getAsyncQueue;
         public MockSubscriptionClientFactory(List<string> tenants, Queue<List<string>> subscriptions)
         {
             _tenants = tenants;
@@ -49,6 +50,11 @@ public static string GetSubscriptionNameFromId(string id)
             return "Sub-" + id;
         }
 
+        public static void SetGetAsyncResponses(Queue<Func<GetSubscriptionResult>> responses)
+        {
+            _getAsyncQueue = responses;
+        }
+
         public SubscriptionClient GetSubscriptionClient()
         {
             var tenantMock = new Mock<ITenantOperations>();
@@ -66,6 +72,10 @@ public SubscriptionClient GetSubscriptionClient()
                 s => s.GetAsync(It.IsAny<string>(), It.IsAny<CancellationToken>())).Returns(
                     (string subId, CancellationToken token) =>
                     {
+                        if (_getAsyncQueue != null && _getAsyncQueue.Any())
+                        {
+                            return Task.FromResult(_getAsyncQueue.Dequeue().Invoke());
+                        }
                         GetSubscriptionResult result = new GetSubscriptionResult
                         {
                             RequestId = Guid.NewGuid().ToString(),