From 72b693d91cca3880bd95f8f73483594bd0857086 Mon Sep 17 00:00:00 2001
From: Svein Helge <svein.helge.grinden@gmail.com>
Date: Fri, 31 Jul 2015 00:33:30 +0200
Subject: [PATCH 1/5] New column with name Description added to the table.

---
 EfEnumToLookup/LookupGenerator/EnumParser.cs  |  9 +--
 .../LookupGenerator/EnumToLookup.cs           | 22 ++++--
 EfEnumToLookup/LookupGenerator/IDbHandler.cs  | 14 ++--
 EfEnumToLookup/LookupGenerator/LookupValue.cs |  3 +-
 .../LookupGenerator/SqlServerHandler.cs       | 34 +++++----
 .../EfEnumToLookupTests.csproj                |  3 +
 EfEnumToLookupTests/Tests/EnumParserTests.cs  |  6 +-
 EfEnumToLookupTests/Tests/TestStuff.cs        | 11 +--
 ExampleUsage/EnumExample.cs                   | 69 +++++++++++++++++--
 ExampleUsage/ExampleUsage.csproj              |  7 +-
 10 files changed, 130 insertions(+), 48 deletions(-)

diff --git a/EfEnumToLookup/LookupGenerator/EnumParser.cs b/EfEnumToLookup/LookupGenerator/EnumParser.cs
index 3bd9351..9ecbe48 100644
--- a/EfEnumToLookup/LookupGenerator/EnumParser.cs
+++ b/EfEnumToLookup/LookupGenerator/EnumParser.cs
@@ -57,7 +57,8 @@ public IEnumerable<LookupValue> GetLookupValues(Type lookup)
 				{
 					Id = (int)numericValue,
 					Name = EnumName(value),
-				});
+                    Description = EnumDescriptionValue(value)
+                });
 			}
 			return values;
 		}
@@ -70,12 +71,6 @@ public IEnumerable<LookupValue> GetLookupValues(Type lookup)
 		/// </summary>
 		private string EnumName(Enum value)
 		{
-			var description = EnumDescriptionValue(value);
-			if (description != null)
-			{
-				return description;
-			}
-
 			var name = value.ToString();
 
 			if (SplitWords)
diff --git a/EfEnumToLookup/LookupGenerator/EnumToLookup.cs b/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
index f8138ef..546cabd 100644
--- a/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
+++ b/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
@@ -31,7 +31,8 @@ public EnumToLookup()
 		{
 			// set default behaviour, can be overridden by setting properties on object before calling Apply()
 			NameFieldLength = 255;
-			TableNamePrefix = "Enum_";
+            DescriptionFieldLength = 255;
+            TableNamePrefix = "Enum_";
 			_enumParser = new EnumParser { SplitWords = true };
 		}
 
@@ -51,12 +52,18 @@ public bool SplitWords
 		/// </summary>
 		public int NameFieldLength { get; set; }
 
-		/// <summary>
-		/// Prefix to add to all the generated tables to separate help group them together
-		/// and make them stand out as different from other tables.
-		/// Defaults to "Enum_" set to null or "" to not have any prefix.
+        /// <summary>
+		/// The size of the Description field that will be added to the generated lookup tables.
+		/// Adjust to suit your data if required, defaults to 255.
 		/// </summary>
-		public string TableNamePrefix { get; set; }
+		public int DescriptionFieldLength { get; set; }
+
+        /// <summary>
+        /// Prefix to add to all the generated tables to separate help group them together
+        /// and make them stand out as different from other tables.
+        /// Defaults to "Enum_" set to null or "" to not have any prefix.
+        /// </summary>
+        public string TableNamePrefix { get; set; }
 
 		/// <summary>
 		/// Suffix to add to all the generated tables to separate help group them together
@@ -110,7 +117,8 @@ private IDbHandler GetDbHandler()
 			IDbHandler dbHandler = new SqlServerHandler
 			{
 				NameFieldLength = NameFieldLength,
-				TableNamePrefix = TableNamePrefix,
+                DescriptionFieldLength = DescriptionFieldLength,
+                TableNamePrefix = TableNamePrefix,
 				TableNameSuffix = TableNameSuffix,
 			};
 			return dbHandler;
diff --git a/EfEnumToLookup/LookupGenerator/IDbHandler.cs b/EfEnumToLookup/LookupGenerator/IDbHandler.cs
index c26c47c..2f4ca31 100644
--- a/EfEnumToLookup/LookupGenerator/IDbHandler.cs
+++ b/EfEnumToLookup/LookupGenerator/IDbHandler.cs
@@ -12,11 +12,17 @@ internal interface IDbHandler
 		/// </summary>
 		int NameFieldLength { get; set; }
 
-		/// <summary>
-		/// Prefix to add to all the generated tables to separate help group them together
-		/// and make them stand out as different from other tables.
+        /// <summary>
+		/// The size of the Description field that will be added to the generated lookup tables.
+		/// Adjust to suit your data if required.
 		/// </summary>
-		string TableNamePrefix { get; set; }
+		int DescriptionFieldLength { get; set; }
+
+        /// <summary>
+        /// Prefix to add to all the generated tables to separate help group them together
+        /// and make them stand out as different from other tables.
+        /// </summary>
+        string TableNamePrefix { get; set; }
 
 		/// <summary>
 		/// Suffix to add to all the generated tables to separate help group them together
diff --git a/EfEnumToLookup/LookupGenerator/LookupValue.cs b/EfEnumToLookup/LookupGenerator/LookupValue.cs
index ac1c0ae..4b62398 100644
--- a/EfEnumToLookup/LookupGenerator/LookupValue.cs
+++ b/EfEnumToLookup/LookupGenerator/LookupValue.cs
@@ -4,5 +4,6 @@ internal class LookupValue
 	{
 		public int Id { get; set; }
 		public string Name { get; set; }
-	}
+        public string Description { get; set; }
+    }
 }
diff --git a/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs b/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
index 7830b6b..ed8cdc4 100644
--- a/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
+++ b/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
@@ -13,12 +13,18 @@ class SqlServerHandler : IDbHandler
 		/// </summary>
 		public int NameFieldLength { get; set; }
 
-		/// <summary>
-		/// Prefix to add to all the generated tables to separate help group them together
-		/// and make them stand out as different from other tables.
-		/// Defaults to "Enum_" set to null or "" to not have any prefix.
+        /// <summary>
+		/// The size of the Description field that will be added to the generated lookup tables.
+		/// Adjust to suit your data if required, defaults to 255.
 		/// </summary>
-		public string TableNamePrefix { get; set; }
+        public int DescriptionFieldLength { get; set; }
+
+        /// <summary>
+        /// Prefix to add to all the generated tables to separate help group them together
+        /// and make them stand out as different from other tables.
+        /// Defaults to "Enum_" set to null or "" to not have any prefix.
+        /// </summary>
+        public string TableNamePrefix { get; set; }
 
 		/// <summary>
 		/// Suffix to add to all the generated tables to separate help group them together
