Test for CreateIndexStatementCollection and some refactorings.

Author: msallin

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

@@ -71,6 +71,7 @@
     <Compile Include="Statement\StatementTestBase.cs" />
     <Compile Include="Statement\ForeignKeyStatementTest.cs" />
     <Compile Include="Statement\CreateTableStatementTest.cs" />
+    <Compile Include="Statement\CreateIndexStatementCollectionTest.cs" />
   </ItemGroup>
   <ItemGroup>
     <Folder Include="Builder\" />

SQLite.CodeFirst.Test/Statement/CreateIndexStatementCollectionTest.cs

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

SQLite.CodeFirst.Test/Statement/CreateTableStatementTest.cs

@@ -1,5 +1,4 @@
 using Microsoft.VisualStudio.TestTools.UnitTesting;
-using Moq;
 using SQLite.CodeFirst.Statement;
 
 namespace SQLite.CodeFirst.Test.Statement
@@ -10,13 +9,10 @@ public class CreateTableStatementTest : StatementTestBase
         [TestMethod]
         public void CreateStatementTest()
         {
-            var statementCollectionMock = new Mock<IStatementCollection>();
-            statementCollectionMock.Setup(s => s.CreateStatement()).Returns("dummyColumnDefinition");
-
             var createTableStatement = new CreateTableStatement
             {
                 TableName = "dummyTable",
-                ColumnStatementCollection = statementCollectionMock.Object
+                ColumnStatementCollection = CreateStatementCollectionMock("dummyColumnDefinition").Object
             };
 
             string output = createTableStatement.CreateStatement();

SQLite.CodeFirst.Test/Statement/ForeignKeyStatementTest.cs

@@ -5,7 +5,7 @@
 namespace SQLite.CodeFirst.Test.Statement
 {
     [TestClass]
-    public class ForeignKeyStatementTest
+    public class ForeignKeyStatementTest : StatementTestBase
     {
         [TestMethod]
         public void CreateStatementOneForeignKeyTest()

SQLite.CodeFirst.Test/Statement/PrimaryKeyStatementTest.cs

@@ -1,5 +1,4 @@
-using System;
-using System.Collections.Generic;
+using System.Collections.Generic;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using SQLite.CodeFirst.Statement;
 

SQLite.CodeFirst.Test/Statement/StatementTestBase.cs

@@ -11,5 +11,12 @@ protected Mock<IStatement> CreateStatementMock(string createStatementReturnValue
             statementMock.Setup(s => s.CreateStatement()).Returns(createStatementReturnValue);
             return statementMock;
         }
+
+        protected Mock<IStatementCollection> CreateStatementCollectionMock(string createStatementReturnValue)
+        {
+            var statementCollectionMock = new Mock<IStatementCollection>();
+            statementCollectionMock.Setup(s => s.CreateStatement()).Returns(createStatementReturnValue);
+            return statementCollectionMock;
+        }
     }
 }

SQLite.CodeFirst/Statement/CreateIndexStatementCollection.cs

@@ -5,11 +5,11 @@
 
 namespace SQLite.CodeFirst.Statement
 {
-    internal class CreateIndexStatementCollection : Collection<CreateIndexStatement>, IStatement
+    internal class CreateIndexStatementCollection : Collection<IStatement>, IStatementCollection
     {
         private const string StatementSeperator = "\r\n";
 
-        public CreateIndexStatementCollection(IEnumerable<CreateIndexStatement> createIndexStatements)
+        public CreateIndexStatementCollection(IEnumerable<IStatement> createIndexStatements)
         {
             foreach (var createIndexStatement in createIndexStatements)
             {

SQLite.CodeFirst/Statement/IStatementCollection.cs

@@ -2,7 +2,7 @@
 
 namespace SQLite.CodeFirst.Statement
 {
-    internal interface IStatementCollection : IStatement, ICollection<IStatement>
+    public interface IStatementCollection : IStatement, ICollection<IStatement>
     {
     }
 }