From 720b701fac13f464addb2116301d9a9a2998d041 Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Wed, 27 Aug 2025 20:48:27 +0100
Subject: Add MockDBPool

Feels like this should be provided by libdbpp, but it isn't.
---
 test/test-util.hpp | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'test/test-util.hpp')

diff --git a/test/test-util.hpp b/test/test-util.hpp
index 20020f8..5b2a968 100644
--- a/test/test-util.hpp
+++ b/test/test-util.hpp
@@ -1,7 +1,8 @@
 #pragma once
 
-#include "pq-mock.h"
+#include <connectionPool.h>
 #include <filesystem>
+#include <pq-mock.h>
 
 namespace WebStat {
 #define XSTR(s) STR(s)
@@ -20,5 +21,16 @@ namespace WebStat {
 		MockDB();
 	};
 
+	class MockDBPool : public DB::BasicConnectionPool {
+	public:
+		MockDBPool(std::string);
+
+	protected:
+		DB::ConnectionPtr createResource() const;
+
+	private:
+		std::string name;
+	};
+
 	template<typename Out> using ParseData = std::tuple<std::string_view, Out>;
 }
-- 
cgit v1.2.3