Merge pull request #131 from msallin/#127

#127

Author: msallin

.gitignore

@@ -183,3 +183,4 @@ FakesAssemblies/
 
 # Custom
 *.GhostDoc.xml
+/.vs/*

SQLite.CodeFirst.Test/IntegrationTests/InMemoryDbCreationTest.cs

@@ -0,0 +1,64 @@
+using System.Collections.Generic;
+using System.Linq;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using SQLite.CodeFirst.Console;
+using SQLite.CodeFirst.Console.Entity;
+
+namespace SQLite.CodeFirst.Test.IntegrationTests
+{
+
+    [TestClass]
+    public class InMemoryDbCreationTest : InMemoryDbTest<FootballDbContext>
+    {
+        [TestMethod]
+        public void CreateInMemoryDatabaseTest()
+        {
+            using (var context = GetDbContext())
+            {
+                context.Set<Team>().Add(new Team
+                {
+                    Name = "New",
+                    Coach = new Coach
+                    {
+                        City = "New",
+                        FirstName = "New",
+                        LastName = "New",
+                        Street = "New"
+                    },
+                    Players = new List<Player>
+                        {
+                            new Player
+                            {
+                                City = "New",
+                                FirstName = "New",
+                                LastName = "New",
+                                Street = "New",
+                                Number = 1
+                            },
+                            new Player
+                            {
+                                City = "New",
+                                FirstName = "New",
+                                LastName = "New",
+                                Street = "New",
+                                Number = 2
+                            }
+                        },
+                    Stadion = new Stadion
+                    {
+                        Name = "New",
+                        City = "New",
+                        Street = "New"
+                    }
+                });
+
+                context.SaveChanges();
+            }
+
+            using (var context = GetDbContext())
+            {
+                Assert.AreEqual(1, context.Set<Team>().Count());
+            }
+        }
+    }
+}

SQLite.CodeFirst.Test/IntegrationTests/InMemoryDbTest.cs

@@ -0,0 +1,48 @@
+using System;
+using System.Data.Common;
+using System.Data.Entity;
+using System.Data.SQLite;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+
+namespace SQLite.CodeFirst.Test.IntegrationTests
+{
+    /// <summary>
+    ///  Inherit this class to create a test which uses a SQLite in memory database. 
+    ///  This class provides the necessary logic to run multiple tests again the in memory database in a row.
+    /// </summary>
+    public abstract class InMemoryDbTest<TDbContext> 
+        where TDbContext : DbContext
+    {
+        private bool dbInitialized;
+
+        protected DbConnection Connection { get; private set; }
+
+        protected DbContext GetDbContext()
+        {
+            TDbContext context = (TDbContext)Activator.CreateInstance(typeof(TDbContext), Connection, false);
+            if (!dbInitialized)
+            {
+                context.Database.Initialize(true);
+                dbInitialized = true;
+            }
+            return context;
+        }
+
+        [TestInitialize]
+        public void Initialize()
+        {
+            Connection = new SQLiteConnection("data source=:memory:");
+
+            // This is important! Else the in memory database will not work.
+            Connection.Open();
+
+            dbInitialized = false;
+        }
+
+        [TestCleanup]
+        public void Cleanup()
+        {
+            Connection.Dispose();
+        }
+    }
+}

SQLite.CodeFirst.Test/IntegrationTests/MemoryDbTest.cs

@@ -86,10 +86,11 @@
     <Compile Include="..\Shared\AssemblySharedInfo.cs">
       <Link>Properties\AssemblySharedInfo.cs</Link>
     </Compile>
+    <Compile Include="IntegrationTests\InMemoryDbCreationTest.cs" />
+    <Compile Include="IntegrationTests\InMemoryDbTest.cs" />
     <Compile Include="IntegrationTests\SqlGenerationTest.cs" />
     <Compile Include="UnitTests\Builder\NameCreators\IndexNameCreatorTest.cs" />
     <Compile Include="UnitTests\Builder\NameCreators\NameCreatorTest.cs" />
-    <Compile Include="IntegrationTests\MemoryDbTest.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
     <Compile Include="UnitTests\Statement\ColumnConstraint\MaxLengthConstraint.cs" />
     <Compile Include="UnitTests\Statement\ColumnConstraint\PrimaryKeyConstraintTest.cs" />