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

//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//

#nullable disable

using System.Threading.Tasks;
using Microsoft.SqlTools.ServiceLayer.IntegrationTests.Utility;
using Microsoft.SqlTools.ServiceLayer.Security;
using Microsoft.SqlTools.ServiceLayer.Test.Common;
using NUnit.Framework;

namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Security
{
    /// <summary>
    /// Tests for the Credential management component
    /// </summary>
    public class CredentialTests
    {
        /// <summary>
        /// TestHandleCreateCredentialRequest
        /// </summary>
        [Test]
        public async Task TestHandleCreateCredentialRequest()
        {
            using (SelfCleaningTempFile queryTempFile = new SelfCleaningTempFile())
            {
                // setup
                var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", queryTempFile.FilePath);
                var service = new SecurityService();
                var credential = SecurityTestUtils.GetTestCredentialInfo();
                await SecurityTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, SecurityTestUtils.GetCredentialURN(credential.Name));

                // test
                await SecurityTestUtils.CreateCredential(service, connectionResult, credential);

                // cleanup
                await SecurityTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, SecurityTestUtils.GetCredentialURN(credential.Name));
            }
        }

        /// <summary>
        /// TestHandleUpdateCredentialRequest
        /// </summary>
        [Test]
        public async Task TestHandleUpdateCredentialRequest()
        {
            using (SelfCleaningTempFile queryTempFile = new SelfCleaningTempFile())
            {
                // setup
                var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", queryTempFile.FilePath);
                var service = new SecurityService();
                var credential = SecurityTestUtils.GetTestCredentialInfo();
                await SecurityTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, SecurityTestUtils.GetCredentialURN(credential.Name));
                await SecurityTestUtils.CreateCredential(service, connectionResult, credential);

                // test
                await SecurityTestUtils.UpdateCredential(service, connectionResult, credential);

                // cleanup
                await SecurityTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, SecurityTestUtils.GetCredentialURN(credential.Name));
            }
        }

        /// <summary>
        /// TestHandleDeleteCredentialRequest
        /// </summary>
        [Test]
        public async Task TestHandleDeleteCredentialRequest()
        {
            using (SelfCleaningTempFile queryTempFile = new SelfCleaningTempFile())
            {
                // setup
                var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", queryTempFile.FilePath);
                var service = new SecurityService();
                var credential = SecurityTestUtils.GetTestCredentialInfo();
                await SecurityTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, SecurityTestUtils.GetCredentialURN(credential.Name));
                await SecurityTestUtils.CreateCredential(service, connectionResult, credential);

                // test
                await SecurityTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, SecurityTestUtils.GetCredentialURN(credential.Name));
            }
        }
    }
}