Test for the PrimaryKeyStatement.

Author: msallin

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

@@ -66,13 +66,13 @@
   </Choose>
   <ItemGroup>
     <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="Statement\PrimaryKeyStatementTest.cs" />
     <Compile Include="Statement\StatementTestBase.cs" />
   </ItemGroup>
   <ItemGroup>
     <Folder Include="Builder\" />
     <Folder Include="Convention\" />
     <Folder Include="Extensions\" />
-    <Folder Include="Statement\" />
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\SQLite.CodeFirst\SQLite.CodeFirst.csproj">

SQLite.CodeFirst.Test/Statement/PrimaryKeyStatementTest.cs

@@ -0,0 +1,32 @@
+using System;
+using System.Collections.Generic;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using SQLite.CodeFirst.Statement;
+
+namespace SQLite.CodeFirst.Test.Statement
+{
+    [TestClass]
+    public class PrimaryKeyStatementTest : StatementTestBase
+    {
+        [TestMethod]
+        public void CreateStatementWithOneKeyTest()
+        {
+            const string keyMember1 = "keyMember1";
+
+            var primaryKeyStatement = new PrimaryKeyStatement(new List<string> { keyMember1 });
+            Assert.AreEqual(primaryKeyStatement.Count, 1);
+            Assert.AreEqual(primaryKeyStatement.CreateStatement(), "PRIMARY KEY(keyMember1)");
+        }
+
+        [TestMethod]
+        public void CreateStatementWithTwoKeyTest()
+        {
+            const string keyMember1 = "keyMember1";
+            const string keyMember2 = "keyMember2";
+
+            var primaryKeyStatement = new PrimaryKeyStatement(new List<string> { keyMember1, keyMember2 });
+            Assert.AreEqual(primaryKeyStatement.Count, 2);
+            Assert.AreEqual(primaryKeyStatement.CreateStatement(), "PRIMARY KEY(keyMember1, keyMember2)");
+        }
+    }
+}