add fixture trait for use in database component tests

1 files changed, 19 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/core/database/database.scala b/src/main/scala/xyz/driver/core/database/database.scala
index 4c65385..8d61899 100644
--- a/src/main/scala/xyz/driver/core/database/database.scala
+++ b/src/main/scala/xyz/driver/core/database/database.scala
@@ -131,6 +131,25 @@ package database {
     def dropNamespaceSchema: StreamingDBIO[Vector[Unit], Unit]
   }
 
+  trait CreateAndDropSchema {
+    val slickDal: xyz.driver.core.database.SlickDal
+    val tables: GeneratedTables
+
+    import tables.profile.api._
+    import scala.concurrent.Await
+    import scala.concurrent.duration.Duration
+
+    // Note: Does not call `super.beforeEach()` (for org.scalatest.BeforeAndAfterEach)
+    def beforeEach(): Unit = {
+      Await.result(slickDal.execute(tables.createNamespaceSchema >> tables.schema.create), Duration.Inf)
+    }
+
+    // Note: Does not call `super.afterEach()` (for org.scalatest.BeforeAndAfterEach)
+    def afterEach(): Unit = {
+      Await.result(slickDal.execute(tables.schema.drop >> tables.dropNamespaceSchema), Duration.Inf)
+    }
+  }
+
   trait DatabaseObject extends ColumnTypes {
     def createTables(): Future[Unit]
     def disconnect(): Unit