Add the SqlServerDatabase.ClearData() method.

Author: GillesTourreau

src/Directory.Build.props

@@ -12,11 +12,12 @@
     <PackageLicenseExpression>MIT</PackageLicenseExpression>
     <PackageReleaseNotes>
       2.2.0
+      - Add SqlServerDatabase.ClearDataAsync() method.
       - Add SqlServer.CreateDatabase() method to create database from an Entity Framework DbContext.
       - Add SqlServer.CreateEmptyDatabaseAsync() method.
       - Add SqlServer.DeleteDatabaseAsync() method.
-      - Add SqlServer.ExecuteScriptAsync() method.
-      - Add SqlServer.InsertIntoAsync() method.
+      - Add SqlServerDatabase.ExecuteScriptAsync() method.
+      - Add SqlServerDatabase.InsertIntoAsync() method.
 
       2.1.0
       - PosInformatique.Testing.Databases.SqlServer target the .NET Standard 2.0 platform.

src/Testing.Databases.SqlServer/SqlServerDatabaseExtensions.cs

@@ -100,34 +100,24 @@ public static Task<int> InsertIntoAsync<T>(this SqlServerDatabase database, stri
         /// <param name="database">SQL Server database which the data have to be deleted.</param>
         public static void ClearAllData(this SqlServerDatabase database)
         {
-            database.ExecuteNonQuery("EXEC sp_msforeachtable 'ALTER TABLE ? NOCHECK CONSTRAINT all'");
-
-            database.ExecuteNonQuery("EXEC sp_msforeachtable 'SET QUOTED_IDENTIFIER ON; DELETE FROM ?'");
-
-            // Re-initialize the seed of the IDENTITY columns.
-            // For each table which contains an IDENTITY column, execute the following SQL statement:
-            //   DBCC CHECKIDENT ('[<schema>].[<table>]', RESEED, <seed>)
-            database.ExecuteNonQuery(@"
-                DECLARE @sqlcmd VARCHAR(MAX);
-
-                SET @sqlcmd = (
-	                SELECT STRING_AGG(CAST('DBCC CHECKIDENT (''[' + [s].[name] + '].[' + [t].[name] + ']'', RESEED, ' + CAST([ic].[seed_value] AS VARCHAR(20)) + ')' AS NVARCHAR(MAX)),';' + CHAR(10)) WITHIN GROUP (ORDER BY [t].[name])
-	                FROM
-		                [sys].[schemas] AS [s],
-		                [sys].[tables] AS [t],
-		                [sys].[columns] AS [c],
-		                [sys].[identity_columns] AS [ic]
-	                WHERE
-			                [s].[schema_id] = [t].[schema_id]
-		                AND [t].[object_id] = [c].[object_id]
-		                AND [c].[is_identity] = 1
-		                AND [c].[object_id] = [ic].[object_id]
-		                AND [c].[column_id] = [ic].[column_id]
-                )
-
-                EXEC (@sqlcmd)");
+            foreach (var statement in GetClearDataStatements())
+            {
+                database.ExecuteNonQuery(statement);
+            }
+        }
 
-            database.ExecuteNonQuery("EXEC sp_msforeachtable 'ALTER TABLE ? WITH CHECK CHECK CONSTRAINT all'");
+        /// <summary>
+        /// Clear all in the database asynchronously.
+        /// </summary>
+        /// <param name="database">SQL Server database which the data have to be deleted.</param>
+        /// <param name="cancellationToken"><see cref="CancellationToken"/> used to cancel the asynchronous operation.</param>
+        /// <returns>A <see cref="Task"/> which represents the asynchronous operation.</returns>
+        public static async Task ClearAllDataAsync(this SqlServerDatabase database, CancellationToken cancellationToken = default)
+        {
+            foreach (var statement in GetClearDataStatements())
+            {
+                await database.ExecuteNonQueryAsync(statement, cancellationToken);
+            }
         }
 
         /// <summary>
@@ -243,6 +233,34 @@ Type t when Array.Exists(AuthorizedNonStringTypes, at => at == t) => builder.Add
             return statement;
         }
 
+        private static string[] GetClearDataStatements()
+        {
+            return
+            [
+                "EXEC sp_msforeachtable 'ALTER TABLE ? NOCHECK CONSTRAINT all'",
+                "EXEC sp_msforeachtable 'SET QUOTED_IDENTIFIER ON; DELETE FROM ?'",
+                @"DECLARE @sqlcmd VARCHAR(MAX);
+
+                SET @sqlcmd = (
+	                SELECT STRING_AGG(CAST('DBCC CHECKIDENT (''[' + [s].[name] + '].[' + [t].[name] + ']'', RESEED, ' + CAST([ic].[seed_value] AS VARCHAR(20)) + ')' AS NVARCHAR(MAX)),';' + CHAR(10)) WITHIN GROUP (ORDER BY [t].[name])
+	                FROM
+		                [sys].[schemas] AS [s],
+		                [sys].[tables] AS [t],
+		                [sys].[columns] AS [c],
+		                [sys].[identity_columns] AS [ic]
+	                WHERE
+			                [s].[schema_id] = [t].[schema_id]
+		                AND [t].[object_id] = [c].[object_id]
+		                AND [c].[is_identity] = 1
+		                AND [c].[object_id] = [ic].[object_id]
+		                AND [c].[column_id] = [ic].[column_id]
+                )
+
+                EXEC (@sqlcmd)",
+                "EXEC sp_msforeachtable 'ALTER TABLE ? WITH CHECK CHECK CONSTRAINT all'",
+            ];
+        }
+
         private sealed class SqlInsertStatementBuilder
         {
             private readonly string tableName;