ODBC mocking and tests from Project2

1 files changed, 32 insertions, 0 deletions

diff --git a/libodbcpp/mock.cpp b/libodbcpp/mock.cpp
new file mode 100644
index 0000000..58d5afd
--- /dev/null
+++ b/libodbcpp/mock.cpp
@@ -0,0 +1,32 @@
+#include "connection.h"
+#include "mock.h"
+#include <buffer.h>
+
+namespace ODBC {
+
+Mock::Mock(const std::string & masterdb, const std::string & name, const std::vector<boost::filesystem::path> & ss) :
+	MockServerDatabase(masterdb, name, "odbc")
+{
+	CreateNewDatabase();
+	PlaySchemaScripts(ss);
+}
+
+DB::Connection *
+Mock::openConnection() const
+{
+	return new Connection(stringbf("Driver=postgresql;Database=%s;uid=postgres;servername=/run/postgresql", testDbName));
+}
+
+Mock::~Mock()
+{
+	DropDatabase();
+}
+
+void Mock::DropDatabase() const
+{
+	master->execute("SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = '" + testDbName + "'");
+	MockServerDatabase::DropDatabase();
+}
+
+}
+