Fixing bug

Author: athipp

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.ServiceManagement/Commands.AnalysisServices.ServiceManagement.csproj

@@ -90,11 +90,11 @@
   <ItemGroup>
     <Compile Include="Commands\AddAzureASAccount.cs" />
     <Compile Include="Models\AsAzureAccount.cs" />
-    <Compile Include="Models\AsAzureClientUtility.cs" />
+    <Compile Include="Models\AsAzureClientSession.cs" />
     <Compile Include="Models\ASAzureContext.cs" />
     <Compile Include="Models\AsAzureEnvironment.cs" />
     <Compile Include="Models\AsAzureProfile.cs" />
-    <Compile Include="Commands\Restart-AzureAnalysisServer.cs" />
+    <Compile Include="Commands\Restart-AzureAsInstance.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
     <Compile Include="Properties\Resources.Designer.cs">
       <DependentUpon>Resources.resx</DependentUpon>

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.ServiceManagement/Commands/AddAzureASAccount.cs

@@ -26,8 +26,8 @@ namespace Microsoft.Azure.Commands.AnalysisServices.ServiceManagement
     /// <summary>
     /// Cmdlet to log into an Analysis Services environment
     /// </summary>
-    [Cmdlet("Add", "AzureASAccount", SupportsShouldProcess=true)]
-    [Alias("Login-AzureASAccount")]
+    [Cmdlet("Add", "AzureAsAccount", SupportsShouldProcess=true)]
+    [Alias("Login-AzureAsAccount")]
     [OutputType(typeof(AsAzureProfile))]
     public class AddAzureASAccountCommand : AzurePSCmdlet, IModuleAssemblyInitializer
     {

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.ServiceManagement/Commands/Restart-AzureAnalysisServer.cs renamed to src/ResourceManager/AnalysisServices/Commands.AnalysisServices.ServiceManagement/Commands/Restart-AzureAsInstance.cs

@@ -27,7 +27,7 @@ namespace Microsoft.Azure.Commands.AnalysisServices.ServiceManagement
     /// <summary>
     /// Cmdlet to log into an Analysis Services environment
     /// </summary>
-    [Cmdlet("Restart", "AzureAnalysisServer", SupportsShouldProcess=true)]
+    [Cmdlet("Restart", "AzureAsInstance", SupportsShouldProcess=true)]
     [Alias("Restart-AzureAs")]
     [OutputType(typeof(AsAzureProfile))]
     public class RestartAzureAnalysisServer: AzurePSCmdlet, IModuleAssemblyInitializer

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.ServiceManagement/Models/AsAzureClientUtility.cs renamed to src/ResourceManager/AnalysisServices/Commands.AnalysisServices.ServiceManagement/Models/AsAzureClientSession.cs

@@ -120,6 +120,11 @@ public static string GetAadAuthenticatedToken(AsAzureContext asAzureContext, Sec
             return result.AccessToken;
         }
 
+        public static string GetAuthorityUrlForEnvironment(AsAzureEnvironment environment)
+        {
+            return environment.Name.Contains("asazure-int") ? AadAuthorityUrlPpe : AadAuthorityUrlProd;
+        }
+
         public void SetCurrentContext(AsAzureAccount azureAccount, AsAzureEnvironment asEnvironment)
         {
             _profile.Context = new AsAzureContext(azureAccount, asEnvironment)

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.ServiceManagement/Models/AsAzureProfile.cs

@@ -54,7 +54,7 @@ public override string ToString()
         public AsAzureEnvironment CreateEnvironment(string environmentName)
         {
             var env = new AsAzureEnvironment(environmentName);
-            env.Endpoints.Add(AsAzureEnvironment.AsRolloutEndpoints.AdAuthorityBaseUrl, AsAzureClientSession.AadAuthorityUrlProd);
+            env.Endpoints.Add(AsAzureEnvironment.AsRolloutEndpoints.AdAuthorityBaseUrl, AsAzureClientSession.GetAuthorityUrlForEnvironment(env));
             env.Endpoints.Add(AsAzureEnvironment.AsRolloutEndpoints.RestartEndpointFormat, AsAzureClientSession.RestartEndpointPathFormat);
 
             return env;

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.Test/ScenarioTests/AsTests.ps1

@@ -187,8 +187,8 @@ function Test-AnalysisServicesServerRestart
 		$serverCreated = New-AzureRmAnalysisServicesServer -ResourceGroupName $resourceGroupName -Name $serverName -Location $location -Sku 'S1' -Administrators '[email protected],[email protected]'
 		Assert-True {$serverCreated.ProvisioningState -like "Succeeded"}
 
-		Login-AzureASAccount -EnvironmentName $environment
-		Restart-AzureAnalysisServer -Instance $serverName
+		Login-AzureAsAccount -EnvironmentName $environment
+		Restart-AzureAsInstance -Instance $serverName
 	}
 	finally
 	{