@@ -58,14 +64,14 @@ private string CreateTables(IEnumerable<LookupData> enums)
 			foreach (var lookup in enums)
 			{
 				sql.AppendFormat(
-					@"IF OBJECT_ID('{0}', 'U') IS NULL
+                    @"IF OBJECT_ID('{0}', 'U') IS NULL
 begin
-	CREATE TABLE [{0}] (Id {2} CONSTRAINT PK_{0} PRIMARY KEY, Name nvarchar({1}));
+	CREATE TABLE [{0}] (Id {2} CONSTRAINT PK_{0} PRIMARY KEY, Name nvarchar({1}), Description nvarchar({3}));
 	exec sys.sp_addextendedproperty @name=N'MS_Description', @level0type=N'SCHEMA', @level0name=N'dbo', @level1type=N'TABLE',
 		@level1name=N'{0}', @value=N'Automatically generated. Contents will be overwritten on app startup. Table & contents generated by https://github.com/timabell/ef-enum-to-lookup';
 end
 ",
-					TableName(lookup.Name), NameFieldLength, NumericSqlType(lookup.NumericType));
+					TableName(lookup.Name), NameFieldLength, NumericSqlType(lookup.NumericType), DescriptionFieldLength);
 			}
 			return sql.ToString();
 		}
@@ -88,7 +94,7 @@ private string AddForeignKeys(IEnumerable<EnumReference> refs)
 		private string PopulateLookups(IEnumerable<LookupData> lookupData)
 		{
 			var sql = new StringBuilder();
-			sql.AppendLine(string.Format("CREATE TABLE #lookups (Id int, Name nvarchar({0}) COLLATE database_default);", NameFieldLength));
+			sql.AppendLine(string.Format("CREATE TABLE #lookups (Id int, Name nvarchar({0}), Description nvarchar({1}) COLLATE database_default);", NameFieldLength, DescriptionFieldLength));
 			foreach (var lookup in lookupData)
 			{
 				sql.AppendLine(PopulateLookup(lookup));
@@ -102,7 +108,7 @@ private string PopulateLookup(LookupData lookup)
 			var sql = new StringBuilder();
 			foreach (var value in lookup.Values)
 			{
-				sql.AppendFormat("INSERT INTO #lookups (Id, Name) VALUES ({0}, N'{1}');\r\n", value.Id, SanitizeSqlString(value.Name));
+				sql.AppendFormat("INSERT INTO #lookups (Id, Name, Description) VALUES ({0}, N'{1}', N'{2}');\r\n", value.Id, SanitizeSqlString(value.Name), SanitizeSqlString(value.Description));
 			}
 
 			sql.AppendLine(string.Format(@"
@@ -111,12 +117,12 @@ MERGE INTO [{0}] dst
 	WHEN MATCHED AND src.Name <> dst.Name THEN
 		UPDATE SET Name = src.Name
 	WHEN NOT MATCHED THEN
-		INSERT (Id, Name)
-		VALUES (src.Id, src.Name)
+		INSERT (Id, Name, Description)
+		VALUES (src.Id, src.Name, src.Description)
 	WHEN NOT MATCHED BY SOURCE THEN
 		DELETE
 ;"
-				, TableName(lookup.Name)));
+                , TableName(lookup.Name)));
 
 			sql.AppendLine("TRUNCATE TABLE #lookups;");
 			return sql.ToString();
@@ -124,7 +130,7 @@ WHEN NOT MATCHED BY SOURCE THEN
 
 		private static string SanitizeSqlString(string value)
 		{
-			return value.Replace("'", "''");
+			return value == null ? null : value.Replace("'", "''");
 		}
 
 		private string TableName(string enumName)
diff --git a/EfEnumToLookupTests/EfEnumToLookupTests.csproj b/EfEnumToLookupTests/EfEnumToLookupTests.csproj
index c1764a2..b202401 100644
--- a/EfEnumToLookupTests/EfEnumToLookupTests.csproj
+++ b/EfEnumToLookupTests/EfEnumToLookupTests.csproj
@@ -100,6 +100,9 @@
       <Name>EfEnumToLookup</Name>
     </ProjectReference>
   </ItemGroup>
+  <ItemGroup>
+    <Service Include="{82A7F48D-3B50-4B1E-B82E-3ADA8210C358}" />
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.
diff --git a/EfEnumToLookupTests/Tests/EnumParserTests.cs b/EfEnumToLookupTests/Tests/EnumParserTests.cs
index cb04a71..da7c7bd 100644
--- a/EfEnumToLookupTests/Tests/EnumParserTests.cs
+++ b/EfEnumToLookupTests/Tests/EnumParserTests.cs
@@ -48,7 +48,7 @@ public void ReadsByteEnum()
 
 
 		[Test]
-		public void ReadsDecoratedName()
+		public void ReadsDecoratedDescription()
 		{
 			// arrange
 			var parser = new EnumParser { SplitWords = true };
@@ -57,7 +57,7 @@ public void ReadsDecoratedName()
 			var result = parser.GetLookupValues(typeof(DecoratedEnum));
 
 			// assert
-			Assert.AreEqual("Wide boy", result.Single().Name);
+			Assert.AreEqual("Wide boy", result.Single().Description);
 		}
 
 		private enum BareEnum
@@ -78,7 +78,7 @@ private enum DecoratedEnum
 		{
 			// ReSharper disable once UnusedMember.Local
 			// used by test suite
-			[System.ComponentModel.Description("Wide boy")]
+			[System.ComponentModel.Description("Wide boy")]            
 			FooBar
 		}
 	}
diff --git a/EfEnumToLookupTests/Tests/TestStuff.cs b/EfEnumToLookupTests/Tests/TestStuff.cs
index 3bbe946..9837593 100644
--- a/EfEnumToLookupTests/Tests/TestStuff.cs
+++ b/EfEnumToLookupTests/Tests/TestStuff.cs
@@ -19,8 +19,11 @@ public void SetUp()
 			// Using setup rather than teardown to make it easier to inspect the database after running a test.
 			using (var context = new MagicContext())
 			{
-				context.Database.Delete();
-			}
+                if(context.Database.Exists())
+                { 
+                    context.Database.Delete();
+                }
+            }
 
 			Database.SetInitializer(new TestInitializer(new EnumToLookup()));
 			using (var context = new MagicContext())
@@ -63,9 +66,9 @@ public void UsesDescriptionAttribute()
 		{
 			using (var context = new MagicContext())
 			{
-				const string sql = "select @desciption = name from Enum_Importance where id = @id";
+				const string sql = "select @description = description from Enum_Importance where id = @id";
 				var idParam = new SqlParameter("id", (int)Importance.NotBovverd);
-				var outParam = new SqlParameter("desciption", SqlDbType.NVarChar, 255) { Direction = ParameterDirection.Output };
+				var outParam = new SqlParameter("description", SqlDbType.NVarChar, 255) { Direction = ParameterDirection.Output };
 				context.Database.ExecuteSqlCommand(sql, idParam, outParam);
 				var actualName = outParam.Value;
 				Assert.AreEqual(Constants.BovveredDisplay, actualName);
diff --git a/ExampleUsage/EnumExample.cs b/ExampleUsage/EnumExample.cs
index 38e6974..2c82b91 100644
--- a/ExampleUsage/EnumExample.cs
+++ b/ExampleUsage/EnumExample.cs
@@ -1,5 +1,8 @@
 using System;
+using System.Data;
 using System.Data.Entity;
+using System.Data.SqlClient;
+using System.Linq;
 using EfEnumToLookup.LookupGenerator;
 using NUnit.Framework;
 
@@ -39,7 +42,8 @@ public void ExampleOfUsingApply()
 			}
 		}
 
-		[Test] public void ExampleOfGeneratingSql()
+		[Test]
+        public void ExampleOfGeneratingSql()
 		{
 			using (var context = new MyDbContext())
 			{
@@ -55,14 +59,57 @@ [Test] public void ExampleOfGeneratingSql()
 				context.Database.ExecuteSqlCommand(migrationSql);
 			}
 		}
