Drop tables support. Closes #55

Author: stefanmielke

DataCore.Database/Database.cs

@@ -263,7 +263,12 @@ public int CreateTablesIfNotExists(IEnumerable<Type> tables, bool createReferenc
 
         public int DropTable<T>()
         {
-            var tableDefinition = GetTableDefinition(typeof(T));
+            return DropTable(typeof(T));
+        }
+
+        public int DropTable(Type table)
+        {
+            var tableDefinition = GetTableDefinition(table);
 
             var queries = Translator.GetDropTableQuery(tableDefinition.Name);
 
@@ -275,9 +280,24 @@ public int DropTable<T>()
             return 0;
         }
 
+        public int DropTables(params Type[] tables)
+        {
+            foreach (var table in tables)
+            {
+                DropTable(table);
+            }
+
+            return 0;
+        }
+
         public int DropTableIfExists<T>()
         {
-            var tableDefinition = GetTableDefinition(typeof(T));
+            return DropTableIfExists(typeof(T));
+        }
+
+        public int DropTableIfExists(Type table)
+        {
+            var tableDefinition = GetTableDefinition(table);
 
             var queries = Translator.GetDropTableIfExistsQuery(tableDefinition.Name);
 
@@ -289,6 +309,16 @@ public int DropTableIfExists<T>()
             return 0;
         }
 
+        public int DropTablesIfExists(params Type[] tables)
+        {
+            foreach (var table in tables)
+            {
+                DropTableIfExists(table);
+            }
+
+            return 0;
+        }
+
         public int CreateColumn<T>(Expression<Func<T, dynamic>> clause)
         {
             var arguments = ExpressionHelper.GetExpressionsFromDynamic(clause);

DataCore.Database/IDatabase.cs

@@ -20,8 +20,13 @@ public interface IDatabase
         int CreateTableIfNotExists(Type table, bool createReferences = false);
         int CreateTablesIfNotExists(params Type[] tables);
         int CreateTablesIfNotExists(IEnumerable<Type> tables, bool createReferences = false);
+
         int DropTable<T>();
+        int DropTable(Type table);
+        int DropTables(params Type[] tables);
         int DropTableIfExists<T>();
+        int DropTableIfExists(Type table);
+        int DropTablesIfExists(params Type[] tables);
 
         int CreateColumn<T>(Expression<Func<T, dynamic>> clause);
         int CreateColumnIfNotExists<T>(Expression<Func<T, dynamic>> clause);

DataCore.Test/DatabaseCheckedTest.cs

@@ -165,6 +165,24 @@ public void CanDropTable(TestHelper.DatabaseType dbType, string connectionString
             }
         }
 
+        [Test, TestCaseSource(typeof(SqlTestDataFactory), nameof(SqlTestDataFactory.TestCases))]
+        public void CanDropTables(TestHelper.DatabaseType dbType, string connectionString)
+        {
+            using (var connection = TestHelper.GetConnectionFor(dbType, connectionString))
+            {
+                var database = TestHelper.GetDatabaseFor(dbType, connection);
+
+                database.CreateTablesIfNotExists(typeof(TestClass), typeof(TestClass2));
+
+                database.Select(database.From<TestClass>().Where(t => t.Id == 1));
+                database.Select(database.From<TestClass2>().Where(t => t.Id == 1));
+
+                database.DropTablesIfExists(typeof(TestClass), typeof(TestClass2));
+
+                connection.Close();
+            }
+        }
+
         [Test, TestCaseSource(typeof(SqlTestDataFactory), nameof(SqlTestDataFactory.TestCases))]
         public void CanCreateColumn(TestHelper.DatabaseType dbType, string connectionString)
         {

DataCore.Test/DatabaseTest.cs

@@ -147,6 +147,24 @@ public void CanDropTable(TestHelper.DatabaseType dbType, string connectionString
             }
         }
 
+        [Test, TestCaseSource(typeof(SqlTestDataFactory), nameof(SqlTestDataFactory.TestCases))]
+        public void CanDropTables(TestHelper.DatabaseType dbType, string connectionString)
+        {
+            using (var connection = TestHelper.GetConnectionFor(dbType, connectionString))
+            {
+                var database = TestHelper.GetDatabaseFor(dbType, connection);
+
+                database.CreateTables(typeof(TestClass), typeof(TestClass2));
+
+                database.Select(database.From<TestClass>().Where(t => t.Id == 1));
+                database.Select(database.From<TestClass2>().Where(t => t.Id == 1));
+
+                database.DropTables(typeof(TestClass), typeof(TestClass2));
+
+                connection.Close();
+            }
+        }
+
         [Test, TestCaseSource(typeof(SqlTestDataFactory), nameof(SqlTestDataFactory.TestCases))]
         public void CanCreateColumn(TestHelper.DatabaseType dbType, string connectionString)
         {