From d8025ff93d66ec1ff7a6ffd90783e877e9eaafff Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Sun, 13 Jun 2021 02:42:26 +0100
Subject: Support creating a PostgreSQL mock DB and filling it with a schema
 script

---
 test/testdb-postgresql.cpp | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'test/testdb-postgresql.cpp')

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());
-- 
cgit v1.2.3