From be7a8cfb8ccc4d58013bfb03dbdedbcd62c8eaf4 Mon Sep 17 00:00:00 2001
From: Martin McKeaveney <martin@shogunsystems.co.uk>
Date: Mon, 24 May 2021 12:14:04 +0100
Subject: [PATCH] fix tests

---
 packages/server/__mocks__/mssql.js                     | 10 +++++++++-
 .../src/integrations/tests/microsoftSqlServer.spec.js  |  3 ++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/packages/server/__mocks__/mssql.js b/packages/server/__mocks__/mssql.js
index 6119c014da..3809f0a4c6 100644
--- a/packages/server/__mocks__/mssql.js
+++ b/packages/server/__mocks__/mssql.js
@@ -9,6 +9,14 @@ mssql.query = jest.fn(() => ({
   ],
 }))
 
-mssql.connect = jest.fn(() => ({ recordset: [] }))
+// mssql.connect = jest.fn(() => ({ recordset: [] }))
+
+mssql.ConnectionPool = jest.fn(() => ({
+  connect: jest.fn(() => ({
+    request: jest.fn(() => ({
+      query: jest.fn(() => ({})),
+    })),
+  })),
+}))
 
 module.exports = mssql
diff --git a/packages/server/src/integrations/tests/microsoftSqlServer.spec.js b/packages/server/src/integrations/tests/microsoftSqlServer.spec.js
index 29399b840f..5d7d59a647 100644
--- a/packages/server/src/integrations/tests/microsoftSqlServer.spec.js
+++ b/packages/server/src/integrations/tests/microsoftSqlServer.spec.js
@@ -32,7 +32,8 @@ describe("MS SQL Server Integration", () => {
   })
 
   describe("no rows returned", () => {
-    beforeEach(() => {
+    beforeEach(async () => {
+      await config.integration.connect()
       config.integration.client.query.mockImplementation(() => ({ rows: [] }))
     })