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.UnitTests/Profiler/ProfilerServiceTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.UnitTests/Profiler/ProfilerServiceTests.cs
@@ -0,0 +1,68 @@
+//
+// 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.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.SqlServer.Management.XEvent;
+using Microsoft.SqlTools.Hosting.Protocol;
+using Microsoft.SqlTools.ServiceLayer.Connection;
+using Microsoft.SqlTools.ServiceLayer.Profiler;
+using Microsoft.SqlTools.ServiceLayer.Profiler.Contracts;
+using Microsoft.SqlTools.ServiceLayer.UnitTests.Utility;
+using Moq;
+using Xunit;
+
+namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Profiler
+{
+    /// <summary>
+    /// Unit tests for ProfilerService
+    /// </summary>
+    public class ProfilerServiceTests
+    {   
+        /// <summary>
+        /// Test starting a profiling session and receiving event callback
+        /// </summary>
+        /// <returns></returns>
+        [Fact]
+        public async Task TestStartProfilingRequest()
+        {
+            string sessionId = null;
+            string testUri = "profiler_uri";
+            var requestContext = new Mock<RequestContext<StartProfilingResult>>();
+            requestContext.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);
+                });
+
+            var sessionListener = new TestSessionListener();
+
+            var profilerService = new ProfilerService();
+            profilerService.SessionMonitor.AddSessionListener(sessionListener);
+            profilerService.ConnectionServiceInstance = TestObjects.GetTestConnectionService();
+            ConnectionInfo connectionInfo = TestObjects.GetTestConnectionInfo();
+            profilerService.ConnectionServiceInstance.OwnerToConnectionMap.Add(testUri, connectionInfo);
+            profilerService.XEventSessionFactory = new TestXEventSessionFactory();
+
+            var requestParams = new StartProfilingParams();
+            requestParams.OwnerUri = testUri;
+            requestParams.TemplateName = "Standard";
+
+            await profilerService.HandleStartProfilingRequest(requestParams, requestContext.Object);
+
+            // wait a bit for profile sessions to be polled
+            Thread.Sleep(500);
+
+            requestContext.VerifyAll();
+
+            Assert.Equal(sessionListener.PreviousSessionId, sessionId);
+            Assert.Equal(sessionListener.PreviousEvents.Count, 1);
+        }
+    }
+}