1 files changed, 11 insertions, 0 deletions
diff --git a/test/testdb-postgresql.cpp b/test/testdb-postgresql.cpp
index 55f163f..c59640a 100644
--- a/test/testdb-postgresql.cpp
+++ b/test/testdb-postgresql.cpp
@@ -1,6 +1,7 @@
 #include "testdb-postgresql.h"
 #include <compileTimeFormatter.h>
 #include <cstdlib>
+#include <fstream>
 #include <helpers.h>
 #include <output/pq/pqConn.h>
 
@@ -15,6 +16,16 @@ namespace MyGrate {
 			query(("CREATE DATABASE " + mockname).c_str());
 		}
 
+		PqConnDB::PqConnDB(const std::string & schemaFile) : PqConnDB()
+		{
+			auto mdb = mock();
+
+			std::stringstream buffer;
+			buffer << std::ifstream(schemaFile).rdbuf();
+
+			mdb.query(buffer.str().c_str());
+		}
+
 		PqConnDB::~PqConnDB()
 		{
 			query(("DROP DATABASE IF EXISTS " + mockname).c_str());