From 5b1ae1f6fc02506d91276230621bf2e4b2e320ab Mon Sep 17 00:00:00 2001
From: mike12345567 <me@michaeldrury.co.uk>
Date: Fri, 20 May 2022 18:08:48 +0100
Subject: [PATCH] Fixing mock.

---
 packages/server/__mocks__/pg.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/server/__mocks__/pg.ts b/packages/server/__mocks__/pg.ts
index 44aeabcb3..110933ad5 100644
--- a/packages/server/__mocks__/pg.ts
+++ b/packages/server/__mocks__/pg.ts
@@ -14,7 +14,9 @@ module PgMock {
   function Client() {}
 
   Client.prototype.query = query
-  Client.prototype.end = jest.fn()
+  Client.prototype.end = jest.fn(cb => {
+    if (cb) cb()
+  })
   Client.prototype.connect = jest.fn()
   Client.prototype.release = jest.fn()