ckaczor/sqltoolsservice
1
0
Fork 0
mirror of https://github.com/ckaczor/sqltoolsservice.git synced 2026-01-14 01:25:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

Kusto AdminService Unit Tests (#1257)

Browse Source 
* Added unit tests for AdminService. Added interface for IConnectionService.

* Kusto connection map to ConnectionManager (#1258)

* Created ConnectionManager and moved connectionMap. Changed references from ConnectionService to ConnectionManager.

* Changed creation of ConnectionManager to MEF instead of Instance creation
This commit is contained in:
Justin M
2021-10-07 11:46:13 -07:00
committed by GitHub
parent 2e08873246
commit f2da10f23a
15 changed files with 333 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

141
test/Microsoft.Kusto.ServiceLayer.UnitTests/Admin/AdminServiceTests.cs Normal file
View File

@@ -0,0 +1,141 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.Kusto.ServiceLayer.Admin;
using Microsoft.Kusto.ServiceLayer.Admin.Contracts;
using Microsoft.Kusto.ServiceLayer.Connection;
using Microsoft.Kusto.ServiceLayer.Connection.Contracts;
using Microsoft.Kusto.ServiceLayer.DataSource;
using Microsoft.SqlTools.Hosting.Protocol;
using Microsoft.SqlTools.ServiceLayer.Connection.ReliableConnection;
using Moq;
using NUnit.Framework;
namespace Microsoft.Kusto.ServiceLayer.UnitTests.Admin
{
public class AdminServiceTests
{
[TestCase(null)]
[TestCase("")]
public async Task HandleGetDatabaseInfoRequest_NoDatabase_Returns_Null(string databaseName)
{
var mockServiceHost = new Mock<IProtocolEndpoint>();
var mockDataSourceFactory = new Mock<IDataSourceConnectionFactory>();
var connectionInfo = new ConnectionInfo(mockDataSourceFactory.Object, null, new ConnectionDetails { DatabaseName = databaseName });
var mockConnectionManager = new Mock<IConnectionManager>();
mockConnectionManager
.Setup(x => x.TryGetValue(It.IsAny<string>(), out connectionInfo))
.Returns(true);
var mockRequestContext = new Mock<RequestContext<GetDatabaseInfoResponse>>();
var actualResponse = new GetDatabaseInfoResponse();
mockRequestContext.Setup(x => x.SendResult(It.IsAny<GetDatabaseInfoResponse>()))
.Callback<GetDatabaseInfoResponse>(actual => actualResponse = actual)
.Returns(Task.CompletedTask);
var adminService = new AdminService();
adminService.InitializeService(mockServiceHost.Object, mockConnectionManager.Object);
await adminService.HandleGetDatabaseInfoRequest(new GetDatabaseInfoParams(), mockRequestContext.Object);
Assert.AreEqual(null, actualResponse.DatabaseInfo);
}
[Test]
public async Task HandleGetDatabaseInfoRequest_Returns_DatabaseName()
{
var expectedDatabaseInfo = new DatabaseInfo()
{
Options = new Dictionary<string, object>
{
{ "FakeDatabaseName", "FakeSizeInMB" }
}
};
var mockDatasource = new Mock<IDataSource>();
mockDatasource
.Setup(x => x.GetDatabaseInfo(It.IsAny<string>(), It.IsAny<string>()))
.Returns(expectedDatabaseInfo);
var mockDataSourceFactory = new Mock<IDataSourceFactory>();
mockDataSourceFactory
.Setup(x => x.Create(It.IsAny<ConnectionDetails>(), It.IsAny<string>()))
.Returns(mockDatasource.Object);
var expectedConnectionDetails = new ConnectionDetails
{
DatabaseName = "FakeDatabaseName"
};
var mockConnectionFactory = new Mock<IDataSourceConnectionFactory>();
var connectionInfo = new ConnectionInfo(mockConnectionFactory.Object, null, expectedConnectionDetails);
var connection = new ReliableDataSourceConnection(expectedConnectionDetails, RetryPolicyFactory.NoRetryPolicy,
RetryPolicyFactory.NoRetryPolicy, mockDataSourceFactory.Object, "");
connectionInfo.AddConnection(ConnectionType.Default, connection);
var mockConnectionManager = new Mock<IConnectionManager>();
mockConnectionManager
.Setup(x => x.TryGetValue(It.IsAny<string>(), out connectionInfo))
.Returns(true);
var mockRequestContext = new Mock<RequestContext<GetDatabaseInfoResponse>>();
var actualResponse = new GetDatabaseInfoResponse();
mockRequestContext.Setup(x => x.SendResult(It.IsAny<GetDatabaseInfoResponse>()))
.Callback<GetDatabaseInfoResponse>(actual => actualResponse = actual)
.Returns(Task.CompletedTask);
var mockServiceHost = new Mock<IProtocolEndpoint>();
var adminService = new AdminService();
adminService.InitializeService(mockServiceHost.Object, mockConnectionManager.Object);
await adminService.HandleGetDatabaseInfoRequest(new GetDatabaseInfoParams(), mockRequestContext.Object);
Assert.AreEqual("FakeDatabaseName", actualResponse.DatabaseInfo.Options.First().Key);
Assert.AreEqual("FakeSizeInMB", actualResponse.DatabaseInfo.Options.First().Value);
}
[Test]
public async Task HandleGetDatabaseInfoRequest_NoConnection_Returns_Null()
{
var mockServiceHost = new Mock<IProtocolEndpoint>();
var mockConnectionManager = new Mock<IConnectionManager>();
var mockRequestContext = new Mock<RequestContext<GetDatabaseInfoResponse>>();
var actualResponse = new GetDatabaseInfoResponse();
mockRequestContext.Setup(x => x.SendResult(It.IsAny<GetDatabaseInfoResponse>()))
.Callback<GetDatabaseInfoResponse>(actual => actualResponse = actual)
.Returns(Task.CompletedTask);
var adminService = new AdminService();
adminService.InitializeService(mockServiceHost.Object, mockConnectionManager.Object);
await adminService.HandleGetDatabaseInfoRequest(new GetDatabaseInfoParams(), mockRequestContext.Object);
Assert.AreEqual(null, actualResponse.DatabaseInfo);
}
[Test]
public async Task HandleDatabaseInfoRequest_ThrowsException_Returns_Error()
{
var mockServiceHost = new Mock<IProtocolEndpoint>();
var mockConnectionManager = new Mock<IConnectionManager>();
ConnectionInfo connectionInfo;
var expectedException = new Exception("Fake Error Message");
var actualException = new Exception();
mockConnectionManager
.Setup(x => x.TryGetValue(It.IsAny<string>(), out connectionInfo))
.Throws(expectedException);
var mockRequestContext = new Mock<RequestContext<GetDatabaseInfoResponse>>();
mockRequestContext
.Setup(x => x.SendError(It.IsAny<Exception>()))
.Callback<Exception>(ex => actualException = ex)
.Returns(Task.CompletedTask);
var adminService = new AdminService();
adminService.InitializeService(mockServiceHost.Object, mockConnectionManager.Object);
await adminService.HandleGetDatabaseInfoRequest(new GetDatabaseInfoParams(), mockRequestContext.Object);
Assert.AreEqual(expectedException.GetType(), actualException.GetType());
Assert.AreEqual(expectedException.Message, actualException.Message);
}
}
}

27
test/Microsoft.Kusto.ServiceLayer.UnitTests/Admin/Contracts/AdminServiceTests.cs
View File

@@ -1,27 +0,0 @@
using Microsoft.Kusto.ServiceLayer.Admin;
using Microsoft.Kusto.ServiceLayer.Connection;
using Microsoft.Kusto.ServiceLayer.Connection.Contracts;
using Moq;
using NUnit.Framework;
namespace Microsoft.Kusto.ServiceLayer.UnitTests.Admin.Contracts
{
public class AdminServiceTests
{
[TestCase(null)]
[TestCase("")]
public void GetDatabaseInfo_Returns_Null_For_Invalid_DatabaseName(string databaseName)
{
var dataSourceConnectionFactory = new Mock<IDataSourceConnectionFactory>();
var connectionDetails = new ConnectionDetails
{
DatabaseName = databaseName
};
var connectionInfo = new ConnectionInfo(dataSourceConnectionFactory.Object, "", connectionDetails);
var adminService = new AdminService();
var databaseInfo = adminService.GetDatabaseInfo(connectionInfo);
Assert.IsNull(databaseInfo);
}
}
}