From b177499d168936da9e64a964beb22f6057880703 Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Thu, 24 Dec 2015 04:00:01 +0000
Subject: PostgreSQL files prefixed with pq-

---
 libpqpp/pq-mock.cpp | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 libpqpp/pq-mock.cpp

(limited to 'libpqpp/pq-mock.cpp')

diff --git a/libpqpp/pq-mock.cpp b/libpqpp/pq-mock.cpp
new file mode 100644
index 0000000..db7eafb
--- /dev/null
+++ b/libpqpp/pq-mock.cpp
@@ -0,0 +1,33 @@
+#include "pq-mock.h"
+#include "pq-connection.h"
+#include <buffer.h>
+
+namespace PQ {
+
+Mock::Mock(const std::string & masterdb, const std::string & name, const std::vector<boost::filesystem::path> & ss) :
+	MockServerDatabase(masterdb, name, "postgresql")
+{
+	CreateNewDatabase();
+	PlaySchemaScripts(ss);
+}
+
+DB::Connection *
+Mock::openConnection() const
+{
+	return new Connection(stringbf("user=postgres dbname=%s", testDbName));
+}
+
+Mock::~Mock()
+{
+	DropDatabase();
+}
+
+void
+Mock::DropDatabase() const
+{
+	master->execute("SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = '" + testDbName + "'");
+	MockServerDatabase::DropDatabase();
+}
+
+}
+
-- 
cgit v1.2.3