-	}
+
+        [Test]        
+        public void CheckIfDescriptionColumnsExists()
+        {
+            using (var context = new MyDbContext())
+            {
+                var enumToLookup = new EnumToLookup();
+
+                // if you need to get at the raw sql to run a migration separately then use:
+                var migrationSql = enumToLookup.GenerateMigrationSql(context);
+                // you'd probably want to write this to a file and then add it to source control, but for
+                // the purpose of demonstration we'll write it to the console instead:
+                Console.Out.WriteLine(migrationSql);
+
+                // at some point you'd then run the sql (probably not like this, but this serves as a test that it's working)
+                context.Database.ExecuteSqlCommand(migrationSql);
+
+                var descriptions = context.Database.SqlQuery<Enum>("Select Id, Name, Description From Enum_Size");
+
+                var descriptionList = descriptions.ToList();
+
+                Assert.AreEqual(4, descriptionList.Count);                
+
+                Assert.AreEqual(string.Empty, descriptionList.Single(x => x.Id == (int)Size.Small).Description);
+                Assert.AreEqual("It's average", descriptionList.Single(x => x.Id == (int)Size.Medium).Description);
+                Assert.AreEqual(string.Empty, descriptionList.Single(x => x.Id == (int)Size.ReallyVeryBig).Description);
+                Assert.AreEqual("Huge you know?", descriptionList.Single(x => x.Id == (int)Size.Huge).Description);
+
+                var shapes = context.Database.SqlQuery<Enum>("Select Id, Name, Description From Enum_Shape");
+
+                var shapeList = shapes.ToList();
+
+                Assert.AreEqual(2, shapeList.Count);
+
+                Assert.AreEqual(string.Empty, shapeList.Single(x => x.Id == (int)Shape.Square).Description);
+                Assert.AreEqual("Round it is!", shapeList.Single(x => x.Id == (int)Shape.Round).Description);
+            }
+        }
+    }
 
 	/// <summary>
 	/// Example context
 	/// </summary>
 	public class MyDbContext : DbContext
-	{
-		public DbSet<Foo> Foos { get; set; }
+    {
+        
+        public MyDbContext()
+        {
+        }
+
+        public DbSet<Foo> Foos { get; set; }
 	}
 
 	/// <summary>
@@ -87,9 +134,9 @@ public enum Size
 		Medium,
 
 		ReallyVeryBig,
-
+        
 		// this is only fully qualified because of a name clash with NUnit, you wouldn't normally need to.
-		[System.ComponentModel.Description("Huge you know?")] // give it a different name in the lookup table
+		[System.ComponentModel.Description("Huge you know?")]
 		Huge,
 
 		[RuntimeOnly] // this won't exist in the database, handy to prevent unwanted data creeping in (enforced by foreign key constraint).
@@ -99,6 +146,14 @@ public enum Size
 	public enum Shape
 	{
 		Square,
-		Round
+        [System.ComponentModel.Description("Round it is!")]
+        Round
 	}
+
+    public class Enum
+    {
+        public int Id { get; set; }
+        public string Name { get; set; }
+        public string Description { get; set; }
+    }
 }
diff --git a/ExampleUsage/ExampleUsage.csproj b/ExampleUsage/ExampleUsage.csproj
index 21e06e1..8e44a8a 100644
--- a/ExampleUsage/ExampleUsage.csproj
+++ b/ExampleUsage/ExampleUsage.csproj
@@ -53,7 +53,9 @@
     <Compile Include="Properties\AssemblyInfo.cs" />
   </ItemGroup>
   <ItemGroup>
-    <None Include="App.config" />
+    <None Include="App.config">
+      <SubType>Designer</SubType>
+    </None>
     <None Include="packages.config" />
   </ItemGroup>
   <ItemGroup>
@@ -62,6 +64,9 @@
       <Name>EfEnumToLookup</Name>
     </ProjectReference>
   </ItemGroup>
+  <ItemGroup>
+    <Service Include="{82A7F48D-3B50-4B1E-B82E-3ADA8210C358}" />
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.

From 5f93bebc866d1bf62c71ac926c6e439efaf9f844 Mon Sep 17 00:00:00 2001
From: Svein Helge <svein.helge.grinden@gmail.com>
Date: Fri, 31 Jul 2015 00:33:30 +0200
Subject: [PATCH 2/5] New column with name Description added to the table.

---
 EfEnumToLookup/LookupGenerator/EnumParser.cs  |  9 +--
 .../LookupGenerator/EnumToLookup.cs           | 22 +++++--
 EfEnumToLookup/LookupGenerator/IDbHandler.cs  | 14 ++--
 EfEnumToLookup/LookupGenerator/LookupValue.cs |  3 +-
 .../LookupGenerator/SqlServerHandler.cs       | 34 ++++++----
 .../EfEnumToLookupTests.csproj                |  3 +
 EfEnumToLookupTests/Tests/EnumParserTests.cs  |  6 +-
 EfEnumToLookupTests/Tests/TestStuff.cs        | 11 ++--
 ExampleUsage/EnumExample.cs                   | 64 +++++++++++++++++--
 ExampleUsage/ExampleUsage.csproj              |  7 +-
 10 files changed, 125 insertions(+), 48 deletions(-)

diff --git a/EfEnumToLookup/LookupGenerator/EnumParser.cs b/EfEnumToLookup/LookupGenerator/EnumParser.cs
index 3bd9351..9ecbe48 100644
--- a/EfEnumToLookup/LookupGenerator/EnumParser.cs
+++ b/EfEnumToLookup/LookupGenerator/EnumParser.cs
@@ -57,7 +57,8 @@ public IEnumerable<LookupValue> GetLookupValues(Type lookup)
 				{
 					Id = (int)numericValue,
 					Name = EnumName(value),
-				});
+                    Description = EnumDescriptionValue(value)
+                });
 			}
 			return values;
 		}
