Register mocked database instances with the plugin manager and wrap up opening connections to them by name

2 files changed, 21 insertions, 0 deletions

diff --git a/libdbpp/mockDatabase.cpp b/libdbpp/mockDatabase.cpp
index 3ef8728..b611c70 100644
--- a/libdbpp/mockDatabase.cpp
+++ b/libdbpp/mockDatabase.cpp
@@ -2,6 +2,18 @@
 #include <buffer.h>
 #include <fstream>
 #include <modifycommand.h>
+#include <plugins.impl.h>
+
+namespace AdHoc {
+	template <>
+	PluginOf<DB::MockDatabase>::~PluginOf()
+	{
+		// This implementation doesn't delete .implementation as
+		// mock databases simply unregister themselves (via destructor)
+		// when the mock framework tears them down.
+	}
+}
+INSTANIATEPLUGINOF(DB::MockDatabase);
 
 namespace DB {
 
@@ -10,10 +22,18 @@ unsigned int MockDatabase::mocked = 0;
 MockDatabase::MockDatabase(const std::string & name) :
 	mockName(name)
 {
+	AdHoc::PluginManager::getDefault()->add(AdHoc::PluginPtr(new AdHoc::PluginOf<MockDatabase>(this, mockName, __FILE__, __LINE__)));
 }
 
 MockDatabase::~MockDatabase()
 {
+	AdHoc::PluginManager::getDefault()->remove<MockDatabase>(mockName);
+}
+
+Connection *
+MockDatabase::openConnectionTo(const std::string & mockName)
+{
+	return AdHoc::PluginManager::getDefault()->get<DB::MockDatabase>(mockName)->implementation()->openConnection();
 }
 
 void
diff --git a/libdbpp/mockDatabase.h b/libdbpp/mockDatabase.h
index a1d935b..3e26201 100644
--- a/libdbpp/mockDatabase.h
+++ b/libdbpp/mockDatabase.h
@@ -15,6 +15,7 @@ class DLL_PUBLIC MockDatabase {
 		virtual ~MockDatabase();
 
 		virtual DB::Connection * openConnection() const = 0;
+		static Connection * openConnectionTo(const std::string &);
 
 	protected:
 		virtual void CreateNewDatabase() const = 0;