fix test

dotnet · Jun 23, 2021 · d800dd4 · d800dd4
1 parent 481b6a0
commit d800dd4
Showing 1 changed file with 16 additions and 15 deletions.
diff --git a/...Microsoft.Data.SqlClient/tests/ManualTests/SQL/RetryLogic/SqlConnectionReliabilityTest.cs b/...Microsoft.Data.SqlClient/tests/ManualTests/SQL/RetryLogic/SqlConnectionReliabilityTest.cs
@@ -66,34 +66,35 @@ public void CreateDatabaseWhileTryingToConnect(string cnnString, SqlRetryLogicBa
             using (var cnn1 = new SqlConnection(new SqlConnectionStringBuilder(cnnString) { ConnectTimeout = 60, Pooling = false }.ConnectionString))
             {
                 cnn1.Open();
-                Task createDBTask = null;
-                try
+                using (var cmd = cnn1.CreateCommand())
                 {
-                    provider.Retrying += (s, e) =>
+                    Task createDBTask = null;
+                    try
                     {
-                        currentRetries = e.RetryCount;
-                        using (var cmd = cnn1.CreateCommand())
+                        provider.Retrying += (s, e) =>
                         {
+                            currentRetries = e.RetryCount;
                             // Try to create database just after first faliure.
                             if (createDBTask == null || createDBTask.Status == TaskStatus.Faulted)
                             {
+                                Console.WriteLine($"createDBTask.Status:{createDBTask?.Status}");
                                 cmd.CommandText = $"IF (NOT EXISTS(SELECT 1 FROM sys.databases WHERE name = '{database}')) CREATE DATABASE [{database}];";
                                 createDBTask = cmd.ExecuteNonQueryAsync();
                             }
-                        }
-                    };
+                        };
 
-                    using (var cnn2 = new SqlConnection(builder.ConnectionString))
+                        using (var cnn2 = new SqlConnection(builder.ConnectionString))
+                        {
+                            cnn2.RetryLogicProvider = provider;
+                            cnn2.Open();
+                        }
+                    }
+                    finally
                     {
-                        cnn2.RetryLogicProvider = provider;
-                        cnn2.Open();
+                        createDBTask?.Wait();
+                        DataTestUtility.DropDatabase(cnn1, database);
                     }
                 }
-                finally
-                {
-                    createDBTask?.Wait();
-                    DataTestUtility.DropDatabase(cnn1, database);
-                }
             }
             Assert.True(currentRetries > 0);
         }