Updating tests for new environment class

Author: markcowl

src/ServiceManagement/Services/Commands.Test/Environment/AddAzureEnvironmentTests.cs

@@ -62,7 +62,7 @@ public void AddsAzureEnvironment()
             cmdlet.ExecuteCmdlet();
             cmdlet.InvokeEndProcessing();
 
-            commandRuntimeMock.Verify(f => f.WriteObject(It.IsAny<PSObject>()), Times.Once());
+            commandRuntimeMock.Verify(f => f.WriteObject(It.IsAny<PSAzureEnvironment>()), Times.Once());
             ProfileClient client = new ProfileClient(profile);
             AzureEnvironment env = client.GetEnvironmentOrDefault("KaTaL");
             Assert.Equal(env.Name, cmdlet.Name);
@@ -91,7 +91,7 @@ public void AddsEnvironmentWithMinimumInformation()
             cmdlet.ExecuteCmdlet();
             cmdlet.InvokeEndProcessing();
 
-            commandRuntimeMock.Verify(f => f.WriteObject(It.IsAny<PSObject>()), Times.Once());
+            commandRuntimeMock.Verify(f => f.WriteObject(It.IsAny<PSAzureEnvironment>()), Times.Once());
             ProfileClient client = new ProfileClient(profile);
             AzureEnvironment env = client.Profile.Environments["KaTaL"];
             Assert.Equal(env.Name, cmdlet.Name);
@@ -146,9 +146,9 @@ public void AddsEnvironmentWithStorageEndpoint()
         {
             var profile = new AzureProfile();
             Mock<ICommandRuntime> commandRuntimeMock = new Mock<ICommandRuntime>();
-            PSObject actual = null;
+            PSAzureEnvironment actual = null;
             commandRuntimeMock.Setup(f => f.WriteObject(It.IsAny<object>()))
-                .Callback((object output) => actual = (PSObject)output);
+                .Callback((object output) => actual = (PSAzureEnvironment)output);
             AddAzureEnvironmentCommand cmdlet = new AddAzureEnvironmentCommand()
             {
                 CommandRuntime = commandRuntimeMock.Object,
@@ -162,11 +162,11 @@ public void AddsEnvironmentWithStorageEndpoint()
             cmdlet.ExecuteCmdlet();
             cmdlet.InvokeEndProcessing();
 
-            commandRuntimeMock.Verify(f => f.WriteObject(It.IsAny<PSObject>()), Times.Once());
+            commandRuntimeMock.Verify(f => f.WriteObject(It.IsAny<PSAzureEnvironment>()), Times.Once());
             ProfileClient client = new ProfileClient(profile);
             AzureEnvironment env = client.Profile.Environments["KaTaL"];
             Assert.Equal(env.Name, cmdlet.Name);
-            Assert.Equal(env.Endpoints[AzureEnvironment.Endpoint.PublishSettingsFileUrl], actual.GetVariableValue<string>(AzureEnvironment.Endpoint.PublishSettingsFileUrl.ToString()));
+            Assert.Equal(env.Endpoints[AzureEnvironment.Endpoint.PublishSettingsFileUrl], actual.PublishSettingsFileUrl);
         }
 
         [Fact]