Fix batch results with GO N (#587)

- Fixes https://github.com/Microsoft/vscode-mssql/issues/1066
2026-01-14 01:25:40 -05:00 · 2018-03-21 13:41:58 -07:00
parent 2c7de499d1
commit 30c4b97f2a
2 changed files with 58 additions and 51 deletions
--- a/test/Microsoft.SqlTools.ServiceLayer.UnitTests/QueryExecution/Execution/BatchTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.UnitTests/QueryExecution/Execution/BatchTests.cs
@@ -373,6 +373,10 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.QueryExecution.Execution

            // Make sure the number of result sets matches
            Assert.Equal(expectedResultSets, batch.ResultSets.Count);
+            for (int i = 0; i < expectedResultSets; i++)
+            {
+                Assert.Equal(i, batch.ResultSets[i].Id);
+            }
            Assert.Equal(expectedResultSets, batch.ResultSummaries.Length);

            // Make sure that the error state is set properly