@@ -70,12 +71,6 @@ public IEnumerable<LookupValue> GetLookupValues(Type lookup)
 		/// </summary>
 		private string EnumName(Enum value)
 		{
-			var description = EnumDescriptionValue(value);
-			if (description != null)
-			{
-				return description;
-			}
-
 			var name = value.ToString();
 
 			if (SplitWords)
diff --git a/EfEnumToLookup/LookupGenerator/EnumToLookup.cs b/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
index f8138ef..546cabd 100644
--- a/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
+++ b/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
@@ -31,7 +31,8 @@ public EnumToLookup()
 		{
 			// set default behaviour, can be overridden by setting properties on object before calling Apply()
 			NameFieldLength = 255;
-			TableNamePrefix = "Enum_";
+            DescriptionFieldLength = 255;
+            TableNamePrefix = "Enum_";
 			_enumParser = new EnumParser { SplitWords = true };
 		}
 
@@ -51,12 +52,18 @@ public bool SplitWords
 		/// </summary>
 		public int NameFieldLength { get; set; }
 
-		/// <summary>
-		/// Prefix to add to all the generated tables to separate help group them together
-		/// and make them stand out as different from other tables.
-		/// Defaults to "Enum_" set to null or "" to not have any prefix.
+        /// <summary>
+		/// The size of the Description field that will be added to the generated lookup tables.
+		/// Adjust to suit your data if required, defaults to 255.
 		/// </summary>
-		public string TableNamePrefix { get; set; }
+		public int DescriptionFieldLength { get; set; }
+
+        /// <summary>
+        /// Prefix to add to all the generated tables to separate help group them together
+        /// and make them stand out as different from other tables.
+        /// Defaults to "Enum_" set to null or "" to not have any prefix.
+        /// </summary>
+        public string TableNamePrefix { get; set; }
 
 		/// <summary>
 		/// Suffix to add to all the generated tables to separate help group them together
@@ -110,7 +117,8 @@ private IDbHandler GetDbHandler()
 			IDbHandler dbHandler = new SqlServerHandler
 			{
 				NameFieldLength = NameFieldLength,
-				TableNamePrefix = TableNamePrefix,
+                DescriptionFieldLength = DescriptionFieldLength,
+                TableNamePrefix = TableNamePrefix,
 				TableNameSuffix = TableNameSuffix,
 			};
 			return dbHandler;
diff --git a/EfEnumToLookup/LookupGenerator/IDbHandler.cs b/EfEnumToLookup/LookupGenerator/IDbHandler.cs
index c26c47c..2f4ca31 100644
--- a/EfEnumToLookup/LookupGenerator/IDbHandler.cs
+++ b/EfEnumToLookup/LookupGenerator/IDbHandler.cs
@@ -12,11 +12,17 @@ internal interface IDbHandler
 		/// </summary>
 		int NameFieldLength { get; set; }
 
-		/// <summary>
-		/// Prefix to add to all the generated tables to separate help group them together
-		/// and make them stand out as different from other tables.
+        /// <summary>
+		/// The size of the Description field that will be added to the generated lookup tables.
+		/// Adjust to suit your data if required.
 		/// </summary>
-		string TableNamePrefix { get; set; }
+		int DescriptionFieldLength { get; set; }
+
+        /// <summary>
+        /// Prefix to add to all the generated tables to separate help group them together
+        /// and make them stand out as different from other tables.
+        /// </summary>
+        string TableNamePrefix { get; set; }
 
 		/// <summary>
 		/// Suffix to add to all the generated tables to separate help group them together
diff --git a/EfEnumToLookup/LookupGenerator/LookupValue.cs b/EfEnumToLookup/LookupGenerator/LookupValue.cs
index ac1c0ae..4b62398 100644
--- a/EfEnumToLookup/LookupGenerator/LookupValue.cs
+++ b/EfEnumToLookup/LookupGenerator/LookupValue.cs
@@ -4,5 +4,6 @@ internal class LookupValue
 	{
 		public int Id { get; set; }
 		public string Name { get; set; }
-	}
+        public string Description { get; set; }
+    }
 }
diff --git a/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs b/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
index 7830b6b..ed8cdc4 100644
--- a/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
+++ b/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
@@ -13,12 +13,18 @@ class SqlServerHandler : IDbHandler
 		/// </summary>
 		public int NameFieldLength { get; set; }
 
-		/// <summary>
-		/// Prefix to add to all the generated tables to separate help group them together
-		/// and make them stand out as different from other tables.
-		/// Defaults to "Enum_" set to null or "" to not have any prefix.
+        /// <summary>
+		/// The size of the Description field that will be added to the generated lookup tables.
+		/// Adjust to suit your data if required, defaults to 255.
 		/// </summary>
-		public string TableNamePrefix { get; set; }
+        public int DescriptionFieldLength { get; set; }
+
+        /// <summary>
+        /// Prefix to add to all the generated tables to separate help group them together
+        /// and make them stand out as different from other tables.
+        /// Defaults to "Enum_" set to null or "" to not have any prefix.
+        /// </summary>
+        public string TableNamePrefix { get; set; }
 
 		/// <summary>
 		/// Suffix to add to all the generated tables to separate help group them together
@@ -58,14 +64,14 @@ private string CreateTables(IEnumerable<LookupData> enums)
 			foreach (var lookup in enums)
 			{
 				sql.AppendFormat(
-					@"IF OBJECT_ID('{0}', 'U') IS NULL
+                    @"IF OBJECT_ID('{0}', 'U') IS NULL
 begin
-	CREATE TABLE [{0}] (Id {2} CONSTRAINT PK_{0} PRIMARY KEY, Name nvarchar({1}));
+	CREATE TABLE [{0}] (Id {2} CONSTRAINT PK_{0} PRIMARY KEY, Name nvarchar({1}), Description nvarchar({3}));
 	exec sys.sp_addextendedproperty @name=N'MS_Description', @level0type=N'SCHEMA', @level0name=N'dbo', @level1type=N'TABLE',
 		@level1name=N'{0}', @value=N'Automatically generated. Contents will be overwritten on app startup. Table & contents generated by https://github.com/timabell/ef-enum-to-lookup';
 end
 ",
-					TableName(lookup.Name), NameFieldLength, NumericSqlType(lookup.NumericType));
+					TableName(lookup.Name), NameFieldLength, NumericSqlType(lookup.NumericType), DescriptionFieldLength);
 			}
 			return sql.ToString();
 		}
@@ -88,7 +94,7 @@ private string AddForeignKeys(IEnumerable<EnumReference> refs)
 		private string PopulateLookups(IEnumerable<LookupData> lookupData)
 		{
 			var sql = new StringBuilder();
-			sql.AppendLine(string.Format("CREATE TABLE #lookups (Id int, Name nvarchar({0}) COLLATE database_default);", NameFieldLength));
+			sql.AppendLine(string.Format("CREATE TABLE #lookups (Id int, Name nvarchar({0}), Description nvarchar({1}) COLLATE database_default);", NameFieldLength, DescriptionFieldLength));
 			foreach (var lookup in lookupData)
 			{
 				sql.AppendLine(PopulateLookup(lookup));
@@ -102,7 +108,7 @@ private string PopulateLookup(LookupData lookup)
 			var sql = new StringBuilder();
 			foreach (var value in lookup.Values)
 			{
-				sql.AppendFormat("INSERT INTO #lookups (Id, Name) VALUES ({0}, N'{1}');\r\n", value.Id, SanitizeSqlString(value.Name));
+				sql.AppendFormat("INSERT INTO #lookups (Id, Name, Description) VALUES ({0}, N'{1}', N'{2}');\r\n", value.Id, SanitizeSqlString(value.Name), SanitizeSqlString(value.Description));
 			}
 
 			sql.AppendLine(string.Format(@"
@@ -111,12 +117,12 @@ MERGE INTO [{0}] dst
 	WHEN MATCHED AND src.Name <> dst.Name THEN
 		UPDATE SET Name = src.Name
 	WHEN NOT MATCHED THEN
-		INSERT (Id, Name)
-		VALUES (src.Id, src.Name)
+		INSERT (Id, Name, Description)
+		VALUES (src.Id, src.Name, src.Description)
 	WHEN NOT MATCHED BY SOURCE THEN
 		DELETE
 ;"
-				, TableName(lookup.Name)));
+                , TableName(lookup.Name)));
 
 			sql.AppendLine("TRUNCATE TABLE #lookups;");
 			return sql.ToString();
@@ -124,7 +130,7 @@ WHEN NOT MATCHED BY SOURCE THEN
 
 		private static string SanitizeSqlString(string value)
 		{
-			return value.Replace("'", "''");
+			return value == null ? null : value.Replace("'", "''");
 		}
 
 		private string TableName(string enumName)
diff --git a/EfEnumToLookupTests/EfEnumToLookupTests.csproj b/EfEnumToLookupTests/EfEnumToLookupTests.csproj
index c1764a2..b202401 100644
--- a/EfEnumToLookupTests/EfEnumToLookupTests.csproj
+++ b/EfEnumToLookupTests/EfEnumToLookupTests.csproj
@@ -100,6 +100,9 @@
       <Name>EfEnumToLookup</Name>
     </ProjectReference>
   </ItemGroup>
