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

Feat/result streaming - Fix for issue #746 in toolsservice and issue Microsoft/azuredatastudio#3348 (#753)

Browse Source 
* fix for issues 746 & azuredatastudio issue 3348

* test coverage improvement for results streaming

* addressed minor review comments

* adding generated file test/CodeCoverage/package-lock.json to workaround code coverage issue.
This commit is contained in:
Arvind Ranasaria
2018-12-04 20:49:05 -08:00
committed by GitHub
parent a1946edca3
commit e4808c12aa
6 changed files with 3996 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
test/Microsoft.SqlTools.ServiceLayer.UnitTests/QueryExecution/Common.cs
View File

@@ -23,6 +23,7 @@ using Microsoft.SqlTools.ServiceLayer.Test.Common;
using Moq;
using Moq.Protected;
using HostingProtocol = Microsoft.SqlTools.Hosting.Protocol;
using System.Collections.Generic;
namespace Microsoft.SqlTools.ServiceLayer.UnitTests.QueryExecution
{
@@ -40,8 +41,20 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.QueryExecution
public const int StandardRows = 5;
public const int ZeroRows = 0;
public const int HundredRows = 100;
public const int ThousandRows = 1000;
public const int HundredKRows = 100000;
public const int MillionRows = 1000000;
public const int TenMillionRows = 10000000;
public const int HundredMillionRows = 100000000;
public const SelectionData WholeDocument = null;
public static readonly ConnectionDetails StandardConnectionDetails = new ConnectionDetails
@@ -55,18 +68,53 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.QueryExecution
public static readonly SelectionData SubsectionDocument = new SelectionData(0, 0, 2, 2);
#endregion
public static TestResultSet ZeroRowTestResultSet => new TestResultSet(StandardColumns, ZeroRows);
public static TestResultSet[] ZeroRowTestDataSet => new[] { ZeroRowTestResultSet };
public static TestResultSet StandardTestResultSet => new TestResultSet(StandardColumns, StandardRows);
private static readonly Lazy<TestResultSet> TestResultSet = new Lazy<TestResultSet>(new TestResultSet(StandardColumns, MillionRows));
public static TestResultSet[] StandardTestDataSet => new[] { StandardTestResultSet };
public static TestResultSet MillionRowTestResultSet => TestResultSet.Value;
private static readonly Lazy<TestResultSet[]> MillionRowResultSet = new Lazy<TestResultSet[]>(new[]{MillionRowTestResultSet});
public static TestResultSet HundredRowTestResultSet => new TestResultSet(StandardColumns, HundredRows);
public static TestResultSet[] StandardTestDataSet => new [] {StandardTestResultSet};
public static TestResultSet[] HundredRowTestDataSet => new[] { HundredRowTestResultSet };
public static TestResultSet[] MillionRowTestDataSet => MillionRowResultSet.Value;
public static TestResultSet ThousandRowTestResultSet => new TestResultSet(StandardColumns, ThousandRows);
public static TestResultSet[] ThousandRowTestDataSet => new[] { ThousandRowTestResultSet };
private static readonly Lazy<TestResultSet> HundredKRowTestResultSetLazy = new Lazy<TestResultSet>(new TestResultSet(StandardColumns, HundredKRows));
public static TestResultSet HundredKRowTestResultSet => HundredKRowTestResultSetLazy.Value;
private static readonly Lazy<TestResultSet[]> HundredKRowTestDataSetLazy = new Lazy<TestResultSet[]>(new[] { HundredKRowTestResultSet });
public static TestResultSet[] HundredKRowTestDataSet => HundredKRowTestDataSetLazy.Value;
private static readonly Lazy<TestResultSet> MillionRowTestResultSetLazy = new Lazy<TestResultSet>(new TestResultSet(StandardColumns, MillionRows));
public static TestResultSet MillionRowTestResultSet => MillionRowTestResultSetLazy.Value;
private static readonly Lazy<TestResultSet[]> MillionRowTestDataSetLazy = new Lazy<TestResultSet[]>(new[]{ MillionRowTestResultSet });
public static TestResultSet[] MillionRowTestDataSet => MillionRowTestDataSetLazy.Value;
public static IEnumerable<TestResultSet> TestResultSetsEnumeration
{
get
{
yield return StandardTestResultSet;
yield return HundredRowTestResultSet;
yield return ThousandRowTestResultSet;
yield return ZeroRowTestResultSet;
yield return HundredKRowTestResultSet;
yield return MillionRowTestResultSet;
}
}
public static TestResultSet[] ExecutionPlanTestDataSet
{
@@ -230,7 +278,8 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.QueryExecution
bool throwOnExecute,
bool throwOnRead,
WorkspaceService<SqlToolsSettings> workspaceService,
out ConcurrentDictionary<string, byte[]> storage)
out ConcurrentDictionary<string, byte[]> storage,
int sizeFactor=1)
{
// Create a place for the temp "files" to be written
storage = new ConcurrentDictionary<string, byte[]>();
@@ -244,7 +293,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.QueryExecution
.OutCallback((string owner, out ConnectionInfo connInfo) => connInfo = isConnected ? ci : null)
.Returns(isConnected);
return new QueryExecutionService(connectionService.Object, workspaceService) { BufferFileStreamFactory = MemoryFileSystem.GetFileStreamFactory(storage) };
return new QueryExecutionService(connectionService.Object, workspaceService) { BufferFileStreamFactory = MemoryFileSystem.GetFileStreamFactory(storage, sizeFactor) };
}
public static QueryExecutionService GetPrimedExecutionService(
@@ -252,10 +301,11 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.QueryExecution
bool isConnected,
bool throwOnExecute,
bool throwOnRead,
WorkspaceService<SqlToolsSettings> workspaceService)
WorkspaceService<SqlToolsSettings> workspaceService,
int sizeFactor=1)
{
ConcurrentDictionary<string, byte[]> storage;
return GetPrimedExecutionService(data, isConnected, throwOnExecute, throwOnRead, workspaceService, out storage);
return GetPrimedExecutionService(data, isConnected, throwOnExecute, throwOnRead, workspaceService, out storage, sizeFactor);
}
public static WorkspaceService<SqlToolsSettings> GetPrimedWorkspaceService(string query)