mirror of
https://github.com/ckaczor/sqltoolsservice.git
synced 2026-01-14 09:59:48 -05:00
366 lines
14 KiB
C#
366 lines
14 KiB
C#
//
|
|
// 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;
|
|
using System.Collections.Generic;
|
|
using System.Data.Common;
|
|
using System.Linq;
|
|
using System.Text.RegularExpressions;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.SqlTools.ServiceLayer.EditData;
|
|
using Microsoft.SqlTools.ServiceLayer.EditData.Contracts;
|
|
using Microsoft.SqlTools.ServiceLayer.EditData.UpdateManagement;
|
|
using Microsoft.SqlTools.ServiceLayer.QueryExecution;
|
|
using Microsoft.SqlTools.ServiceLayer.QueryExecution.Contracts;
|
|
using Microsoft.SqlTools.ServiceLayer.Test.Common;
|
|
using Microsoft.SqlTools.ServiceLayer.UnitTests.Utility;
|
|
using NUnit.Framework;
|
|
|
|
namespace Microsoft.SqlTools.ServiceLayer.UnitTests.EditData
|
|
{
|
|
public class RowEditBaseTests
|
|
{
|
|
[Test]
|
|
public void ConstructWithoutExtendedMetadata()
|
|
{
|
|
// Setup: Create a table metadata that has not been extended
|
|
EditTableMetadata etm = new EditTableMetadata();
|
|
|
|
// If: I construct a new EditRowBase implementation without an extended metadata
|
|
// Then: I should get an exception
|
|
Assert.Throws<ArgumentException>(() => new RowEditTester(null, etm));
|
|
}
|
|
|
|
[Test]
|
|
public async Task ValidateUpdatableColumnOutOfRange([Values(-1,2,100)]int columnId)
|
|
{
|
|
// Setup: Create a result set
|
|
var rs = await GetResultSet(
|
|
new DbColumn[] {
|
|
new TestDbColumn("id") {IsKey = true, IsAutoIncrement = true, IsIdentity = true},
|
|
new TestDbColumn("col1")
|
|
},
|
|
new object[] { "id", "1" });
|
|
var etm = Common.GetCustomEditTableMetadata(rs.Columns.Cast<DbColumn>().ToArray());
|
|
|
|
// If: I validate a column ID that is out of range
|
|
// Then: It should throw
|
|
RowEditTester tester = new RowEditTester(rs, etm);
|
|
Assert.Throws<ArgumentOutOfRangeException>(() => tester.ValidateColumn(columnId));
|
|
}
|
|
|
|
[Test]
|
|
public async Task ValidateUpdatableColumnNotUpdatable()
|
|
{
|
|
// Setup: Create a result set with an identity column
|
|
var rs = await GetResultSet(
|
|
new DbColumn[] {
|
|
new TestDbColumn("id") {IsKey = true, IsAutoIncrement = true, IsIdentity = true},
|
|
new TestDbColumn("col1")
|
|
},
|
|
new object[] { "id", "1" });
|
|
var etm = Common.GetCustomEditTableMetadata(rs.Columns.Cast<DbColumn>().ToArray());
|
|
|
|
// If: I validate a column ID that is not updatable
|
|
// Then: It should throw
|
|
RowEditTester tester = new RowEditTester(rs, etm);
|
|
Assert.Throws<InvalidOperationException>(() => tester.ValidateColumn(0));
|
|
}
|
|
|
|
[Test]
|
|
[TestCaseSource(nameof(GetWhereClauseIsNotNullData))]
|
|
public async Task GetWhereClauseSimple(DbColumn col, object val, string nullClause)
|
|
{
|
|
// Setup: Create a result set and metadata provider with a single column
|
|
var cols = new[] {col};
|
|
ResultSet rs = await GetResultSet(cols, new[] {val});
|
|
EditTableMetadata etm = Common.GetCustomEditTableMetadata(cols);
|
|
|
|
RowEditTester rt = new RowEditTester(rs, etm);
|
|
if (val == DBNull.Value)
|
|
{
|
|
rt.ValidateWhereClauseNullKey(nullClause);
|
|
}
|
|
else
|
|
{
|
|
rt.ValidateWhereClauseSingleKey(nullClause);
|
|
}
|
|
}
|
|
|
|
public static IEnumerable<object[]> GetWhereClauseIsNotNullData
|
|
{
|
|
get
|
|
{
|
|
yield return new object[] {new TestDbColumn("col"), DBNull.Value, "IS NULL"};
|
|
yield return new object[] {
|
|
new TestDbColumn
|
|
{
|
|
DataTypeName = "BINARY",
|
|
DataType = typeof(byte[])
|
|
},
|
|
new byte[5],
|
|
"= 0x0000000000"
|
|
};
|
|
yield return new object[]
|
|
{
|
|
new TestDbColumn
|
|
{
|
|
DataType = typeof(string),
|
|
DataTypeName = "TEXT"
|
|
},
|
|
"abc",
|
|
"= N'abc'"
|
|
};
|
|
yield return new object[]
|
|
{
|
|
new TestDbColumn
|
|
{
|
|
DataType = typeof(string),
|
|
DataTypeName = "NTEXT",
|
|
|
|
},
|
|
"abc",
|
|
"= N'abc'"
|
|
};
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public async Task GetWhereClauseMultipleKeyColumns()
|
|
{
|
|
// Setup: Create a result set and metadata provider with multiple key columns
|
|
DbColumn[] cols = {new TestDbColumn("col1"), new TestDbColumn("col2")};
|
|
ResultSet rs = await GetResultSet(cols, new object[] {"abc", "def"});
|
|
EditTableMetadata etm = Common.GetCustomEditTableMetadata(cols);
|
|
|
|
RowEditTester rt = new RowEditTester(rs, etm);
|
|
rt.ValidateWhereClauseMultipleKeys();
|
|
}
|
|
|
|
[Test]
|
|
public async Task GetWhereClauseNoKeyColumns()
|
|
{
|
|
// Setup: Create a result set and metadata provider with no key columns
|
|
DbColumn[] cols = {new TestDbColumn("col1"), new TestDbColumn("col2")};
|
|
ResultSet rs = await GetResultSet(cols, new object[] {"abc", "def"});
|
|
EditTableMetadata etm = Common.GetCustomEditTableMetadata(new DbColumn[] {});
|
|
|
|
RowEditTester rt = new RowEditTester(rs, etm);
|
|
rt.ValidateWhereClauseNoKeys();
|
|
}
|
|
|
|
[Test]
|
|
public async Task SortingByTypeTest()
|
|
{
|
|
// Setup: Create a result set and metadata we can reuse
|
|
var data = new Common.TestDbColumnsWithTableMetadata(false, false, 0, 0);
|
|
var rs = await Common.GetResultSet(data.DbColumns, false);
|
|
|
|
// If: I request to sort a list of the three different edit operations
|
|
List<RowEditBase> rowEdits = new List<RowEditBase>
|
|
{
|
|
new RowDelete(0, rs, data.TableMetadata),
|
|
new RowUpdate(0, rs, data.TableMetadata),
|
|
new RowCreate(0, rs, data.TableMetadata)
|
|
};
|
|
rowEdits.Sort();
|
|
|
|
// Then:
|
|
// (we don't care about the order of the other two)
|
|
Assert.That(rowEdits.Last(), Is.InstanceOf<RowDelete>(), "Delete should be the last operation to execute");
|
|
}
|
|
|
|
[Test]
|
|
public async Task SortingUpdatesByRowIdTest()
|
|
{
|
|
// Setup: Create a result set and metadata we can reuse
|
|
var data = new Common.TestDbColumnsWithTableMetadata(false, false, 0, 0);
|
|
var rs = await Common.GetResultSet(data.DbColumns, false, 4);
|
|
|
|
// If: I sort 3 edit operations of the same type
|
|
List<RowEditBase> rowEdits = new List<RowEditBase>
|
|
{
|
|
new RowUpdate(3, rs, data.TableMetadata),
|
|
new RowUpdate(1, rs, data.TableMetadata),
|
|
new RowUpdate(2, rs, data.TableMetadata)
|
|
};
|
|
rowEdits.Sort();
|
|
|
|
// Then: They should be in order by row ID ASCENDING
|
|
Assert.AreEqual(1, rowEdits[0].RowId);
|
|
Assert.AreEqual(2, rowEdits[1].RowId);
|
|
Assert.AreEqual(3, rowEdits[2].RowId);
|
|
}
|
|
|
|
[Test]
|
|
public async Task SortingCreatesByRowIdTest()
|
|
{
|
|
// Setup: Create a result set and metadata we can reuse
|
|
var data = new Common.TestDbColumnsWithTableMetadata(false, false, 0, 0);
|
|
var rs = await Common.GetResultSet(data.DbColumns, false);
|
|
|
|
// If: I sort 3 edit operations of the same type
|
|
List<RowEditBase> rowEdits = new List<RowEditBase>
|
|
{
|
|
new RowCreate(3, rs, data.TableMetadata),
|
|
new RowCreate(1, rs, data.TableMetadata),
|
|
new RowCreate(2, rs, data.TableMetadata)
|
|
};
|
|
rowEdits.Sort();
|
|
|
|
// Then: They should be in order by row ID ASCENDING
|
|
Assert.AreEqual(1, rowEdits[0].RowId);
|
|
Assert.AreEqual(2, rowEdits[1].RowId);
|
|
Assert.AreEqual(3, rowEdits[2].RowId);
|
|
}
|
|
|
|
[Test]
|
|
public async Task SortingDeletesByRowIdTest()
|
|
{
|
|
// Setup: Create a result set and metadata we can reuse
|
|
var data = new Common.TestDbColumnsWithTableMetadata(false, false, 0, 0);
|
|
var rs = await Common.GetResultSet(data.DbColumns, false);
|
|
|
|
// If: I sort 3 delete operations of the same type
|
|
List<RowEditBase> rowEdits = new List<RowEditBase>
|
|
{
|
|
new RowDelete(1, rs, data.TableMetadata),
|
|
new RowDelete(3, rs, data.TableMetadata),
|
|
new RowDelete(2, rs, data.TableMetadata)
|
|
};
|
|
rowEdits.Sort();
|
|
|
|
// Then: They should be in order by row ID DESCENDING
|
|
Assert.AreEqual(3, rowEdits[0].RowId);
|
|
Assert.AreEqual(2, rowEdits[1].RowId);
|
|
Assert.AreEqual(1, rowEdits[2].RowId);
|
|
}
|
|
|
|
private static async Task<ResultSet> GetResultSet(DbColumn[] columns, object[] row)
|
|
{
|
|
object[][] rows = {row};
|
|
var testResultSet = new TestResultSet(columns, rows);
|
|
var testReader = new TestDbDataReader(new [] {testResultSet}, false);
|
|
var resultSet = new ResultSet(0,0, MemoryFileSystem.GetFileStreamFactory());
|
|
await resultSet.ReadResultToEnd(testReader, CancellationToken.None);
|
|
return resultSet;
|
|
}
|
|
|
|
private sealed class RowEditTester : RowEditBase
|
|
{
|
|
public RowEditTester(ResultSet rs, EditTableMetadata meta) : base(0, rs, meta) { }
|
|
|
|
public void ValidateColumn(int columnId)
|
|
{
|
|
ValidateColumnIsUpdatable(columnId);
|
|
}
|
|
|
|
// ReSharper disable once UnusedParameter.Local
|
|
public void ValidateWhereClauseNullKey(string nullValue)
|
|
{
|
|
// If: I generate a where clause with one is null column value
|
|
WhereClause wc = GetWhereClause(false);
|
|
|
|
// Then:
|
|
// ... There should only be one component
|
|
Assert.AreEqual(1, wc.ClauseComponents.Count);
|
|
|
|
Assert.That(wc.Parameters, Is.Empty, "Parameterization should be empty");
|
|
|
|
// ... The component should contain the name of the column and be null
|
|
Assert.AreEqual(
|
|
$"({AssociatedObjectMetadata.Columns.First().EscapedName} {nullValue})",
|
|
wc.ClauseComponents[0]);
|
|
|
|
// ... The complete clause should contain a single WHERE
|
|
Assert.AreEqual($"WHERE {wc.ClauseComponents[0]}", wc.CommandText);
|
|
}
|
|
|
|
public void ValidateWhereClauseSingleKey(string clauseValue)
|
|
{
|
|
// If: I generate a where clause with one is null column value
|
|
WhereClause wc = GetWhereClause(false);
|
|
|
|
// Then:
|
|
// ... There should only be one component
|
|
Assert.AreEqual(1, wc.ClauseComponents.Count);
|
|
|
|
// ... Parameterization should be empty
|
|
Assert.IsEmpty(wc.Parameters);
|
|
|
|
// ... The component should contain the name of the column and the value
|
|
Assert.True(wc.ClauseComponents[0].Contains(AssociatedObjectMetadata.Columns.First().EscapedName));
|
|
Regex r = new Regex($@"\(CONVERT \([A-Z]*\(MAX\), {AssociatedObjectMetadata.Columns.First().EscapedName}\) {clauseValue}\)");
|
|
Assert.True(r.IsMatch(wc.ClauseComponents[0]));
|
|
|
|
// ... The complete clause should contain a single WHERE
|
|
Assert.AreEqual($"WHERE {wc.ClauseComponents[0]}", wc.CommandText);
|
|
}
|
|
|
|
public void ValidateWhereClauseMultipleKeys()
|
|
{
|
|
// If: I generate a where clause with multiple key columns
|
|
WhereClause wc = GetWhereClause(false);
|
|
|
|
// Then:
|
|
// ... There should two components
|
|
var keys = AssociatedObjectMetadata.KeyColumns.ToArray();
|
|
Assert.AreEqual(keys.Length, wc.ClauseComponents.Count);
|
|
|
|
Assert.That(wc.Parameters, Is.Empty, "Parameterization should be empty");
|
|
|
|
Assert.That(wc.ClauseComponents, Has.All.Match(@"\([0-9a-z]+ = .+\)"), "The components should contain the name of the column and the value");
|
|
|
|
// ... The complete clause should contain multiple cause components joined
|
|
// with and
|
|
Assert.True(wc.CommandText.StartsWith("WHERE "));
|
|
Assert.True(wc.CommandText.EndsWith(string.Join(" AND ", wc.ClauseComponents)));
|
|
}
|
|
|
|
public void ValidateWhereClauseNoKeys()
|
|
{
|
|
// If: I generate a where clause from metadata that doesn't have keys
|
|
// Then: An exception should be thrown
|
|
Assert.Throws<InvalidOperationException>(() => GetWhereClause(false));
|
|
}
|
|
|
|
public override string GetScript()
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
|
|
public override EditRow GetEditRow(DbCellValue[] cells)
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
|
|
public override EditRevertCellResult RevertCell(int columnId)
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
|
|
public override EditUpdateCellResult SetCell(int columnId, string newValue)
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
|
|
public override void ApplyChanges(DbDataReader reader)
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
|
|
public override DbCommand GetCommand(DbConnection conn)
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
|
|
protected override int SortId => 0;
|
|
}
|
|
}
|
|
}
|