Fix for Issue #43

Author: msallin

SQLite.CodeFirst.Console/App.config

@@ -4,7 +4,7 @@
     <section name="entityFramework" type="System.Data.Entity.Internal.ConfigFile.EntityFrameworkSection, EntityFramework, Version=6.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089" requirePermission="false" />
   </configSections>
   <connectionStrings>
-    <add name="footballDb" connectionString="data source=.\footballDb.sqlite" providerName="System.Data.SQLite" />
+    <add name="footballDb" connectionString="data source=.\footballDb.sqlite;foreign keys=true" providerName="System.Data.SQLite" />
   </connectionStrings>
   <startup>
     <supportedRuntime version="v4.0" sku=".NETFramework,Version=v4.5.2" />

SQLite.CodeFirst.Console/FootballDbContext.cs

@@ -16,12 +16,13 @@ public FootballDbContext()
 
         protected override void OnModelCreating(DbModelBuilder modelBuilder)
         {
+            modelBuilder.Conventions.Remove<PluralizingTableNameConvention>();
+
             ConfigureTeamEntity(modelBuilder);
             ConfigureStadionEntity(modelBuilder);
+            ConfigureCoachEntity(modelBuilder);
             ConfigurePlayerEntity(modelBuilder);
 
-            modelBuilder.Conventions.Remove<PluralizingTableNameConvention>();
-
             var initializer = new FootballDbInitializer(modelBuilder);
             Database.SetInitializer(initializer);
         }
@@ -31,9 +32,14 @@ private static void ConfigureTeamEntity(DbModelBuilder modelBuilder)
             modelBuilder.Entity<Team>();
 
             modelBuilder.Entity<Team>()
-                .HasOptional(p => p.Coach)
+                .HasRequired(t => t.Coach)
                 .WithMany()
                 .WillCascadeOnDelete(false);
+
+            modelBuilder.Entity<Team>()
+                .HasRequired(t => t.Stadion)
+                .WithRequiredPrincipal()
+                .WillCascadeOnDelete(true);
         }
 
         private static void ConfigureStadionEntity(DbModelBuilder modelBuilder)
@@ -45,8 +51,8 @@ private static void ConfigureCoachEntity(DbModelBuilder modelBuilder)
         {
             modelBuilder.Entity<Coach>()
                 .HasRequired(p => p.Team)
-                .WithMany()
-                .WillCascadeOnDelete(true);
+                .WithRequiredPrincipal(t => t.Coach)
+                .WillCascadeOnDelete(false);
         }
 
         private static void ConfigurePlayerEntity(DbModelBuilder modelBuilder)

SQLite.CodeFirst/Builder/CreateDatabaseStatementBuilder.cs

@@ -2,6 +2,7 @@
 using System.Data.Entity.Core.Metadata.Edm;
 using System.Linq;
 using SQLite.CodeFirst.Statement;
+using SQLite.CodeFirst.Utility;
 
 namespace SQLite.CodeFirst.Builder
 {
@@ -30,7 +31,14 @@ private IEnumerable<CreateTableStatement> GetCreateTableStatements()
                 ICollection<AssociationType> associationTypes =
                     edmModel.AssociationTypes.Where(a => a.Constraint.ToRole.Name == entitySet.Name).ToList();
 
-                var tableStatementBuilder = new CreateTableStatementBuilder(entitySet, associationTypes);
+                var b = associationTypes.Select(a => new AssociationTypeWrapper
+                {
+                    AssociationType = a,
+                    FromTableName = edmModel.Container.GetEntitySetByName(a.Constraint.FromRole.Name, true).Table,
+                    ToTableName = edmModel.Container.GetEntitySetByName(a.Constraint.ToRole.Name, true).Table
+                });
+
+                var tableStatementBuilder = new CreateTableStatementBuilder(entitySet, b);
                 yield return tableStatementBuilder.BuildStatement();
             }
         }

SQLite.CodeFirst/Builder/CreateTableStatementBuilder.cs

@@ -1,16 +1,16 @@
 using System.Collections.Generic;
 using System.Data.Entity.Core.Metadata.Edm;
-using SQLite.CodeFirst.Extensions;
 using SQLite.CodeFirst.Statement;
+using SQLite.CodeFirst.Utility;
 
 namespace SQLite.CodeFirst.Builder
 {
     internal class CreateTableStatementBuilder : IStatementBuilder<CreateTableStatement>
     {
         private readonly EntitySet entitySet;
-        private readonly IEnumerable<AssociationType> associationTypes;
+        private readonly IEnumerable<AssociationTypeWrapper> associationTypes;
 
-        public CreateTableStatementBuilder(EntitySet entitySet, IEnumerable<AssociationType> associationTypes)
+        public CreateTableStatementBuilder(EntitySet entitySet, IEnumerable<AssociationTypeWrapper> associationTypes)
         {
             this.entitySet = entitySet;
             this.associationTypes = associationTypes;

SQLite.CodeFirst/Builder/ForeignKeyStatementBuilder.cs

@@ -2,14 +2,15 @@
 using System.Data.Entity.Core.Metadata.Edm;
 using System.Linq;
 using SQLite.CodeFirst.Statement;
+using SQLite.CodeFirst.Utility;
 
 namespace SQLite.CodeFirst.Builder
 {
     internal class ForeignKeyStatementBuilder : IStatementBuilder<ColumnStatementCollection>
     {
-        private readonly IEnumerable<AssociationType> associationTypes;
+        private readonly IEnumerable<AssociationTypeWrapper> associationTypes;
 
-        public ForeignKeyStatementBuilder(IEnumerable<AssociationType> associationTypes)
+        public ForeignKeyStatementBuilder(IEnumerable<AssociationTypeWrapper> associationTypes)
         {
             this.associationTypes = associationTypes;
         }
@@ -26,10 +27,10 @@ private IEnumerable<ForeignKeyStatement> GetForeignKeyStatements()
             {
                 yield return new ForeignKeyStatement
                 {
-                    ForeignKey = associationType.Constraint.ToProperties.Select(x => x.Name),
-                    ForeignTable = associationType.Constraint.FromRole.Name,
-                    ForeignPrimaryKey = associationType.Constraint.FromProperties.Select(x => x.Name),
-                    CascadeDelete = associationType.Constraint.FromRole.DeleteBehavior == OperationAction.Cascade
+                    ForeignKey = associationType.AssociationType.Constraint.ToProperties.Select(x => x.Name),
+                    ForeignTable = associationType.FromTableName,
+                    ForeignPrimaryKey = associationType.AssociationType.Constraint.FromProperties.Select(x => x.Name),
+                    CascadeDelete = associationType.AssociationType.Constraint.FromRole.DeleteBehavior == OperationAction.Cascade
                 };
             }
         }

SQLite.CodeFirst/SQLite.CodeFirst.csproj

@@ -73,6 +73,7 @@
     <Reference Include="System.Data" />
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="Utility\AssociationTypeWrapper.cs" />
     <Compile Include="Builder\ColumnStatementCollectionBuilder.cs" />
     <Compile Include="Builder\CreateIndexStatementBuilder.cs" />
     <Compile Include="Builder\ForeignKeyStatementBuilder.cs" />

SQLite.CodeFirst/Utility/AssociationTypeWrapper.cs

@@ -0,0 +1,11 @@
+using System.Data.Entity.Core.Metadata.Edm;
+
+namespace SQLite.CodeFirst.Utility
+{
+    internal class AssociationTypeWrapper
+    {
+        public AssociationType AssociationType { get; set; }
+        public string FromTableName { get; set; }
+        public string ToTableName { get; set; }
+    }
+}