summaryrefslogtreecommitdiff
path: root/libpqpp/mock.cpp
diff options
context:
space:
mode:
authorDan Goodliffe <dan@randomdan.homeip.net>2015-09-24 03:02:07 +0100
committerDan Goodliffe <dan@randomdan.homeip.net>2015-09-24 03:06:06 +0100
commitf8317be5c2b2f6d31622d863a41646fa456afbbe (patch)
tree6124ba5d953fea59f5b8e6132261003f258e49cf /libpqpp/mock.cpp
parentConnector compatibility fix (diff)
downloadlibdbpp-postgresql-f8317be5c2b2f6d31622d863a41646fa456afbbe.tar.bz2
libdbpp-postgresql-f8317be5c2b2f6d31622d863a41646fa456afbbe.tar.xz
libdbpp-postgresql-f8317be5c2b2f6d31622d863a41646fa456afbbe.zip
PostgreSQL mocking and tests from Project2
Diffstat (limited to 'libpqpp/mock.cpp')
-rw-r--r--libpqpp/mock.cpp33
1 files changed, 33 insertions, 0 deletions
diff --git a/libpqpp/mock.cpp b/libpqpp/mock.cpp
new file mode 100644
index 0000000..20d4340
--- /dev/null
+++ b/libpqpp/mock.cpp
@@ -0,0 +1,33 @@
+#include "mock.h"
+#include "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();
+}
+
+}
+