Add missing transactions to test methods

gentoorax · Jan 28, 2025 · bd944bd · bd944bd
1 parent 1765b4b
commit bd944bd
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/Promethix.Framework.Ado.Tests/TestSupport/DataAccess/Mssql/SimpleMssqlTestRepository.cs b/Promethix.Framework.Ado.Tests/TestSupport/DataAccess/Mssql/SimpleMssqlTestRepository.cs
@@ -26,26 +26,28 @@ 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;
 
         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);
         }
     }
 }