Fix | Fix issue with SqlCommand OnStatementCompleted event not being triggered

src/Microsoft.Data.SqlClient/netcore/src/Microsoft/Data/SqlClient/TdsParser.cs

@@ -2792,6 +2792,11 @@ private bool TryProcessDone(SqlCommand cmd, SqlDataReader reader, ref RunBehavio
                         cmd.InternalRecordsAffected = count;
                     }
                 }
+                // Skip the bogus DONE counts sent by the server
+                if (stateObj._receivedColMetaData || (curCmd != TdsEnums.SELECT))
+                {
+                    cmd.OnStatementCompleted(count);
+                }
             }
 
             stateObj._receivedColMetaData = false;

src/Microsoft.Data.SqlClient/tests/ManualTests/Microsoft.Data.SqlClient.ManualTesting.Tests.csproj

@@ -75,8 +75,9 @@
     <Compile Include="DDBasics\DDMARSTest\DDMARSTest.cs" />
     <Compile Include="ProviderAgnostic\MultipleResultsTest\MultipleResultsTest.cs" />
     <Compile Include="ProviderAgnostic\ReaderTest\ReaderTest.cs" />
-    <Compile Include="SQL\AsyncTest\AsyncTest.cs" /> 
-    <Compile Include="SQL\CommandCancelTest\CommandCancelTest.cs" />
+    <Compile Include="SQL\AsyncTest\AsyncTest.cs" />
+    <Compile Include="SQL\SqlCommand\SqlCommandCompletedTest.cs" /> 
+    <Compile Include="SQL\SqlCommand\SqlCommandCancelTest.cs" />
     <Compile Include="SQL\ConnectionPoolTest\ConnectionPoolTest.cs" />
     <Compile Include="SQL\ConnectivityTests\ConnectivityTest.cs" />
     <Compile Include="SQL\DataBaseSchemaTest\ConnectionSchemaTest.cs" />

src/Microsoft.Data.SqlClient/tests/ManualTests/SQL/CommandCancelTest/CommandCancelTest.cs → src/Microsoft.Data.SqlClient/tests/ManualTests/SQL/SqlCommand/SqlCommandCancelTest.cs

@@ -10,7 +10,7 @@
 
 namespace Microsoft.Data.SqlClient.ManualTesting.Tests
 {
-    public static class CommandCancelTest
+    public static class SqlCommandCancelTest
     {
         // Shrink the packet size - this should make timeouts more likely
         private static readonly string s_connStr = (new SqlConnectionStringBuilder(DataTestUtility.TcpConnStr) { PacketSize = 512 }).ConnectionString;
@@ -139,7 +139,7 @@ private static void MultiThreadedCancel(string constr, bool async)
 
                 Task.WaitAll(tasks, 15 * 1000);
 
-                CommandCancelTest.VerifyConnection(command);
+                SqlCommandCancelTest.VerifyConnection(command);
             }
         }
 

src/Microsoft.Data.SqlClient/tests/ManualTests/SQL/SqlCommand/SqlCommandCompletedTest.cs

@@ -0,0 +1,32 @@
+using System.Data;
+using Xunit;
+
+namespace Microsoft.Data.SqlClient.ManualTesting.Tests
+{
+    public static class SqlCommandCompletedTest
+    {
+        private static readonly string s_connStr = (new SqlConnectionStringBuilder(DataTestUtility.TcpConnStr) { PacketSize = 512 }).ConnectionString;
+        private static bool completedHandlerExecuted = false;
+
+        [CheckConnStrSetupFact]
+        public static void VerifyStatmentCompletedCalled()
+        {
+            using (var conn = new SqlConnection(s_connStr))
+            {
+                using (var cmd = conn.CreateCommand())
+                {
+                    cmd.CommandText = "SELECT COUNT(*) FROM sys.databases";
+                    cmd.StatementCompleted += StatementCompletedHandler;
+                    conn.Open();
+                    var res = cmd.ExecuteScalar();
+                }
+            }
+            Assert.True(completedHandlerExecuted);
+        }
+
+        private static void StatementCompletedHandler(object sender, StatementCompletedEventArgs args)
+        {
+            completedHandlerExecuted = true;
+        }
+    }
+}