#27: Added UniqueConstraint and tests for it.

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\UniqueConstraintTest.cs" />
     <Compile Include="Statement\ColumnStatementCollectionTest.cs" />
     <Compile Include="Statement\PrimaryKeyStatementTest.cs" />
     <Compile Include="Statement\CreateDatabaseStatementTest.cs" />

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

@@ -0,0 +1,27 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using SQLite.CodeFirst.Statement.ColumnConstraint;
+
+namespace SQLite.CodeFirst.Test.Statement.ColumnConstraint
+{
+    [TestClass]
+    public class UniqueConstraintTest
+    {
+        [TestMethod]
+        public void CreateStatement_StatementIsCorrect_NoConstraint()
+        {
+            var uniqueConstraint = new UniqueConstraint();
+            uniqueConstraint.OnConflict = OnConflictAction.None;
+            string output = uniqueConstraint.CreateStatement();
+            Assert.AreEqual(output, "UNIQUE");
+        }
+
+        [TestMethod]
+        public void CreateStatement_StatementIsCorrect_WithConstraint()
+        {
+            var uniqueConstraint = new UniqueConstraint();
+            uniqueConstraint.OnConflict = OnConflictAction.Rollback;
+            string output = uniqueConstraint.CreateStatement();
+            Assert.AreEqual(output, "UNIQUE ON CONFLICT ROLLBACK");
+        }
+    }
+}

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

@@ -0,0 +1,20 @@
+using System.Text;
+
+namespace SQLite.CodeFirst.Statement.ColumnConstraint
+{
+    internal class UniqueConstraint : IColumnConstraint
+    {
+        private const string Template = "UNIQUE {conflict-clause}";
+
+        public OnConflictAction OnConflict { get; set; }
+
+        public string CreateStatement()
+        {
+            var sb = new StringBuilder(Template);
+
+            sb.Replace("{conflict-clause}", OnConflict != OnConflictAction.None ? "ON CONFLICT " + OnConflict.ToString().ToUpperInvariant() : string.Empty);
+
+            return sb.ToString().Trim();
+        }
+    }
+}

SQLite.CodeFirst/SQLite.CodeFirst.csproj

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