Test for the ForeignKeyStatement.

Author: msallin

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

@@ -69,6 +69,7 @@
     <Compile Include="Statement\PrimaryKeyStatementTest.cs" />
     <Compile Include="Statement\CreateDatabaseStatementTest.cs" />
     <Compile Include="Statement\StatementTestBase.cs" />
+    <Compile Include="Statement\ForeignKeyStatementTest.cs" />
   </ItemGroup>
   <ItemGroup>
     <Folder Include="Builder\" />

SQLite.CodeFirst.Test/Statement/ForeignKeyStatementTest.cs

@@ -0,0 +1,85 @@
+using System.Collections.Generic;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using SQLite.CodeFirst.Statement;
+
+namespace SQLite.CodeFirst.Test.Statement
+{
+    [TestClass]
+    public class ForeignKeyStatementTest
+    {
+        [TestMethod]
+        public void CreateStatementOneForeignKeyTest()
+        {
+            var foreignKeyStatement = new ForeignKeyStatement
+            {
+                CascadeDelete = false,
+                ForeignKey = new List<string> { "dummyForeignKey1" },
+                ForeignPrimaryKey = new List<string> { "dummForeignPrimaryKey1" },
+                ForeignTable = "dummyForeignTable"
+            };
+
+            string output = foreignKeyStatement.CreateStatement();
+            Assert.AreEqual(output, "FOREIGN KEY (dummyForeignKey1) REFERENCES dummyForeignTable(dummForeignPrimaryKey1)");
+        }
+
+        [TestMethod]
+        public void CreateStatementOneForeignKeyCascadeDeleteTest()
+        {
+            var foreignKeyStatement = new ForeignKeyStatement
+            {
+                CascadeDelete = true,
+                ForeignKey = new List<string> { "dummyForeignKey1" },
+                ForeignPrimaryKey = new List<string> { "dummForeignPrimaryKey1" },
+                ForeignTable = "dummyForeignTable"
+            };
+
+            string output = foreignKeyStatement.CreateStatement();
+            Assert.AreEqual(output, "FOREIGN KEY (dummyForeignKey1) REFERENCES dummyForeignTable(dummForeignPrimaryKey1) ON DELETE CASCADE");
+        }
+
+        [TestMethod]
+        public void CreateStatementTwoForeignKeyTest()
+        {
+            var foreignKeyStatement = new ForeignKeyStatement
+            {
+                CascadeDelete = false,
+                ForeignKey = new List<string> { "dummyForeignKey1", "dummyForeignKey2" },
+                ForeignPrimaryKey = new List<string> { "dummForeignPrimaryKey1" },
+                ForeignTable = "dummyForeignTable"
+            };
+
+            string output = foreignKeyStatement.CreateStatement();
+            Assert.AreEqual(output, "FOREIGN KEY (dummyForeignKey1, dummyForeignKey2) REFERENCES dummyForeignTable(dummForeignPrimaryKey1)");
+        }
+
+        [TestMethod]
+        public void CreateStatementTwoForeignKeyTwoPrimaryKeyTest()
+        {
+            var foreignKeyStatement = new ForeignKeyStatement
+            {
+                CascadeDelete = false,
+                ForeignKey = new List<string> { "dummyForeignKey1", "dummyForeignKey2" },
+                ForeignPrimaryKey = new List<string> { "dummForeignPrimaryKey1", "dummForeignPrimaryKey2" },
+                ForeignTable = "dummyForeignTable"
+            };
+
+            string output = foreignKeyStatement.CreateStatement();
+            Assert.AreEqual(output, "FOREIGN KEY (dummyForeignKey1, dummyForeignKey2) REFERENCES dummyForeignTable(dummForeignPrimaryKey1, dummForeignPrimaryKey2)");
+        }
+
+        [TestMethod]
+        public void CreateStatementOneForeignKeyTwoPrimaryKeyTest()
+        {
+            var foreignKeyStatement = new ForeignKeyStatement
+            {
+                CascadeDelete = false,
+                ForeignKey = new List<string> { "dummyForeignKey1" },
+                ForeignPrimaryKey = new List<string> { "dummForeignPrimaryKey1", "dummForeignPrimaryKey2" },
+                ForeignTable = "dummyForeignTable"
+            };
+
+            string output = foreignKeyStatement.CreateStatement();
+            Assert.AreEqual(output, "FOREIGN KEY (dummyForeignKey1) REFERENCES dummyForeignTable(dummForeignPrimaryKey1, dummForeignPrimaryKey2)");
+        }
+    }
+}