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

make list databases return in their original order (#1859)

Browse Source 
* return databases in their original order

* remove using statement

* fix unit tests

* fix error
This commit is contained in:
Alan Ren
2023-02-09 14:27:25 -08:00
committed by GitHub
parent 497118ed82
commit 15250a9178
2 changed files with 12 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/Microsoft.SqlTools.ServiceLayer/Connection/ListDatabaseRequestHandler.cs
View File

@@ -9,7 +9,6 @@ using System.Collections.Generic;
using System.Data;
using System.Data.Common;
using System.Diagnostics;
using System.Linq;
using Microsoft.SqlServer.Management.Common;
using Microsoft.SqlTools.ServiceLayer.Admin.Contracts;
using Microsoft.SqlTools.ServiceLayer.Connection.Contracts;
@@ -61,8 +60,6 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection
/// </summary>
abstract class ListDatabaseRequestHandler<T> : IListDatabaseRequestHandler
{
private static readonly string[] SystemDatabases = new string[] { "master", "model", "msdb", "tempdb" };
public abstract string QueryText { get; }
public ListDatabasesResponse HandleRequest(ISqlConnectionFactory connectionFactory, ConnectionInfo connectionInfo)
@@ -122,9 +119,6 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection
{
results.Add(this.CreateItem(reader));
}
// Put system databases at the top of the list
results = results.Where(s => SystemDatabases.Any(x => this.NameMatches(x, s))).Concat(
results.Where(s => SystemDatabases.All(x => !this.NameMatches(x, s)))).ToList();
}
}
connection.Close();

24
test/Microsoft.SqlTools.ServiceLayer.UnitTests/Connection/ConnectionServiceTests.cs
View File

@@ -946,7 +946,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Connection
TestDbColumn[] cols = { new TestDbColumn("name") };
object[][] rows =
{
new object[] {"mydatabase"}, // this should be sorted to the end in the response
new object[] {"mydatabase"},
new object[] {"master"},
new object[] {"model"},
new object[] {"msdb"},
@@ -958,11 +958,11 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Connection
string[] databaseNames = response.DatabaseNames;
Assert.AreEqual(5, databaseNames.Length);
Assert.AreEqual("master", databaseNames[0]);
Assert.AreEqual("model", databaseNames[1]);
Assert.AreEqual("msdb", databaseNames[2]);
Assert.AreEqual("tempdb", databaseNames[3]);
Assert.AreEqual("mydatabase", databaseNames[4]);
Assert.AreEqual("mydatabase", databaseNames[0]);
Assert.AreEqual("master", databaseNames[1]);
Assert.AreEqual("model", databaseNames[2]);
Assert.AreEqual("msdb", databaseNames[3]);
Assert.AreEqual("tempdb", databaseNames[4]);
}
/// <summary>
@@ -980,7 +980,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Connection
};
object[][] rows =
{
new object[] {"mydatabase", "Online", "10", "2010-01-01 11:11:11"}, // this should be sorted to the end in the response
new object[] {"mydatabase", "Online", "10", "2010-01-01 11:11:11"},
new object[] {"master", "Online", "11", "2010-01-01 11:11:12"},
new object[] {"model", "Offline", "12", "2010-01-01 11:11:13"},
new object[] {"msdb", "Online", "13", "2010-01-01 11:11:14"},
@@ -990,11 +990,11 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Connection
var response = await RunListDatabasesRequestHandler(testdata: data, includeDetails: true);
Assert.AreEqual(5, response.Databases.Length);
VerifyDatabaseDetail(rows[0], response.Databases[4]);
VerifyDatabaseDetail(rows[1], response.Databases[0]);
VerifyDatabaseDetail(rows[2], response.Databases[1]);
VerifyDatabaseDetail(rows[3], response.Databases[2]);
VerifyDatabaseDetail(rows[4], response.Databases[3]);
VerifyDatabaseDetail(rows[0], response.Databases[0]);
VerifyDatabaseDetail(rows[1], response.Databases[1]);
VerifyDatabaseDetail(rows[2], response.Databases[2]);
VerifyDatabaseDetail(rows[3], response.Databases[3]);
VerifyDatabaseDetail(rows[4], response.Databases[4]);
}
private void VerifyDatabaseDetail(object[] expected, DatabaseInfo actual)