1 files changed, 57 insertions, 0 deletions
diff --git a/libdbpp/unittests/testConnection.cpp b/libdbpp/unittests/testConnection.cpp
new file mode 100644
index 0000000..3472dde
--- /dev/null
+++ b/libdbpp/unittests/testConnection.cpp
@@ -0,0 +1,57 @@
+#define BOOST_TEST_MODULE DbConnection
+#include <boost/test/unit_test.hpp>
+
+#include <factory.h>
+#include <connection.h>
+
+class MockDb : public DB::Connection {
+	public:
+		MockDb(const std::string &) {}
+
+		void	finish() const {}
+		int		beginTx() const { return 0; }
+		int		commitTx() const { return 0; }
+		int		rollbackTx() const { return 0; }
+		bool	inTx() const { return false; }
+		void	savepoint(const std::string &) const {}
+		void	rollbackToSavepoint(const std::string &) const {}
+		void	releaseSavepoint(const std::string &) const {}
+		void	ping() const {}
+		DB::BulkDeleteStyle bulkDeleteStyle() const { return DB::BulkDeleteUsingUsing; }
+		DB::BulkUpdateStyle bulkUpdateStyle() const { return DB::BulkUpdateUsingJoin; }
+
+		void execute(const std::string &) const {}
+		DB::SelectCommand * newSelectCommand(const std::string &) const { return nullptr; }
+		DB::ModifyCommand * newModifyCommand(const std::string &) const { return nullptr; }
+
+		void beginBulkUpload(const char *, const char *) const {}
+		void endBulkUpload(const char *) const {}
+		size_t bulkUploadData(const char *, size_t) const {return 0;}
+};
+
+FACTORY(MockDb, DB::ConnectionFactory);
+
+BOOST_AUTO_TEST_CASE( plugins )
+{
+	auto pm = AdHoc::PluginManager::getDefault();
+	BOOST_REQUIRE(pm);
+	BOOST_REQUIRE_EQUAL(1, pm->count());
+	BOOST_REQUIRE_EQUAL(1, pm->getAll().size());
+	BOOST_REQUIRE_EQUAL(1, pm->getAll<DB::ConnectionFactory>().size());
+}
+
+BOOST_AUTO_TEST_CASE( create )
+{
+	auto mock = DB::ConnectionFactory::create("MockDb", "doesn't matter");
+	BOOST_REQUIRE(mock);
+	// MockDb is fake, just returns nullptr, but the call should otherwise succeed.
+	BOOST_REQUIRE(!mock->newModifyCommand(""));
+	BOOST_REQUIRE(!mock->newSelectCommand(""));
+	delete mock;
+}
+
+BOOST_AUTO_TEST_CASE( resolve )
+{
+	BOOST_REQUIRE_THROW(DB::ConnectionFactory::create("otherdb", "doesn't matter"), AdHoc::LoadLibraryException);
+}
+