//
// 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)));
}
}
}
}