+  <ItemGroup>
+    <Service Include="{82A7F48D-3B50-4B1E-B82E-3ADA8210C358}" />
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.
diff --git a/EfEnumToLookupTests/Tests/EnumParserTests.cs b/EfEnumToLookupTests/Tests/EnumParserTests.cs
index cb04a71..da7c7bd 100644
--- a/EfEnumToLookupTests/Tests/EnumParserTests.cs
+++ b/EfEnumToLookupTests/Tests/EnumParserTests.cs
@@ -48,7 +48,7 @@ public void ReadsByteEnum()
 
 
 		[Test]
-		public void ReadsDecoratedName()
+		public void ReadsDecoratedDescription()
 		{
 			// arrange
 			var parser = new EnumParser { SplitWords = true };
@@ -57,7 +57,7 @@ public void ReadsDecoratedName()
 			var result = parser.GetLookupValues(typeof(DecoratedEnum));
 
 			// assert
-			Assert.AreEqual("Wide boy", result.Single().Name);
+			Assert.AreEqual("Wide boy", result.Single().Description);
 		}
 
 		private enum BareEnum
@@ -78,7 +78,7 @@ private enum DecoratedEnum
 		{
 			// ReSharper disable once UnusedMember.Local
 			// used by test suite
-			[System.ComponentModel.Description("Wide boy")]
+			[System.ComponentModel.Description("Wide boy")]            
 			FooBar
 		}
 	}
diff --git a/EfEnumToLookupTests/Tests/TestStuff.cs b/EfEnumToLookupTests/Tests/TestStuff.cs
index 3bbe946..9837593 100644
--- a/EfEnumToLookupTests/Tests/TestStuff.cs
+++ b/EfEnumToLookupTests/Tests/TestStuff.cs
@@ -19,8 +19,11 @@ public void SetUp()
 			// Using setup rather than teardown to make it easier to inspect the database after running a test.
 			using (var context = new MagicContext())
 			{
-				context.Database.Delete();
-			}
+                if(context.Database.Exists())
+                { 
+                    context.Database.Delete();
+                }
+            }
 
 			Database.SetInitializer(new TestInitializer(new EnumToLookup()));
 			using (var context = new MagicContext())
@@ -63,9 +66,9 @@ public void UsesDescriptionAttribute()
 		{
 			using (var context = new MagicContext())
 			{
-				const string sql = "select @desciption = name from Enum_Importance where id = @id";
+				const string sql = "select @description = description from Enum_Importance where id = @id";
 				var idParam = new SqlParameter("id", (int)Importance.NotBovverd);
-				var outParam = new SqlParameter("desciption", SqlDbType.NVarChar, 255) { Direction = ParameterDirection.Output };
+				var outParam = new SqlParameter("description", SqlDbType.NVarChar, 255) { Direction = ParameterDirection.Output };
 				context.Database.ExecuteSqlCommand(sql, idParam, outParam);
 				var actualName = outParam.Value;
 				Assert.AreEqual(Constants.BovveredDisplay, actualName);
diff --git a/ExampleUsage/EnumExample.cs b/ExampleUsage/EnumExample.cs
index 38e6974..5008e6d 100644
--- a/ExampleUsage/EnumExample.cs
+++ b/ExampleUsage/EnumExample.cs
@@ -1,5 +1,8 @@
 using System;
+using System.Data;
 using System.Data.Entity;
+using System.Data.SqlClient;
+using System.Linq;
 using EfEnumToLookup.LookupGenerator;
 using NUnit.Framework;
 
@@ -39,7 +42,8 @@ public void ExampleOfUsingApply()
 			}
 		}
 
-		[Test] public void ExampleOfGeneratingSql()
+		[Test]
+        public void ExampleOfGeneratingSql()
 		{
 			using (var context = new MyDbContext())
 			{
@@ -55,14 +59,52 @@ [Test] public void ExampleOfGeneratingSql()
 				context.Database.ExecuteSqlCommand(migrationSql);
 			}
 		}
-	}
+
+        [Test]        
+        public void CheckIfDescriptionColumnsExists()
+        {
+            using (var context = new MyDbContext())
+            {
+                var enumToLookup = new EnumToLookup();
+
+                // if you need to get at the raw sql to run a migration separately then use:
+                var migrationSql = enumToLookup.GenerateMigrationSql(context);
+                // you'd probably want to write this to a file and then add it to source control, but for
+                // the purpose of demonstration we'll write it to the console instead:
+                Console.Out.WriteLine(migrationSql);
+
+                // at some point you'd then run the sql (probably not like this, but this serves as a test that it's working)
+                context.Database.ExecuteSqlCommand(migrationSql);
+
+                var descriptions = context.Database.SqlQuery<Enum>("Select Id, Name, Description From Enum_Size");
+
+                var descriptionList = descriptions.ToList();
+
+                Assert.AreEqual(4, descriptionList.Count);                
+
+                Assert.AreEqual(string.Empty, descriptionList.Single(x => x.Id == (int)Size.Small).Description);
+                Assert.AreEqual("It's average", descriptionList.Single(x => x.Id == (int)Size.Medium).Description);
+                Assert.AreEqual(string.Empty, descriptionList.Single(x => x.Id == (int)Size.ReallyVeryBig).Description);
+                Assert.AreEqual("Huge you know?", descriptionList.Single(x => x.Id == (int)Size.Huge).Description);
+
+                var shapes = context.Database.SqlQuery<Enum>("Select Id, Name, Description From Enum_Shape");
+
+                var shapeList = shapes.ToList();
+
+                Assert.AreEqual(2, shapeList.Count);
+
+                Assert.AreEqual(string.Empty, shapeList.Single(x => x.Id == (int)Shape.Square).Description);
+                Assert.AreEqual("Round it is!", shapeList.Single(x => x.Id == (int)Shape.Round).Description);
+            }
+        }
+    }
 
 	/// <summary>
 	/// Example context
 	/// </summary>
 	public class MyDbContext : DbContext
-	{
-		public DbSet<Foo> Foos { get; set; }
+    {
+        public DbSet<Foo> Foos { get; set; }
 	}
 
 	/// <summary>
@@ -87,9 +129,9 @@ public enum Size
 		Medium,
 
 		ReallyVeryBig,
-
+        
 		// this is only fully qualified because of a name clash with NUnit, you wouldn't normally need to.
-		[System.ComponentModel.Description("Huge you know?")] // give it a different name in the lookup table
+		[System.ComponentModel.Description("Huge you know?")]
 		Huge,
 
 		[RuntimeOnly] // this won't exist in the database, handy to prevent unwanted data creeping in (enforced by foreign key constraint).
@@ -99,6 +141,14 @@ public enum Size
 	public enum Shape
 	{
 		Square,
-		Round
+        [System.ComponentModel.Description("Round it is!")]
+        Round
 	}
+
+    public class Enum
+    {
+        public int Id { get; set; }
+        public string Name { get; set; }
+        public string Description { get; set; }
+    }
 }
diff --git a/ExampleUsage/ExampleUsage.csproj b/ExampleUsage/ExampleUsage.csproj
index 21e06e1..8e44a8a 100644
--- a/ExampleUsage/ExampleUsage.csproj
+++ b/ExampleUsage/ExampleUsage.csproj
@@ -53,7 +53,9 @@
     <Compile Include="Properties\AssemblyInfo.cs" />
   </ItemGroup>
   <ItemGroup>
-    <None Include="App.config" />
+    <None Include="App.config">
+      <SubType>Designer</SubType>
+    </None>
     <None Include="packages.config" />
   </ItemGroup>
   <ItemGroup>
@@ -62,6 +64,9 @@
       <Name>EfEnumToLookup</Name>
     </ProjectReference>
   </ItemGroup>
