Add SqlServer.ExecuteScriptAsync() method.

Author: GillesTourreau

src/Directory.Build.props

@@ -14,6 +14,7 @@
       2.2.0
       - Add SqlServer.CreateEmptyDatabaseAsync() method.
       - Add SqlServer.DeleteDatabaseAsync() method.
+      - Add SqlServer.ExecuteScriptAsync() method.
 
       2.1.0
       - PosInformatique.Testing.Databases.SqlServer target the .NET Standard 2.0 platform.

src/Testing.Databases.SqlServer/SqlServerDatabaseExtensions.cs

@@ -164,6 +164,44 @@ public static void ExecuteScript(this SqlServerDatabase database, StringReader s
             }
         }
 
+        /// <summary>
+        /// Execute an T-SQL script on the <paramref name="database"/>.
+        /// </summary>
+        /// <param name="database"><see cref="SqlServerDatabase"/> where the <paramref name="script"/> will be executed.</param>
+        /// <param name="script">T-SQL script to execute.</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 ExecuteScriptAsync(this SqlServerDatabase database, string script, CancellationToken cancellationToken = default)
+        {
+            using var stringReader = new StringReader(script);
+
+            await ExecuteScriptAsync(database, stringReader);
+        }
+
+        /// <summary>
+        /// Execute an T-SQL script on the <paramref name="database"/> asynchronously.
+        /// </summary>
+        /// <param name="database"><see cref="SqlServerDatabase"/> where the <paramref name="script"/> will be executed.</param>
+        /// <param name="script"><see cref="StringReader"/> which contains the T-SQL script to execute.</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 ExecuteScriptAsync(this SqlServerDatabase database, StringReader script, CancellationToken cancellationToken = default)
+        {
+            var parser = new SqlServerScriptParser(script);
+
+            var block = parser.ReadNextBlock();
+
+            while (block is not null)
+            {
+                for (var i = 0; i < block.Count; i++)
+                {
+                    await database.ExecuteNonQueryAsync(block.Code, cancellationToken);
+                }
+
+                block = parser.ReadNextBlock();
+            }
+        }
+
         private sealed class SqlInsertStatementBuilder
         {
             private readonly string tableName;

tests/Testing.Databases.SqlServer.Tests/SqlServerDatabaseExtensionsTest.cs

@@ -219,5 +219,134 @@ GO 10
 
             table.Rows[0]["Id"].Should().Be(10);
         }
+
+        [Fact]
+        public async Task ExecuteScriptAsync_String()
+        {
+            var server = new SqlServer(ConnectionString);
+
+            var database = server.CreateEmptyDatabase("SqlServerDatabaseExtensionsTest");
+
+            await database.ExecuteScriptAsync(@"
+                CREATE TABLE TableTest
+                (
+                    Id          INT         NOT NULL
+                )
+
+                GO
+                GO
+
+                INSERT INTO [TableTest] ([Id]) VALUES (0)
+
+                GO
+                UPDATE [TableTest]
+                SET [Id] = [Id] + 1
+
+                GO 10");
+
+            var table = await database.ExecuteQueryAsync("SELECT * FROM [TableTest]");
+
+            table.Rows.Should().HaveCount(1);
+
+            table.Rows[0]["Id"].Should().Be(10);
+        }
+
+        [Fact]
+        public async Task ExecuteScriptAsync_String_WithEmptyLinesAtTheEnd()
+        {
+            var server = new SqlServer(ConnectionString);
+
+            var database = server.CreateEmptyDatabase("SqlServerDatabaseExtensionsTest");
+
+            await database.ExecuteScriptAsync(@"
+                CREATE TABLE TableTest
+                (
+                    Id          INT         NOT NULL
+                )
+
+                GO
+                GO
+
+                INSERT INTO [TableTest] ([Id]) VALUES (0)
+
+                GO
+                UPDATE [TableTest]
+                SET [Id] = [Id] + 1
+
+                GO 10
+
+                ");
+
+            var table = await database.ExecuteQueryAsync("SELECT * FROM [TableTest]");
+
+            table.Rows.Should().HaveCount(1);
+
+            table.Rows[0]["Id"].Should().Be(10);
+        }
+
+        [Fact]
+        public async Task ExecuteScriptAsync_StringReader()
+        {
+            var server = new SqlServer(ConnectionString);
+
+            var database = server.CreateEmptyDatabase("SqlServerDatabaseExtensionsTest");
+
+            await database.ExecuteScriptAsync(new StringReader(@"
+                CREATE TABLE TableTest
+                (
+                    Id          INT         NOT NULL
+                )
+
+                GO
+                GO
+
+                INSERT INTO [TableTest] ([Id]) VALUES (0)
+
+                GO
+                UPDATE [TableTest]
+                SET [Id] = [Id] + 1
+
+                GO 10"));
+
+            var table = await database.ExecuteQueryAsync("SELECT * FROM [TableTest]");
+
+            table.Rows.Should().HaveCount(1);
+
+            table.Rows[0]["Id"].Should().Be(10);
+        }
+
+        [Fact]
+        public async Task ExecuteScriptAsync_StringReader_WithEmptyLinesAtTheEnd()
+        {
+            var server = new SqlServer(ConnectionString);
+
+            var database = server.CreateEmptyDatabase("SqlServerDatabaseExtensionsTest");
+
+            await database.ExecuteScriptAsync(new StringReader(@"
+                CREATE TABLE TableTest
+                (
+                    Id          INT         NOT NULL
+                )
+
+                GO
+                GO
+
+                INSERT INTO [TableTest] ([Id]) VALUES (0)
+
+                GO
+                UPDATE [TableTest]
+                SET [Id] = [Id] + 1
+
+                GO 10
+
+
+                "));
+
+            var table = await database.ExecuteQueryAsync("SELECT * FROM [TableTest]");
+
+            table.Rows.Should().HaveCount(1);
+
+            table.Rows[0]["Id"].Should().Be(10);
+        }
     }
 }