Method to insert test data from SQL file

1 files changed, 24 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 8426309..2f20989 100644
--- a/src/main/scala/xyz/driver/core/database/database.scala
+++ b/src/main/scala/xyz/driver/core/database/database.scala
@@ -6,6 +6,7 @@ import xyz.driver.core.date.Date
 import xyz.driver.core.time.Time
 
 import scala.concurrent.Future
+import scala.concurrent.ExecutionContext
 
 package database {
 
@@ -136,6 +137,29 @@ package database {
     def dropSchema(): Unit = {
       Await.result(slickDal.execute(tables.schema.drop >> tables.dropNamespaceSchema), Duration.Inf)
     }
+
+    def insertTestData(database: xyz.driver.core.database.Database, filePath: String)(
+            implicit executionContext: ExecutionContext) = {
+
+      import database.profile.api.{DBIO => _, _}
+
+      val file    = java.nio.file.Paths.get(filePath)
+      val sqlLine = new String(java.nio.file.Files.readAllBytes(file), "UTF-8")
+
+      val createInsertProcedure =
+        sqlu"""CREATE PROCEDURE INSERT_TEST_DATA()
+                MODIFIES SQL DATA
+               BEGIN ATOMIC
+               #$sqlLine
+               END;
+             """
+
+      val callInserts = sqlu"""{call INSERT_TEST_DATA()}"""
+
+      val dropInsertProcedure = sqlu"""drop PROCEDURE INSERT_TEST_DATA;"""
+
+      Future.sequence(Seq(createInsertProcedure, callInserts, dropInsertProcedure).map(slickDal.execute))
+    }
   }
 
   trait DatabaseObject extends ColumnTypes {