From bd944bda96f3707363f6375a4ef0e708682f9ec2 Mon Sep 17 00:00:00 2001
From: Christopher Law <chris@chrislaw.me>
Date: Tue, 28 Jan 2025 11:17:12 +0000
Subject: [PATCH] Add missing transactions to test methods

---
 .../DataAccess/Mssql/SimpleMssqlTestRepository.cs  | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/Promethix.Framework.Ado.Tests/TestSupport/DataAccess/Mssql/SimpleMssqlTestRepository.cs b/Promethix.Framework.Ado.Tests/TestSupport/DataAccess/Mssql/SimpleMssqlTestRepository.cs
index 895c139..2d7d982 100644
--- a/Promethix.Framework.Ado.Tests/TestSupport/DataAccess/Mssql/SimpleMssqlTestRepository.cs
+++ b/Promethix.Framework.Ado.Tests/TestSupport/DataAccess/Mssql/SimpleMssqlTestRepository.cs
@@ -26,6 +26,8 @@ public SimpleMssqlTestRepository(IAmbientAdoContextLocator ambientAdoContextLoca
 
         private IDbConnection SqlConnection2 => ambientAdoContextLocator.GetContext<MssqlContextExample2>().Connection;
 
+        private IDbTransaction SqlTransaction2 => ambientAdoContextLocator.GetContext<MssqlContextExample2>().Transaction;
+
         private IDbConnection SqlConnection3 => ambientAdoContextLocator.GetContext<MssqlContextExample3>().Connection;
 
         private IDbTransaction SqlTransaction3 => ambientAdoContextLocator.GetContext<MssqlContextExample3>().Transaction;
@@ -33,19 +35,19 @@ public SimpleMssqlTestRepository(IAmbientAdoContextLocator ambientAdoContextLoca
         public void Add(TestEntity entity)
         {
             const string query = "INSERT INTO TestEntity (Name, Description, Quantity) VALUES (@Name, @Description, @Quantity)";
-            SqlConnection1.Execute(query, entity);
+            SqlConnection1.Execute(query, entity, SqlTransaction1);
         }
 
         public void AddWithDifferentContext(TestEntity entity)
         {
             const string query = "INSERT INTO TestEntity (Name, Description, Quantity) VALUES (@Name, @Description, @Quantity)";
-            SqlConnection2.Execute(query, entity);
+            SqlConnection2.Execute(query, entity, SqlTransaction2);
         }
 
         public TestEntity? GetEntityByName(string name)
         {
             const string query = "SELECT * FROM TestEntity WHERE Name = @Name";
-            return SqlConnection1.QuerySingleOrDefault<TestEntity>(query, new { Name = name });
+            return SqlConnection1.QuerySingleOrDefault<TestEntity>(query, new { Name = name }, SqlTransaction1);
         }
 
         public IEnumerable<TestEntity> GetEntities()
@@ -63,7 +65,7 @@ public IEnumerable<TestEntity> GetEntitiesNullTransaction()
         public int GetEntityCount()
         {
             const string query = "SELECT COUNT(*) FROM TestEntity";
-            return SqlConnection1.ExecuteScalar<int>(query);
+            return SqlConnection1.ExecuteScalar<int>(query, SqlTransaction1);
         }
 
         public void DivideByZero()
@@ -75,8 +77,8 @@ public void DivideByZero()
         public void DeleteAll()
         {
             const string query = "DELETE FROM TestEntity";
-            SqlConnection1.Execute(query);
-            SqlConnection2.Execute(query);
+            SqlConnection1.Execute(query, SqlTransaction1);
+            SqlConnection2.Execute(query, SqlTransaction2);
         }
     }
 }