Add checkin test for export logfile cmdlet

Author: Aman Orazaev

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.Dataplane/Commands/Export-AzureAsInstanceLog.cs

@@ -151,13 +151,11 @@ public override void ExecuteCmdlet()
                 UriBuilder resolvedUriBuilder = new UriBuilder(logfileBaseUri);
                 resolvedUriBuilder.Host = ClusterResolve(logfileBaseUri, accessToken, serverName);
 
-                var logfileEndpoint =
-                    string.Format(
-                        (string)
-                            context.Environment.Endpoints[AsAzureEnvironment.AsRolloutEndpoints.LogfileEndpointFormat],
+                var logfileEndpoint = string.Format(
+                        (string) context.Environment.Endpoints[AsAzureEnvironment.AsRolloutEndpoints.LogfileEndpointFormat],
                         serverName);
 
-                this.AsAzureHttpClient = new AsAzureHttpClient(() => new HttpClient());
+                this.AsAzureHttpClient.resetHttpClient();
                 using (HttpResponseMessage message = AsAzureHttpClient.CallGetAsync(
                     resolvedUriBuilder.Uri,
                     logfileEndpoint,

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.Dataplane/Models/AsAzureHttpClient.cs

@@ -31,15 +31,20 @@ public interface IAsAzureHttpClient
         Task<HttpResponseMessage> CallPostAsync(Uri baseURI, string requestURL, string accessToken, HttpContent content = null);
 
         Task<HttpResponseMessage> CallGetAsync(Uri baseURI, string requestURL, string accessToken);
+
+        void resetHttpClient();
     }
 
     public class AsAzureHttpClient : IAsAzureHttpClient
     {
         public HttpClient HttpClient { get; set; }
 
+        private Func<HttpClient> HttpClientProvider { get; set; } 
+
         public AsAzureHttpClient(Func<HttpClient> httpClientProvider)
         {
-            HttpClient = httpClientProvider();
+            this.HttpClientProvider = httpClientProvider;
+            this.HttpClient = this.HttpClientProvider();
         }
 
         public async Task<HttpResponseMessage> CallGetAsync(Uri baseURI, string requestURL, string accessToken)
@@ -52,6 +57,11 @@ public async Task<HttpResponseMessage> CallPostAsync(Uri baseURI, string request
             return await CallAsync(HttpMethod.Post, baseURI, requestURL, accessToken, content);
         }
 
+        public void resetHttpClient()
+        {
+            this.HttpClient = this.HttpClientProvider();
+        }
+
         private async Task<HttpResponseMessage> CallAsync(HttpMethod method, Uri baseURI, string requestURL, string accessToken, HttpContent content = null)
         {
             using (HttpClient)

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.Test/InMemoryTests/DataPlaneCommandTests.cs

@@ -269,6 +269,79 @@ public void RestartAzureASInstance_NotLoggedInThrows()
             }
         }
 
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void ExportAzureASInstanceLogTest()
+        {
+            Mock<ICommandRuntime> commandRuntimeMock = new Mock<ICommandRuntime>();
+            // Setup
+            // Clear the the current profile
+            AsAzureClientSession.Instance.Profile.Environments.Clear();
+            var mockAuthenticationProvider = new Mock<IAsAzureAuthenticationProvider>();
+            mockAuthenticationProvider.Setup(
+                authProvider => authProvider.GetAadAuthenticatedToken(
+                    It.IsAny<AsAzureContext>(),
+                    It.IsAny<SecureString>(),
+                    It.IsAny<PromptBehavior>(),
+                    It.IsAny<string>(),
+                    It.IsAny<string>(),
+                    It.IsAny<Uri>())).Returns(testToken);
+            AsAzureClientSession.Instance.SetAsAzureAuthenticationProvider(mockAuthenticationProvider.Object);
+            commandRuntimeMock.Setup(f => f.ShouldProcess(It.IsAny<string>(), It.IsAny<string>())).Returns(true);
+
+            // Set up AsAzureHttpClient mock
+            var mockAsAzureHttpClient = new Mock<IAsAzureHttpClient>();
+            mockAsAzureHttpClient
+                .Setup(obj => obj.CallPostAsync(
+                    It.IsAny<Uri>(),
+                    It.Is<string>(s => s.Contains("clusterResolve")),
+                    It.IsAny<string>(),
+                    It.IsAny<HttpContent>()))
+                .Returns(Task<HttpResponseMessage>.FromResult(
+                    new HttpResponseMessage(HttpStatusCode.OK)
+                    {
+                        Content = new StringContent("{\"clusterFQDN\": \"resolved.westcentralus.asazure.windows.net\"}")
+                    }));
+            mockAsAzureHttpClient.Setup(obj => obj.CallGetAsync(It.IsAny<Uri>(), It.IsAny<string>(), It.IsAny<string>())).Returns(
+                Task<HttpResponseMessage>.FromResult(
+                    new HttpResponseMessage(HttpStatusCode.OK)
+                    {
+                        Content = new StringContent("MOCKED STREAM CONTENT")
+                    }));
+            
+            var mockTokenCacheItemProvider = new Mock<ITokenCacheItemProvider>();
+            mockTokenCacheItemProvider
+                .Setup(obj => obj.GetTokenFromTokenCache(It.IsAny<TokenCache>(), It.IsAny<string>()))
+                .Returns(testToken);
+
+            var exportLogCmdlet = new ExportAzureAnalysisServerLog(mockAsAzureHttpClient.Object, mockTokenCacheItemProvider.Object)
+            {
+                CommandRuntime = commandRuntimeMock.Object
+            };
+            
+            var addAmdlet = new AddAzureASAccountCommand()
+            {
+                CommandRuntime = commandRuntimeMock.Object
+            };
+
+            DoLogin(addAmdlet);
+            exportLogCmdlet.Instance = testServer;
+            try
+            {
+                exportLogCmdlet.OutputPath = System.IO.Path.GetTempFileName();
+                exportLogCmdlet.InvokeBeginProcessing();
+                exportLogCmdlet.ExecuteCmdlet();
+                exportLogCmdlet.InvokeEndProcessing();
+            }
+            finally
+            {
+                if (System.IO.File.Exists(exportLogCmdlet.OutputPath))
+                {
+                    System.IO.File.Delete(exportLogCmdlet.OutputPath);
+                }
+            }
+        }
+
         private void DoLogin(AddAzureASAccountCommand addAmdlet)
         {
             Mock<ICommandRuntime> commandRuntimeMock = new Mock<ICommandRuntime>();