Add live only test to check for context list population

Author: cormacpayne

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

@@ -31,6 +31,7 @@
 using Microsoft.Azure.Commands.Common.Authentication.Abstractions;
 using Microsoft.Azure.Commands.ResourceManager.Common;
 using Microsoft.Azure.Commands.ScenarioTest;
+using System.Linq;
 
 namespace Microsoft.Azure.Commands.Profile.Test
 {
@@ -327,7 +328,7 @@ public void LoginWithCredentialParameterAndMSA()
                              "For more information, please refer to http://go.microsoft.com/fwlink/?linkid=331007&clcid=0x409 " +
                              "for more information about the difference between an organizational account and a Microsoft account.",
                              ex.Message);
-            }            
+            }
         }
 
         [Fact]
@@ -353,6 +354,43 @@ public void LoginWithAccessToken()
             Assert.NotNull(AzureRmProfileProvider.Instance.Profile.DefaultContext);
         }
 
+        [Fact]
+        [Trait(Category.RunType, Category.LiveOnly)]
+        public void LoginPopulatesContextList()
+        {
+            // Before running this test, make sure to clear the contexts on your machine by removing the following two files:
+            // - %APPDATA%/Windows Azure Powershell/AzureRmContext.json
+            // - %APPDATA%/Windows Azure Powershell/AzureRmContextSettings.json
+            // This will clear all existing contexts on your machine so that this test can re-populate the list with a context for each subscription
+
+            var cmdlt = new ConnectAzureRmAccountCommand();
+            // Setup
+            cmdlt.CommandRuntime = commandRuntimeMock;
+
+            // Act
+            cmdlt.InvokeBeginProcessing();
+            cmdlt.ExecuteCmdlet();
+            cmdlt.InvokeEndProcessing();
+
+            var profile = AzureRmProfileProvider.Instance.Profile as AzureRmProfile;
+            Assert.NotNull(profile);
+            Assert.NotNull(profile.Contexts);
+            Assert.NotNull(profile.Subscriptions);
+            Assert.True(profile.Contexts.Count > 1);
+            Assert.True(profile.Subscriptions.Count() > 1);
+            Assert.Equal(profile.Subscriptions.Count(), profile.Contexts.Count);
+
+            foreach (var sub in profile.Subscriptions)
+            {
+                var contextName = string.Format("{0} - {1}", sub.Name, sub.Id);
+                Assert.True(profile.Contexts.ContainsKey(contextName));
+                var context = profile.Contexts[contextName];
+                Assert.NotNull(context);
+                Assert.Equal(sub.Id, context.Subscription.Id);
+                Assert.Equal(sub.GetTenant(), context.Tenant.Id);
+            }
+        }
+
         [Fact]
         [Trait(Category.AcceptanceType, Category.CheckIn)]
         public void ThrowOnUnknownEnvironment()