#42: Added CollateConstraint and tests.

Author: Marc Sallin

SQLite.CodeFirst.Test/SQLite.CodeFirst.Test.csproj

@@ -80,6 +80,7 @@
     <Compile Include="Statement\ColumnConstraint\MaxLengthConstraint.cs" />
     <Compile Include="Statement\ColumnConstraint\ColumnConstraintCollectionTest.cs" />
     <Compile Include="Statement\ColumnConstraint\NotNullConstraintTest.cs" />
+    <Compile Include="Statement\ColumnConstraint\CollateConstraintTest.cs" />
     <Compile Include="Statement\ColumnConstraint\UniqueConstraintTest.cs" />
     <Compile Include="Statement\ColumnStatementCollectionTest.cs" />
     <Compile Include="Statement\PrimaryKeyStatementTest.cs" />

SQLite.CodeFirst.Test/Statement/ColumnConstraint/CollateConstraintTest.cs

@@ -0,0 +1,27 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using SQLite.CodeFirst.Statement.ColumnConstraint;
+
+namespace SQLite.CodeFirst.Test.Statement.ColumnConstraint
+{
+    [TestClass]
+    public class CollateConstraintTest
+    {
+        [TestMethod]
+        public void CreateStatement_StatementIsCorrect_NoConstraint()
+        {
+            var collationConstraint = new CollateConstraint();
+            collationConstraint.CollationFunction = CollationFunction.None;
+            string output = collationConstraint.CreateStatement();
+            Assert.AreEqual(output, "");
+        }
+
+        [TestMethod]
+        public void CreateStatement_StatementIsCorrect_NoCase()
+        {
+            var collationConstraint = new CollateConstraint();
+            collationConstraint.CollationFunction = CollationFunction.NoCase;
+            string output = collationConstraint.CreateStatement();
+            Assert.AreEqual(output, "COLLATE NOCASE");
+        }
+    }
+}

SQLite.CodeFirst/Internal/Statement/ColumnConstraint/CollateConstraint.cs

@@ -0,0 +1,25 @@
+using System.Text;
+
+namespace SQLite.CodeFirst.Statement.ColumnConstraint
+{
+    internal class CollateConstraint : IColumnConstraint
+    {
+        private const string Template = "COLLATE {collation-name}";
+
+        public CollationFunction CollationFunction { get; set; }
+
+        public string CreateStatement()
+        {
+            if (CollationFunction == CollationFunction.None)
+            {
+                return string.Empty;
+            }
+
+            var sb = new StringBuilder(Template);
+
+            sb.Replace("{collation-name}", CollationFunction.ToString().ToUpperInvariant());
+
+            return sb.ToString().Trim();
+        }
+    }
+}

SQLite.CodeFirst/SQLite.CodeFirst.csproj

@@ -91,6 +91,7 @@
     <Compile Include="Internal\Builder\NameCreators\IndexNameCreator.cs" />
     <Compile Include="Internal\Builder\NameCreators\SpecialChars.cs" />
     <Compile Include="IDatabaseCreator.cs" />
+    <Compile Include="Internal\Statement\ColumnConstraint\CollateConstraint.cs" />
     <Compile Include="Internal\Statement\ColumnConstraint\UniqueConstraint.cs" />
     <Compile Include="Internal\Utility\HashCreator.cs" />
     <Compile Include="Internal\Utility\HistoryEntityTypeValidator.cs" />