Couple TransactionScope tests for EF

Jeff Treuting · Jeff Treuting · commit 93a3afde3656 · 2013-02-12T13:50:44.000-08:00
diff --git a/SharpRepository.Tests.Integration/RepositoryAddTests.cs b/SharpRepository.Tests.Integration/RepositoryAddTests.cs
@@ -1,6 +1,6 @@
-using System;
 using System.Collections.Generic;
 using System.Linq;
+using System.Transactions;
 using NUnit.Framework;
 using SharpRepository.Repository;
 using SharpRepository.Tests.Integration.TestAttributes;
@@ -66,5 +66,33 @@ public void Add_Should_Save_And_Assigned_New_Ids_To_Multiple(IRepository<Contact
             var added = repository.GetAll();
             added.Count().ShouldEqual(3);
         }
+
+        [ExecuteForRepositories(RepositoryTypes.Ef5)]
+        public void Using_TransactionScope_Without_Complete_Should_Not_Add(IRepository<Contact, string> repository)
+        {
+            repository.Get("test"); // used to create the SqlCe database before being inside the transaction scope since that throws an error
+
+            using (var trans = new TransactionScope())
+            {
+                repository.Add(new Contact {Name = "Contact 1"});
+            }
+
+            repository.GetAll().Count().ShouldEqual(0);
+        }
+
+        [ExecuteForRepositories(RepositoryTypes.Ef5)]
+        public void Using_TransactionScope_With_Complete_Should_Add(IRepository<Contact, string> repository)
+        {
+            repository.Get("test"); // used to create the SqlCe database before being inside the transaction scope since that throws an error
+
+            using (var trans = new TransactionScope())
+            {
+                repository.Add(new Contact { Name = "Contact 1" });
+
+                trans.Complete();
+            }
+
+            repository.GetAll().Count().ShouldEqual(1);
+        }
     }
 }
diff --git a/SharpRepository.Tests.Integration/SharpRepository.Tests.Integration.csproj b/SharpRepository.Tests.Integration/SharpRepository.Tests.Integration.csproj
@@ -112,6 +112,7 @@
     <Reference Include="System.configuration" />
     <Reference Include="System.Core" />
     <Reference Include="System.Data.Entity" />
+    <Reference Include="System.Transactions" />
     <Reference Include="System.Xml.Linq" />
     <Reference Include="System.Data.DataSetExtensions" />
     <Reference Include="Microsoft.CSharp" />
@@ -134,6 +135,7 @@
     <Compile Include="Spikes\RavenDbEmbeddedTests.cs" />
     <Compile Include="Spikes\StandardCachingSpikes.cs" />
     <Compile Include="TestAttributes\ExecuteForAllRepositoriesAttribute.cs" />
+    <Compile Include="TestAttributes\ExecuteForAllRepositoriesExcept.cs" />
     <Compile Include="TestAttributes\ExecuteForRepositoriesAttribute.cs" />
     <Compile Include="RepositoryAddTests.cs" />
     <Compile Include="RepositoryDeleteTests.cs" />
diff --git a/SharpRepository.Tests.Integration/TestAttributes/ExecuteForAllRepositoriesExcept.cs b/SharpRepository.Tests.Integration/TestAttributes/ExecuteForAllRepositoriesExcept.cs
@@ -0,0 +1,58 @@
+﻿using System.Collections.Generic;
+using System.Linq;
+using NUnit.Framework;
+using SharpRepository.Tests.Integration.Data;
+
+namespace SharpRepository.Tests.Integration.TestAttributes
+{
+    public class ExecuteForAllRepositoriesExceptAttribute : TestCaseSourceAttribute
+    {
+        private static IEnumerable<TestCaseData> ForAllRepositoriesExceptTestCaseData
+        {
+            get
+            {
+                var repositoryTypes = new[]
+                                          {
+                                                                     RepositoryTypes.Dbo4, 
+                                                                     RepositoryTypes.RavenDb, 
+                                                                     RepositoryTypes.Xml,
+                                                                     RepositoryTypes.MongoDb, 
+                                                                     RepositoryTypes.InMemory, 
+                                                                     RepositoryTypes.Ef5, 
+                                                                     RepositoryTypes.Cache, 
+                                          };
+
+                return RepositoryTestCaseDataFactory.Build(RemoveExceptions(repositoryTypes));
+            }
+        }
+
+        private static RepositoryTypes[] RemoveExceptions(RepositoryTypes[] repositoryTypes)
+        {
+            if (_exceptions == null || _exceptions.Length == 0)
+                return repositoryTypes;
+
+            var list = new List<RepositoryTypes>();
+            list.AddRange(repositoryTypes);
+            foreach (var exception in _exceptions.Where(list.Contains))
+            {
+                list.Remove(exception);
+            }
+
+            return list.ToArray();
+        }
+
+        private static RepositoryTypes[] _exceptions;
+        private static string _reason;
+
+        public ExecuteForAllRepositoriesExceptAttribute(string reason, params RepositoryTypes[] exceptions) : this()
+        {
+            _reason = reason; // TODO: it would be nice to find a way to display this in the Unit test results, but helpful in the unit test for human readable in the code
+            _exceptions = exceptions;
+        }
+
+        public ExecuteForAllRepositoriesExceptAttribute()
+            : base(typeof(ExecuteForAllRepositoriesExceptAttribute), "ForAllRepositoriesExceptTestCaseData")
+        {
+        }
+    }
+}