+  <ItemGroup>
+    <Service Include="{82A7F48D-3B50-4B1E-B82E-3ADA8210C358}" />
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.

From a58e5b78ea5653ac0c3ea57b10391c0c54c90964 Mon Sep 17 00:00:00 2001
From: Svein Helge <svein.helge.grinden@gmail.com>
Date: Fri, 31 Jul 2015 14:03:18 +0200
Subject: [PATCH 3/5] Version bump v2.0.0

---
 EfEnumToLookup/Properties/AssemblyInfo.cs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/EfEnumToLookup/Properties/AssemblyInfo.cs b/EfEnumToLookup/Properties/AssemblyInfo.cs
index 69eb094..04fba4a 100644
--- a/EfEnumToLookup/Properties/AssemblyInfo.cs
+++ b/EfEnumToLookup/Properties/AssemblyInfo.cs
@@ -33,6 +33,6 @@
 // by using the '*' as shown below:
 // [assembly: AssemblyVersion("1.0.*")]
 [assembly: AssemblyVersion("1.0.0.0")]
-[assembly: AssemblyFileVersion("1.8.0.0")]
+[assembly: AssemblyFileVersion("2.0.0.0")]
 
 [assembly:InternalsVisibleTo("EfEnumToLookupTests,PublicKey=002400000480000094000000060200000024000052534131000400000100010099a4b9a877a10f75bc4c1a9bf08ee9e21a2df3b9746f2535505b77b3f82971aa8a07caa6cd60b9df6e4b2982ebdb78da960ec1e7688a3b56b6da65e05783ee51f345b1cf0e5a38a02f3cc7c1763d7e469edafc59ef8c21b1a9bcca8548973429577362d992e4a6ed001cb28dc54ecc7ef298e94676bd918fd9e9b521e0cb7ae5")]

From 249483ab4573c35f40e8c3c9ca82959fde259789 Mon Sep 17 00:00:00 2001
From: Svein Helge <svein.helge.grinden@gmail.com>
Date: Fri, 31 Jul 2015 14:12:29 +0200
Subject: [PATCH 4/5] Version bump v2.0.0-beta

---
 ef-enum-to-lookup.nuspec | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/ef-enum-to-lookup.nuspec b/ef-enum-to-lookup.nuspec
index 702b9d4..c95411a 100644
--- a/ef-enum-to-lookup.nuspec
+++ b/ef-enum-to-lookup.nuspec
@@ -2,7 +2,7 @@
 <package xmlns="http://schemas.microsoft.com/packaging/2011/08/nuspec.xsd">
 	<metadata>
 		<id>ef-enum-to-lookup</id>
-		<version>1.8.0</version>
+		<version>2.0.0-beta</version>
 		<title>Enum Lookup-Table Generator for Entity Framework 6</title>
 		<authors>Tim Abell</authors>
 		<licenseUrl>http://opensource.org/licenses/MIT</licenseUrl>
@@ -25,10 +25,12 @@
 			Creates lookup tables and foreign key constraints based on the enums used in your model.
 		</summary>
 		<releaseNotes>
-			v1.8.0
-			* stable sql output ordering https://github.com/timabell/ef-enum-to-lookup/issues/29
-			* stable primary key names https://github.com/timabell/ef-enum-to-lookup/issues/31
-		</releaseNotes>
+      v2.0.0
+      * beta Description column added to enum tables https://github.com/timabell/ef-enum-to-lookup/issues/33
+      v1.8.0
+      * stable sql output ordering https://github.com/timabell/ef-enum-to-lookup/issues/29
+      * stable primary key names https://github.com/timabell/ef-enum-to-lookup/issues/31
+    </releaseNotes>
 		<copyright>Tim Abell</copyright>
 		<language>en-GB</language>
 	</metadata>

From e571410e194c13445fc12e881268041a5e37f48e Mon Sep 17 00:00:00 2001
From: Svein Helge <svein.helge.grinden@gmail.com>
Date: Tue, 11 Aug 2015 23:21:01 +0200
Subject: [PATCH 5/5] Added support for custom Schema.

---
 EfEnumToLookup/EfEnumToLookup.csproj          |  3 +-
 .../LookupGenerator/ContextExtensions.cs      | 48 +++++++++++++
 .../LookupGenerator/EnumToLookup.cs           | 56 +++++++--------
 .../LookupGenerator/LookupDbModel.cs          |  1 +
 .../LookupGenerator/SqlServerHandler.cs       | 34 ++++-----
 EfEnumToLookupTests/App.config                |  2 +-
 ExampleUsage/App.config                       |  2 +-
 ExampleUsage/EnumExample.cs                   | 70 +++++++++++++++----
 8 files changed, 153 insertions(+), 63 deletions(-)
 create mode 100644 EfEnumToLookup/LookupGenerator/ContextExtensions.cs

diff --git a/EfEnumToLookup/EfEnumToLookup.csproj b/EfEnumToLookup/EfEnumToLookup.csproj
index 70a0b0e..32a5d14 100644
--- a/EfEnumToLookup/EfEnumToLookup.csproj
+++ b/EfEnumToLookup/EfEnumToLookup.csproj
@@ -47,13 +47,14 @@
     <Reference Include="System" />
     <Reference Include="System.ComponentModel.DataAnnotations" />
     <Reference Include="System.Core" />
+    <Reference Include="System.Data.Entity" />
     <Reference Include="System.Xml.Linq" />
-    <Reference Include="System.Data.DataSetExtensions" />
     <Reference Include="Microsoft.CSharp" />
     <Reference Include="System.Data" />
     <Reference Include="System.Xml" />
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="LookupGenerator\ContextExtensions.cs" />
     <Compile Include="LookupGenerator\EnumGeneratorException.cs" />
     <Compile Include="LookupGenerator\EnumParser.cs" />
     <Compile Include="LookupGenerator\EnumToLookup.cs" />
