// // 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 { /// /// Tests for the User management component /// public class UserTests { /// /// Test the basic Create User method handler /// // [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>(); createLoginContext.Setup(x => x.SendResult(It.IsAny())) .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 (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>(); createUserContext.Setup(x => x.SendResult(It.IsAny())) .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 (p => p.Success && p.User.UserName != string.Empty))); } } } }