From a785af9617cc4143bb8f9e36dd9548007011674f Mon Sep 17 00:00:00 2001
From: vlad <vlad@driver.xyz>
Date: Wed, 9 Aug 2017 22:42:34 -0700
Subject: Method to insert test data from SQL file

---
 .../scala/xyz/driver/core/database/database.scala  | 24 ++++++++++++++++++++++
 1 file changed, 24 insertions(+)

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 {
-- 
cgit v1.2.3