diff --git a/EfEnumToLookup/LookupGenerator/ContextExtensions.cs b/EfEnumToLookup/LookupGenerator/ContextExtensions.cs
new file mode 100644
index 0000000..3ae32ad
--- /dev/null
+++ b/EfEnumToLookup/LookupGenerator/ContextExtensions.cs
@@ -0,0 +1,48 @@
+namespace EfEnumToLookup.LookupGenerator
+{
+    using System.Data.Entity;
+    using System.Data.Entity.Core.Metadata.Edm;
+    using System.Data.Entity.Core.Objects;
+    using System.Data.Entity.Infrastructure;
+    using System.Linq;
+    using System.Text.RegularExpressions;
+
+    public static class ContextExtensions
+    {
+        public static string GetTableName<T>(this DbContext context) where T : class
+        {
+            var objectContext = ((IObjectContextAdapter)context).ObjectContext;
+
+            return objectContext.GetTableName<T>();
+        }
+
+        public static string GetTableName<T>(this ObjectContext context) where T : class
+        {
+            string sql = context.CreateObjectSet<T>().ToTraceString();
+            Regex regex = new Regex("FROM (?<table>.*) AS");
+            Match match = regex.Match(sql);
+
+            string table = match.Groups["table"].Value;
+            return table;
+        }
+
+        public static string GetDefaultSchema(this DbContext context)
+        {
+            ////var table = (((IObjectContextAdapter)context).ObjectContext).MetadataWorkspace.GetItems(DataSpace.SSpace).FirstOrDefault();
+
+            var table = (((IObjectContextAdapter)context).ObjectContext).MetadataWorkspace.GetItemCollection(DataSpace.SSpace)
+                .GetItems<EntityContainer>()
+                .Single()
+                .BaseEntitySets
+                .OfType<EntitySet>()
+                .SingleOrDefault(s => !s.MetadataProperties.Contains("Type") || s.MetadataProperties["Type"].ToString() == "Tables");
+
+            if(table == null)
+            {
+                return "dbo";
+            }
+
+          return table.MetadataProperties["Schema"].Value.ToString();
+        }
+    }
+}
diff --git a/EfEnumToLookup/LookupGenerator/EnumToLookup.cs b/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
index 546cabd..6578ab5 100644
--- a/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
+++ b/EfEnumToLookup/LookupGenerator/EnumToLookup.cs
@@ -1,29 +1,28 @@
 namespace EfEnumToLookup.LookupGenerator
 {
-	using System;
-	using System.Collections.Generic;
-	using System.ComponentModel;
-	using System.Data.Entity;
-	using System.Data.Entity.Infrastructure;
-	using System.Data.SqlClient;
-	using System.Linq;
-	using System.Reflection;
-	using System.Text;
-	using System.Text.RegularExpressions;
-
-	/// <summary>
-	/// Makes up for a missing feature in Entity Framework 6.1
-	/// Creates lookup tables and foreign key constraints based on the enums
-	/// used in your model.
-	/// Use the properties exposed to control behaviour.
-	/// Run <c>Apply</c> from your Seed method in either your database initializer
-	/// or your EF Migrations.
-	/// It is safe to run repeatedly, and will ensure enum values are kept in line
-	/// with your current code.
-	/// Source code: https://github.com/timabell/ef-enum-to-lookup
-	/// License: MIT
-	/// </summary>
-	public class EnumToLookup : IEnumToLookup
+    using System;
+    using System.Collections.Generic;    
+    using System.Data.Entity;
+    using System.Data.Entity.Core.Metadata.Edm;    
+    using System.Data.Entity.Infrastructure;
+    using System.Data.SqlClient;
+    using System.Linq;
+    using System.Reflection;
+    using System.Text;    
+
+    /// <summary>
+    /// Makes up for a missing feature in Entity Framework 6.1
+    /// Creates lookup tables and foreign key constraints based on the enums
+    /// used in your model.
+    /// Use the properties exposed to control behaviour.
+    /// Run <c>Apply</c> from your Seed method in either your database initializer
+    /// or your EF Migrations.
+    /// It is safe to run repeatedly, and will ensure enum values are kept in line
+    /// with your current code.
+    /// Source code: https://github.com/timabell/ef-enum-to-lookup
+    /// License: MIT
+    /// </summary>
+    public class EnumToLookup : IEnumToLookup
 	{
 		private readonly EnumParser _enumParser;
 
@@ -144,16 +143,17 @@ from enm in enums
 					NumericType = enm.GetEnumUnderlyingType(),
 					Values = _enumParser.GetLookupValues(enm),
 				}).ToList();
-
+            
 			var model = new LookupDbModel
 			{
 				Lookups = lookups,
 				References = enumReferences,
-			};
+                Schema = context.GetDefaultSchema()
+            };
 			return model;
 		}
-
-		private static int ExecuteSqlCommand(DbContext context, string sql, IEnumerable<SqlParameter> parameters = null)
+        
+        private static int ExecuteSqlCommand(DbContext context, string sql, IEnumerable<SqlParameter> parameters = null)
 		{
 			if (parameters == null)
 			{
diff --git a/EfEnumToLookup/LookupGenerator/LookupDbModel.cs b/EfEnumToLookup/LookupGenerator/LookupDbModel.cs
index e8d79b8..eab367b 100644
--- a/EfEnumToLookup/LookupGenerator/LookupDbModel.cs
+++ b/EfEnumToLookup/LookupGenerator/LookupDbModel.cs
@@ -9,6 +9,7 @@ namespace EfEnumToLookup.LookupGenerator
 	/// </summary>
 	internal class LookupDbModel
 	{
+        public string Schema { get; set; }
 		public IList<LookupData> Lookups { get; set; }
 		public IList<EnumReference> References { get; set; }
 	}
diff --git a/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs b/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
index ed8cdc4..8dea874 100644
--- a/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
+++ b/EfEnumToLookup/LookupGenerator/SqlServerHandler.cs
@@ -51,32 +51,33 @@ private string BuildSql(LookupDbModel model)
 			sql.AppendLine("set nocount on;");
 			sql.AppendLine("set xact_abort on; -- rollback on error");
 			sql.AppendLine("begin tran;");
-			sql.AppendLine(CreateTables(model.Lookups));
-			sql.AppendLine(PopulateLookups(model.Lookups));
-			sql.AppendLine(AddForeignKeys(model.References));
+			sql.AppendLine(CreateTables(model.Lookups, model.Schema));
+			sql.AppendLine(PopulateLookups(model.Lookups, model.Schema));
+			sql.AppendLine(AddForeignKeys(model.References, model.Schema));
 			sql.AppendLine("commit;");
 			return sql.ToString();
 		}
 
-		private string CreateTables(IEnumerable<LookupData> enums)
+		private string CreateTables(IEnumerable<LookupData> enums, string schema)
 		{
 			var sql = new StringBuilder();
+            
 			foreach (var lookup in enums)
-			{
+			{                
 				sql.AppendFormat(
                     @"IF OBJECT_ID('{0}', 'U') IS NULL
 begin
-	CREATE TABLE [{0}] (Id {2} CONSTRAINT PK_{0} PRIMARY KEY, Name nvarchar({1}), Description nvarchar({3}));
-	exec sys.sp_addextendedproperty @name=N'MS_Description', @level0type=N'SCHEMA', @level0name=N'dbo', @level1type=N'TABLE',
+	CREATE TABLE [{4}].[{0}] (Id {2} CONSTRAINT PK_{0} PRIMARY KEY, Name nvarchar({1}), Description nvarchar({3}));
+	exec sys.sp_addextendedproperty @name=N'MS_Description', @level0type=N'SCHEMA', @level0name=N'{4}', @level1type=N'TABLE',
 		@level1name=N'{0}', @value=N'Automatically generated. Contents will be overwritten on app startup. Table & contents generated by https://github.com/timabell/ef-enum-to-lookup';
 end
 ",
-					TableName(lookup.Name), NameFieldLength, NumericSqlType(lookup.NumericType), DescriptionFieldLength);
+					TableName(lookup.Name), NameFieldLength, NumericSqlType(lookup.NumericType), DescriptionFieldLength, schema);
 			}
 			return sql.ToString();
 		}
 
-		private string AddForeignKeys(IEnumerable<EnumReference> refs)
+		private string AddForeignKeys(IEnumerable<EnumReference> refs, string schema)
 		{
 			var sql = new StringBuilder();
 			foreach (var enumReference in refs)
@@ -84,26 +85,25 @@ private string AddForeignKeys(IEnumerable<EnumReference> refs)
 				var fkName = string.Format("FK_{0}_{1}", enumReference.ReferencingTable, enumReference.ReferencingField);
 
 				sql.AppendFormat(
-					" IF OBJECT_ID('{0}', 'F') IS NULL ALTER TABLE [{1}] ADD CONSTRAINT {0} FOREIGN KEY ([{2}]) REFERENCES [{3}] (Id);\r\n",
-					fkName, enumReference.ReferencingTable, enumReference.ReferencingField, TableName(enumReference.EnumType.Name)
-				);
+                    " IF OBJECT_ID('{0}', 'F') IS NULL ALTER TABLE [{4}].[{1}] ADD CONSTRAINT {0} FOREIGN KEY ([{2}]) REFERENCES [{4}].[{3}] (Id);\r\n",
+					fkName, enumReference.ReferencingTable, enumReference.ReferencingField, TableName(enumReference.EnumType.Name), schema);
 			}
 			return sql.ToString();
 		}
 
