WIP - test with transactions

Author: doron050

end2end/EndToEndTests/MySqlConnectorDapperTester.generated.cs

@@ -1,8 +1,10 @@
+using MySqlConnector;
 using MySqlConnectorDapperExampleGen;
 using NUnit.Framework;
 using NUnit.Framework.Legacy;
 using System;
 using System.Collections.Generic;
+using System.Diagnostics;
 using System.Linq;
 using System.Threading.Tasks;
 
@@ -26,6 +28,30 @@ public async Task TestOne()
             Assert.That(SingularEquals(expected, actual));
         }
 
+        [Test]
+        public async Task TestTransactionOne()
+        {
+            var connection = new MySqlConnection(Environment.GetEnvironmentVariable(EndToEndCommon.MySqlConnectionStringEnv)!);
+            await connection.OpenAsync();
+            var transaction = connection.BeginTransaction();
+
+            var sqlQueryWithTx = new QuerySql(transaction);
+            await sqlQueryWithTx.CreateAuthor(new QuerySql.CreateAuthorArgs { Id = 1111, Name = "Bojack Horseman", Bio = "Back in the 90s he was in a very famous TV show" });
+  
+            var actualNull = await this.QuerySql.GetAuthor(new QuerySql.GetAuthorArgs { Name = "Bojack Horseman" });
+            Assert.That(actualNull == null, "there is author");
+            await transaction.CommitAsync();
+
+            var expected = new QuerySql.GetAuthorRow
+            {
+                Id = 1111,
+                Name = "Bojack Horseman",
+                Bio = "Back in the 90s he was in a very famous TV show"
+            };
+            var actual = await this.QuerySql.GetAuthor(new QuerySql.GetAuthorArgs { Name = "Bojack Horseman" });
+            Assert.That(SingularEquals(expected, actual));
+        }
+
         private static bool SingularEquals(QuerySql.GetAuthorRow x, QuerySql.GetAuthorRow y)
         {
             return x.Id.Equals(y.Id) && x.Name.Equals(y.Name) && x.Bio.Equals(y.Bio);
@@ -625,4 +651,4 @@ public async Task TestCopyFrom(int batchSize, MysqlTypesCEnum? cEnum)
             Assert.That(actual.CEnum, Is.EqualTo(expected.CEnum));
         }
     }
-}
+}

examples/MySqlConnectorDapperExample/QuerySql.cs

@@ -1,9 +1,9 @@
-// auto-generated by sqlc - do not edit
-// ReSharper disable UseObjectOrCollectionInitializer
-// ReSharper disable UseAwaitUsing
-// ReSharper disable ConvertToUsingDeclaration
-// ReSharper disable NotAccessedPositionalProperty.Global
-// ReSharper disable UnusedAutoPropertyAccessor.Global
+// auto-generated by sqlc - do not edit
+// ReSharper disable UseObjectOrCollectionInitializer
+// ReSharper disable UseAwaitUsing
+// ReSharper disable ConvertToUsingDeclaration
+// ReSharper disable NotAccessedPositionalProperty.Global
+// ReSharper disable UnusedAutoPropertyAccessor.Global
 using System;
 using System.Collections.Generic;
 using System.Threading.Tasks;
@@ -19,13 +19,24 @@
 namespace MySqlConnectorDapperExampleGen;
 public class QuerySql
 {
-    public QuerySql(string connectionString)
+    public QuerySql()
     {
-        this.ConnectionString = connectionString;
         Dapper.DefaultTypeMap.MatchNamesWithUnderscores = true;
     }
 
-    private string ConnectionString { get; }
+    public QuerySql(string connectionString) : this()
+    {
+        this.ConnectionString = connectionString;
+    }
+
+    public QuerySql(MySqlTransaction transaction) : this()
+    {
+        this.mySqlTransaction = transaction;
+    }
+
+    private MySqlTransaction? mySqlTransaction { get; }
+
+    private string? ConnectionString { get; }
 
     private const string GetAuthorSql = "SELECT id, name, bio FROM authors WHERE name = @name LIMIT 1; SELECT LAST_INSERT_ID()";
     public class GetAuthorRow
@@ -40,13 +51,26 @@ public class GetAuthorArgs
     };
     public async Task<GetAuthorRow?> GetAuthor(GetAuthorArgs args)
     {
-        using (var connection = new MySqlConnection(ConnectionString))
+        var queryParams = new Dictionary<string, object?>();
+        queryParams.Add("name", args.Name);
+        if (this.mySqlTransaction == null)
         {
-            var queryParams = new Dictionary<string, object?>();
-            queryParams.Add("name", args.Name);
-            var result = await connection.QueryFirstOrDefaultAsync<GetAuthorRow?>(GetAuthorSql, queryParams);
-            return result;
+            using (var connection = new MySqlConnection(ConnectionString))
+            {
+                var result = await connection.QueryFirstOrDefaultAsync<GetAuthorRow?>(GetAuthorSql, queryParams);
+                return result;
+            }
+        }
+
+        if (this.mySqlTransaction?.Connection == null || this.mySqlTransaction?.Connection.State != System.Data.ConnectionState.Open) {
+            throw new InvalidOperationException("Transaction is provided, but its connection is null.");
         }
+        
+        return await mySqlTransaction.Connection.QueryFirstOrDefaultAsync<GetAuthorRow?>(
+                GetAuthorSql,
+                queryParams,
+                transaction: this.mySqlTransaction);
+        
     }
 
     private const string ListAuthorsSql = "SELECT id, name, bio FROM authors ORDER BY name; SELECT LAST_INSERT_ID()";
@@ -74,14 +98,27 @@ public class CreateAuthorArgs
     };
     public async Task CreateAuthor(CreateAuthorArgs args)
     {
-        using (var connection = new MySqlConnection(ConnectionString))
+        var queryParams = new Dictionary<string, object?>();
+        queryParams.Add("id", args.Id);
+        queryParams.Add("name", args.Name);
+        queryParams.Add("bio", args.Bio);
+        if (this.mySqlTransaction == null)
         {
-            var queryParams = new Dictionary<string, object?>();
-            queryParams.Add("id", args.Id);
-            queryParams.Add("name", args.Name);
-            queryParams.Add("bio", args.Bio);
-            await connection.ExecuteAsync(CreateAuthorSql, queryParams);
+            using (var connection = new MySqlConnection(ConnectionString))
+            {
+                await connection.ExecuteAsync(CreateAuthorSql, queryParams);
+            }
+        }
+
+        if (this.mySqlTransaction?.Connection == null || this.mySqlTransaction?.Connection.State != System.Data.ConnectionState.Open)
+        {
+            throw new InvalidOperationException("Transaction is provided, but its connection is null.");
         }
+        
+        await mySqlTransaction.Connection.ExecuteAsync(
+                CreateAuthorSql,
+                queryParams,
+                transaction: this.mySqlTransaction);
     }
 
     private const string CreateAuthorReturnIdSql = "INSERT INTO authors (name, bio) VALUES (@name, @bio); SELECT LAST_INSERT_ID()";