Add test to reproduce regression in Managed SNI.

dotnet · Oct 18, 2021 · e8f2283 · e8f2283
1 parent 8ebeab0
commit e8f2283
Showing 1 changed file with 56 additions and 0 deletions.
diff --git a/src/Microsoft.Data.SqlClient/tests/ManualTests/SQL/MARSTest/MARSTest.cs b/src/Microsoft.Data.SqlClient/tests/ManualTests/SQL/MARSTest/MARSTest.cs
@@ -639,5 +639,61 @@ public static async Task MarsScenarioClientJoin()
                 Assert.Equal(companyNames[supplier], name);
             }
         }
+
+        [ConditionalFact(typeof(DataTestUtility), nameof(DataTestUtility.AreConnStringsSetup))]
+        public static void MarsConcurrencyTest()
+        {
+            var table = DataTestUtility.GenerateObjectName();
+            using (var conn = new SqlConnection(DataTestUtility.TCPConnectionString))
+            {
+                conn.Open();
+                using var cmd = new SqlCommand
+                {
+                    Connection = conn,
+                    CommandText = @$"
+                        DROP TABLE IF EXISTS [{table}];
+                        CREATE TABLE [{table}] (
+                            [Id] INTEGER,
+                            [IsDeleted] BIT
+                        )"
+                };
+
+                cmd.ExecuteNonQuery();
+            }
+            var connString = new SqlConnectionStringBuilder(DataTestUtility.TCPConnectionString) { MultipleActiveResultSets = true }.ConnectionString;
+            using (var conn = new SqlConnection(connString))
+            {
+                conn.Open();
+                try
+                {
+                    for (int i = 0; i < 5; i++)
+                    {
+                        Parallel.For(
+                            0, 300,
+                            i =>
+                            {
+                                using var cmd = new SqlCommand
+                                {
+                                    Connection = conn,
+                                    CommandText = @$"
+                                SELECT [l].[Id], [l].[IsDeleted]
+                                FROM [{table}] AS [l]
+                                WHERE ([l].[IsDeleted] = CAST(0 AS bit)) AND [l].[Id] IN (1, 2, 3)"
+                                };
+
+                                using SqlDataReader _ = cmd.ExecuteReader();
+                            });
+                    }
+                }
+                catch (Exception e)
+                {
+                    Assert.False(true, "CRITIAL: Test should not fail randomly. Exception occurred: " + e.Message);
+                }
+                finally
+                {
+                    DataTestUtility.DropTable(conn, table);
+                }
+            }
+        }
     }
 }