sqltoolsservice/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/Security/UserTests.cs

//
// 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.ServiceLayer.IntegrationTests.Utility;
using Microsoft.SqlTools.ServiceLayer.Security;
using Microsoft.SqlTools.ServiceLayer.Test.Common;

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] - enable tests in separate change
        public async Task TestHandleCreateUserWithLoginRequest()
        {
            using (SelfCleaningTempFile queryTempFile = new SelfCleaningTempFile())
            {
                // setup
                SecurityService service = new SecurityService();
                UserServiceHandlerImpl userService = new UserServiceHandlerImpl();
                var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", queryTempFile.FilePath);
                var contextId = System.Guid.NewGuid().ToString();

                var login = await SecurityTestUtils.CreateLogin(service, connectionResult, contextId);

                var user = await SecurityTestUtils.CreateUser(userService, connectionResult, contextId, login);

                await SecurityTestUtils.DeleteUser(userService, connectionResult, user);

                await SecurityTestUtils.DeleteLogin(service, connectionResult, login);                
            }
        }

        /// <summary>
        /// Test the basic Update User method handler
        /// </summary>
        //[Test] - enable tests in separate change
        public async Task TestHandleUpdateUserWithLoginRequest()
        {
            using (SelfCleaningTempFile queryTempFile = new SelfCleaningTempFile())
            {
                // setup
                SecurityService service = new SecurityService();
                UserServiceHandlerImpl userService = new UserServiceHandlerImpl();
                var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", queryTempFile.FilePath);
                var contextId = System.Guid.NewGuid().ToString();

                var login = await SecurityTestUtils.CreateLogin(service, connectionResult, contextId);

                var user = await SecurityTestUtils.CreateUser(userService, connectionResult, contextId, login);

                await SecurityTestUtils.UpdateUser(userService, connectionResult, contextId, user);

                await SecurityTestUtils.DeleteUser(userService, connectionResult, user);

                await SecurityTestUtils.DeleteLogin(service, connectionResult, login);
            }
        }
    }
}