Test for ColumnStatementCollection.

Author: msallin

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

@@ -66,6 +66,7 @@
   </Choose>
   <ItemGroup>
     <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="Statement\ColumnStatementCollectionTest.cs" />
     <Compile Include="Statement\PrimaryKeyStatementTest.cs" />
     <Compile Include="Statement\CreateDatabaseStatementTest.cs" />
     <Compile Include="Statement\StatementTestBase.cs" />

SQLite.CodeFirst.Test/Statement/ColumnStatementCollectionTest.cs

@@ -0,0 +1,33 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using SQLite.CodeFirst.Statement;
+
+namespace SQLite.CodeFirst.Test.Statement
+{
+    [TestClass]
+    public class ColumnStatementCollectionTest : StatementTestBase
+    {
+        [TestMethod]
+        public void CreateStatementOneEntryTest()
+        {
+            var columnStatementCollection = new ColumnStatementCollection(new[] { CreateStatementMock("dummy1").Object });
+
+            string output = columnStatementCollection.CreateStatement();
+            Assert.AreEqual(columnStatementCollection.Count, 1);
+            Assert.AreEqual(output, "dummy1");
+        }
+
+        [TestMethod]
+        public void CreateStatementTwoEntryTest()
+        {
+            var createIndexStatementCollection = new ColumnStatementCollection(new[]
+            {
+                CreateStatementMock("dummy1").Object,
+                CreateStatementMock("dummy2").Object
+            });
+
+            string output = createIndexStatementCollection.CreateStatement();
+            Assert.AreEqual(createIndexStatementCollection.Count, 2);
+            Assert.AreEqual(output, "dummy1, dummy2");
+        }
+    }
+}