test: Create MixxxDbTest Fixture base class

1 files changed, 24 insertions, 0 deletions

diff --git a/src/test/mixxxdbtest.h b/src/test/mixxxdbtest.h
new file mode 100644
index 0000000000..e47e5dde13
--- /dev/null
+++ b/src/test/mixxxdbtest.h
@@ -0,0 +1,24 @@
+#include "database/mixxxdb.h"
+#include "test/mixxxtest.h"
+#include "util/db/dbconnectionpooled.h"
+#include "util/db/dbconnectionpooler.h"
+
+class MixxxDbTest : public MixxxTest {
+  protected:
+    MixxxDbTest(bool inMemoryDbConnection = false)
+            : m_mixxxDb(config(), inMemoryDbConnection),
+              m_dbConnectionPooler(m_mixxxDb.connectionPool()) {
+    }
+
+    const mixxx::DbConnectionPoolPtr& dbConnectionPooler() const {
+        return m_dbConnectionPooler;
+    }
+
+    QSqlDatabase dbConnection() const {
+        return mixxx::DbConnectionPooled(m_dbConnectionPooler);
+    }
+
+  private:
+    const MixxxDb m_mixxxDb;
+    const mixxx::DbConnectionPooler m_dbConnectionPooler;
+};