Add test case.

dotnet · Apr 28, 2020 · 81d9908 · 81d9908
1 parent ab5c096
commit 81d9908
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/src/Microsoft.Data.SqlClient/tests/ManualTests/SQL/SqlCommand/SqlCommandCancelTest.cs b/src/Microsoft.Data.SqlClient/tests/ManualTests/SQL/SqlCommand/SqlCommandCancelTest.cs
@@ -120,6 +120,36 @@ public static void TimeOutDuringRead()
             TimeOutDuringRead(s_connStr);
         }
 
+        [CheckConnStrSetupFact]
+        public static void TCPAttentionPacketTest()
+        {
+            CancelFollowedByTransaction(tcp_connStr);
+        }
+
+        [CheckConnStrSetupFact]
+        public static void NPAttentionPacketTest()
+        {
+            CancelFollowedByTransaction(np_connStr);
+        }
+
+        private static void CancelFollowedByTransaction(string constr)
+        {
+            using (SqlConnection connection = new SqlConnection(constr))
+            {
+                connection.Open();
+                using (SqlCommand cmd = connection.CreateCommand())
+                {
+                    cmd.CommandText = @"SELECT @@VERSION";
+                    using (var r = cmd.ExecuteReader())
+                    {
+                        cmd.Cancel();
+                    }
+                }
+                using (var transaction = connection.BeginTransaction())
+                { }
+            }
+        }
+
         private static void MultiThreadedCancel(string constr, bool async)
         {
             using (SqlConnection con = new SqlConnection(constr))