User management support classes (#1856)

* WIP * Fix nullable warnings in UserData class * WIP2 * WIP * Refresh database prototype classes * Fix some typos & merge issues * WIP * WIP * WIP * Additional updates * Remove unneded using
2026-01-14 01:25:40 -05:00 · 2023-02-08 18:02:08 -08:00
parent ee086e2067
commit 2ef5f0918a
24 changed files with 3584 additions and 1336 deletions
--- a/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/Security/UserTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/Security/UserTests.cs
@@ -0,0 +1,69 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using System.Threading.Tasks;
+using Microsoft.SqlTools.Hosting.Protocol;
+using Microsoft.SqlTools.ServiceLayer.IntegrationTests.Utility;
+using Microsoft.SqlTools.ServiceLayer.Security;
+using Microsoft.SqlTools.ServiceLayer.Security.Contracts;
+using Microsoft.SqlTools.ServiceLayer.Test.Common;
+using Moq;
+
+namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Security
+{
+    /// <summary>
+    /// Tests for the User management component
+    /// </summary>
+    public class UserTests
+    {
+        /// <summary>
+        /// Test the basic Create User method handler
+        /// </summary>
+        // [Test]
+        public async Task TestHandleCreateUserRequest()
+        {
+            using (SelfCleaningTempFile queryTempFile = new SelfCleaningTempFile())
+            {
+                // setup
+                var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", queryTempFile.FilePath);
+                var loginParams = new CreateLoginParams
+                {
+                    OwnerUri = connectionResult.ConnectionInfo.OwnerUri,
+                    Login = SecurityTestUtils.GetTestLoginInfo()
+                };
+
+                var createLoginContext = new Mock<RequestContext<CreateLoginResult>>();
+                createLoginContext.Setup(x => x.SendResult(It.IsAny<CreateLoginResult>()))
+                    .Returns(Task.FromResult(new object()));
+
+                // call the create login method
+                SecurityService service = new SecurityService();
+                await service.HandleCreateLoginRequest(loginParams, createLoginContext.Object);
+
+                // verify the result
+                createLoginContext.Verify(x => x.SendResult(It.Is<CreateLoginResult>
+                    (p => p.Success && p.Login.LoginName != string.Empty)));
+
+            
+                var userParams = new CreateUserParams
+                {
+                    OwnerUri = connectionResult.ConnectionInfo.OwnerUri,
+                    User = SecurityTestUtils.GetTestUserInfo(loginParams.Login.LoginName)
+                };
+
+                var createUserContext = new Mock<RequestContext<CreateUserResult>>();
+                createUserContext.Setup(x => x.SendResult(It.IsAny<CreateUserResult>()))
+                    .Returns(Task.FromResult(new object()));
+
+                // call the create login method
+                await service.HandleCreateUserRequest(userParams, createUserContext.Object);
+
+                // verify the result
+                createUserContext.Verify(x => x.SendResult(It.Is<CreateUserResult>
+                    (p => p.Success && p.User.UserName != string.Empty)));
+            }
+        }
+    }
+}