XEvent Profiler initial event handlers (#456)

* Bump SMO to 140.2.5 to pick-up private XEvent binaries * Pick up SMO binaries from the build lab * Add ProfilerService class placeholder * Update SMO nuget package to include DB Scoped XEvents * Stage changes * Stage changes * Update SMO to use RTM dependencies and remove separate SqlScript package * Stage changes * Iterate on profiler service * Fix post-merge break in localization * More refactoring * Continue iterating on profiler * Add test profiler listener * Address a couple of the code review feedback * Fix AppVeyor build break * Use self-cleaning test file
2026-01-14 01:25:40 -05:00 · 2017-09-12 14:08:50 -07:00
parent 2677efb6b8
commit 84ea045572
42 changed files with 1448 additions and 115 deletions
--- a/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/Profiler/ProfilerServiceTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/Profiler/ProfilerServiceTests.cs
@@ -0,0 +1,90 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using System;
+using System.Collections.Generic;
+using System.Data.SqlClient;
+using System.IO;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.SqlServer.Management.Smo;
+using Microsoft.SqlServer.Management.XEvent;
+using Microsoft.SqlTools.Hosting.Protocol;
+using Microsoft.SqlTools.ServiceLayer.Connection;
+using Microsoft.SqlTools.ServiceLayer.IntegrationTests.Utility;
+using Microsoft.SqlTools.ServiceLayer.Profiler;
+using Microsoft.SqlTools.ServiceLayer.Profiler.Contracts;
+using Microsoft.SqlTools.ServiceLayer.Test.Common;
+using Moq;
+using Xunit;
+
+namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Profiler
+{
+    public class ProfilerServiceTests
+    {
+        /// <summary>
+        /// Verify that a start profiling request starts a profiling session
+        /// </summary>
+        [Fact]
+        public async Task TestHandleStartAndStopProfilingRequests()
+        {
+            using (SelfCleaningTempFile queryTempFile = new SelfCleaningTempFile())
+            {
+                var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", queryTempFile.FilePath);
+
+                ProfilerService profilerService = new ProfilerService();
+
+                // start a new session
+                var startParams = new StartProfilingParams();
+                startParams.OwnerUri = connectionResult.ConnectionInfo.OwnerUri;
+                startParams.TemplateName = "Standard";
+
+                string sessionId = null;
+                var startContext = new Mock<RequestContext<StartProfilingResult>>();
+                startContext.Setup(rc => rc.SendResult(It.IsAny<StartProfilingResult>()))
+                    .Returns<StartProfilingResult>((result) => 
+                    {
+                        // capture the session id for sending the stop message
+                        sessionId = result.SessionId;
+                        return Task.FromResult(0);
+                    });
+
+                await profilerService.HandleStartProfilingRequest(startParams, startContext.Object);
+
+                startContext.VerifyAll();
+
+                // wait a bit for the session monitoring to initialize
+                Thread.Sleep(TimeSpan.FromHours(1));
+
+                // stop the session
+                var stopParams = new StopProfilingParams()
+                {
+                    SessionId = sessionId
+                };
+
+                var stopContext = new Mock<RequestContext<StopProfilingResult>>();
+                stopContext.Setup(rc => rc.SendResult(It.IsAny<StopProfilingResult>()))
+                    .Returns(Task.FromResult(0));
+
+                await profilerService.HandleStopProfilingRequest(stopParams, stopContext.Object);
+
+                stopContext.VerifyAll();
+            }           
+        }
+
+        /// <summary>
+        /// Verify  the profiler service XEvent session factory
+        /// </summary>
+        [Fact]
+        public void TestCreateXEventSession()
+        {
+            var liveConnection = LiveConnectionHelper.InitLiveConnectionInfo("master");
+            ProfilerService profilerService = new ProfilerService();
+            IXEventSession xeSession = profilerService.CreateXEventSession(liveConnection.ConnectionInfo);
+            Assert.NotNull(xeSession);
+            Assert.NotNull(xeSession.GetTargetXml());
+        }
+    }
+}