mirror of
https://github.com/ckaczor/sqltoolsservice.git
synced 2026-01-15 17:23:32 -05:00
Adding TestDbResultSet for Better Mocking (#231)
* Replacing the awful and inflexible Dictionary<string, string>[][] with a much better test data set class * Applying all changes * Removing unused RequestParamTests
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
//
|
||||
// Copyright (c) Microsoft. All rights reserved.
|
||||
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
||||
//
|
||||
|
||||
using System.Collections;
|
||||
using System.Collections.Generic;
|
||||
using System.Data.Common;
|
||||
using System.Linq;
|
||||
|
||||
namespace Microsoft.SqlTools.ServiceLayer.Test.Utility
|
||||
{
|
||||
public class TestResultSet : IEnumerable<object[]>
|
||||
{
|
||||
public List<DbColumn> Columns;
|
||||
public List<object[]> Rows;
|
||||
|
||||
public TestResultSet(int columns, int rows)
|
||||
{
|
||||
Columns = Enumerable.Range(0, columns).Select(i => new TestDbColumn($"Col{i}")).Cast<DbColumn>().ToList();
|
||||
Rows = new List<object[]>(rows);
|
||||
for (int i = 0; i < rows; i++)
|
||||
{
|
||||
var row = Enumerable.Range(0, columns).Select(j => $"Cell{i}.{j}").Cast<object>().ToArray();
|
||||
Rows.Add(row);
|
||||
}
|
||||
}
|
||||
|
||||
public TestResultSet(IEnumerable<DbColumn> columns, IEnumerable<object[]> rows)
|
||||
{
|
||||
Columns = new List<DbColumn>(columns);
|
||||
Rows = new List<object[]>(rows);
|
||||
}
|
||||
|
||||
#region IEnumerable<object[]> Impementation
|
||||
|
||||
public IEnumerator<object[]> GetEnumerator()
|
||||
{
|
||||
return (IEnumerator<object[]>) Rows.GetEnumerator();
|
||||
}
|
||||
|
||||
IEnumerator IEnumerable.GetEnumerator()
|
||||
{
|
||||
return GetEnumerator();
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user