-		private string PopulateLookups(IEnumerable<LookupData> lookupData)
+		private string PopulateLookups(IEnumerable<LookupData> lookupData, string schema)
 		{
 			var sql = new StringBuilder();
 			sql.AppendLine(string.Format("CREATE TABLE #lookups (Id int, Name nvarchar({0}), Description nvarchar({1}) COLLATE database_default);", NameFieldLength, DescriptionFieldLength));
 			foreach (var lookup in lookupData)
 			{
-				sql.AppendLine(PopulateLookup(lookup));
+				sql.AppendLine(PopulateLookup(lookup, schema));
 			}
 			sql.AppendLine("DROP TABLE #lookups;");
 			return sql.ToString();
 		}
 
-		private string PopulateLookup(LookupData lookup)
+		private string PopulateLookup(LookupData lookup, string schema)
 		{
 			var sql = new StringBuilder();
 			foreach (var value in lookup.Values)
@@ -112,7 +112,7 @@ private string PopulateLookup(LookupData lookup)
 			}
 
 			sql.AppendLine(string.Format(@"
-MERGE INTO [{0}] dst
+MERGE INTO [{1}].[{0}] dst
 	USING #lookups src ON src.Id = dst.Id
 	WHEN MATCHED AND src.Name <> dst.Name THEN
 		UPDATE SET Name = src.Name
@@ -122,7 +122,7 @@ WHEN NOT MATCHED THEN
 	WHEN NOT MATCHED BY SOURCE THEN
 		DELETE
 ;"
-                , TableName(lookup.Name)));
+                , TableName(lookup.Name), schema));
 
 			sql.AppendLine("TRUNCATE TABLE #lookups;");
 			return sql.ToString();
diff --git a/EfEnumToLookupTests/App.config b/EfEnumToLookupTests/App.config
index 047f1c8..7643ed3 100644
--- a/EfEnumToLookupTests/App.config
+++ b/EfEnumToLookupTests/App.config
@@ -10,7 +10,7 @@
 	<entityFramework>
 		<defaultConnectionFactory type="System.Data.Entity.Infrastructure.LocalDbConnectionFactory, EntityFramework">
 			<parameters>
-				<parameter value="v11.0" />
+				<parameter value="ProjectsV12" />
 			</parameters>
 		</defaultConnectionFactory>
 		<providers>
diff --git a/ExampleUsage/App.config b/ExampleUsage/App.config
index 1e94b1b..def22fe 100644
--- a/ExampleUsage/App.config
+++ b/ExampleUsage/App.config
@@ -7,7 +7,7 @@
   <entityFramework>
     <defaultConnectionFactory type="System.Data.Entity.Infrastructure.LocalDbConnectionFactory, EntityFramework">
       <parameters>
-        <parameter value="v11.0" />
+        <parameter value="ProjectsV12" />
       </parameters>
     </defaultConnectionFactory>
     <providers>
diff --git a/ExampleUsage/EnumExample.cs b/ExampleUsage/EnumExample.cs
index 5008e6d..3d7e3d8 100644
--- a/ExampleUsage/EnumExample.cs
+++ b/ExampleUsage/EnumExample.cs
@@ -67,27 +67,51 @@ public void CheckIfDescriptionColumnsExists()
             {
                 var enumToLookup = new EnumToLookup();
 
-                // if you need to get at the raw sql to run a migration separately then use:
-                var migrationSql = enumToLookup.GenerateMigrationSql(context);
-                // you'd probably want to write this to a file and then add it to source control, but for
-                // the purpose of demonstration we'll write it to the console instead:
-                Console.Out.WriteLine(migrationSql);
+                enumToLookup.Apply(context);
+                
+                var descriptions = context.Database.SqlQuery<Enum>("Select Id, Name, Description From dbo.Enum_Size");
 
-                // at some point you'd then run the sql (probably not like this, but this serves as a test that it's working)
-                context.Database.ExecuteSqlCommand(migrationSql);
+                var descriptionList = descriptions.ToList();
+
+                Assert.AreEqual(4, descriptionList.Count);                
+
+                Assert.AreEqual(string.Empty, descriptionList.Single(x => x.Id == (int)Size.Small).Description);
+                Assert.AreEqual("It's average", descriptionList.Single(x => x.Id == (int)Size.Medium).Description);
+                Assert.AreEqual(string.Empty, descriptionList.Single(x => x.Id == (int)Size.ReallyVeryBig).Description);
+                Assert.AreEqual("Huge you know?", descriptionList.Single(x => x.Id == (int)Size.Huge).Description);
+
+                var shapes = context.Database.SqlQuery<Enum>("Select Id, Name, Description From dbo.Enum_Shape");
+
+                var shapeList = shapes.ToList();
+
+                Assert.AreEqual(2, shapeList.Count);
+
+                Assert.AreEqual(string.Empty, shapeList.Single(x => x.Id == (int)Shape.Square).Description);
+                Assert.AreEqual("Round it is!", shapeList.Single(x => x.Id == (int)Shape.Round).Description);
+            }
+        }
 
-                var descriptions = context.Database.SqlQuery<Enum>("Select Id, Name, Description From Enum_Size");
+        [Test]
+        public void CheckIfDescriptionColumnsExistsUsingCustomSchema()
+        {
+            using (var context = new MyCustomSchemaDbContext())
+            {
+                var enumToLookup = new EnumToLookup();
+
+                enumToLookup.Apply(context);
+                
+                var descriptions = context.Database.SqlQuery<Enum>("Select Id, Name, Description From myschema.Enum_Size");
 
                 var descriptionList = descriptions.ToList();
 
-                Assert.AreEqual(4, descriptionList.Count);                
+                Assert.AreEqual(4, descriptionList.Count);
 
                 Assert.AreEqual(string.Empty, descriptionList.Single(x => x.Id == (int)Size.Small).Description);
                 Assert.AreEqual("It's average", descriptionList.Single(x => x.Id == (int)Size.Medium).Description);
                 Assert.AreEqual(string.Empty, descriptionList.Single(x => x.Id == (int)Size.ReallyVeryBig).Description);
                 Assert.AreEqual("Huge you know?", descriptionList.Single(x => x.Id == (int)Size.Huge).Description);
 
-                var shapes = context.Database.SqlQuery<Enum>("Select Id, Name, Description From Enum_Shape");
+                var shapes = context.Database.SqlQuery<Enum>("Select Id, Name, Description From myschema.Enum_Shape");
 
                 var shapeList = shapes.ToList();
 
@@ -105,12 +129,28 @@ public void CheckIfDescriptionColumnsExists()
 	public class MyDbContext : DbContext
     {
         public DbSet<Foo> Foos { get; set; }
-	}
 
-	/// <summary>
-	/// Example entity that references an enum
-	/// </summary>
-	public class Foo
+        protected override void OnModelCreating(DbModelBuilder modelBuilder)
+        {            
+            base.OnModelCreating(modelBuilder);
+        }
+    }
+
+    public class MyCustomSchemaDbContext : DbContext
+    {
+        public DbSet<Foo> Foos { get; set; }
+
+        protected override void OnModelCreating(DbModelBuilder modelBuilder)
+        {
+            modelBuilder.HasDefaultSchema("myschema");
+            base.OnModelCreating(modelBuilder);
+        }
+    }
+
+    /// <summary>
+    /// Example entity that references an enum
+    /// </summary>
+    public class Foo
 	{
 		public int Id { get; set; }
 		public Size Size { get; set; }