From 5cc2b29257155cbe904fc4549509e64099d07058 Mon Sep 17 00:00:00 2001
From: matt <matt@fiddaman.net>
Date: Thu, 20 Jan 2022 17:57:37 +0000
Subject: [PATCH] Added DatabaseConnectionPool.close()

---
 lib/DatabaseConnectionPool.js           | 7 +++++++
 lib/__mocks__/DatabaseConnectionPool.js | 7 ++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/lib/DatabaseConnectionPool.js b/lib/DatabaseConnectionPool.js
index e1003ec..f83d5a5 100644
--- a/lib/DatabaseConnectionPool.js
+++ b/lib/DatabaseConnectionPool.js
@@ -62,6 +62,13 @@ class DatabaseConnectionPool {
 			await this.#connectionPool.execute(sql, params);
 		return data;
 	}
+
+	/**
+	 * Close the connection to the database
+	 */
+	async close() {
+		await this.#connectionPool.end();
+	}
 }
 
 module.exports = DatabaseConnectionPool;
diff --git a/lib/__mocks__/DatabaseConnectionPool.js b/lib/__mocks__/DatabaseConnectionPool.js
index 3faf3d2..6d539f3 100644
--- a/lib/__mocks__/DatabaseConnectionPool.js
+++ b/lib/__mocks__/DatabaseConnectionPool.js
@@ -21,9 +21,14 @@ const mockRunQuery = jest.fn((sql, params) => {
 	};
 });
 
+const mockClose = jest.fn();
+
 jest.mock('./DatabaseConnectionPool', () => {
 	return jest.fn().mockImplementation(() => {
-		return { runQuery: mockRunQuery };
+		return {
+			runQuery: mockRunQuery,
+			close: mockClose
+		};
 	});
 });