fix(datasource-sql): fix worker crash on authentication testing (#736)

packages/datasource-sql/src/connection/connection-tester.ts

@@ -0,0 +1,18 @@
+import { Sequelize } from 'sequelize';
+
+import { DatabaseConnectError } from './errors';
+
+export default async function testConnectionWithTimeOut(
+  sequelize: Sequelize,
+  databaseUri: string,
+  timeout = 10000,
+) {
+  // Test connection. If this doesn't resolve after 10s, throw a timeout error
+  const timeoutPromise = new Promise(resolve => {
+    setTimeout(resolve, timeout);
+  }).then(() =>
+    Promise.reject(new DatabaseConnectError('Connection to database timed out', databaseUri)),
+  );
+
+  await Promise.race([sequelize.authenticate(), timeoutPromise]);
+}

packages/datasource-sql/src/connection/index.ts

@@ -1,6 +1,7 @@
 import { Sequelize } from 'sequelize';
 
 import ConnectionOptions from './connection-options';
+import testConnectionWithTimeOut from './connection-tester';
 import handleErrors from './handle-errors';
 import ReverseProxy from './services/reverse-proxy';
 import SequelizeFactory from './services/sequelize-factory';
@@ -27,7 +28,8 @@ export default async function connect(options: ConnectionOptions): Promise<Seque
     if (tcpServer) sequelizeFactory.onClose(tcpServer.closeListener.bind(tcpServer));
 
     sequelize = sequelizeFactory.build(await options.buildSequelizeCtorOptions());
-    await sequelize.authenticate(); // Test connection
+
+    await testConnectionWithTimeOut(sequelize, options.debugDatabaseUri);
 
     return sequelize;
   } catch (e) {

packages/datasource-sql/test/connection/connection-tester.test.ts

@@ -0,0 +1,40 @@
+import { Sequelize } from 'sequelize';
+
+import testConnectionWithTimeOut from '../../src/connection/connection-tester';
+import { DatabaseConnectError } from '../../src/connection/errors';
+
+describe('testConnectionWithTimeOut', () => {
+  it('should time out', async () => {
+    jest.useFakeTimers();
+
+    const sequelize = {
+      authenticate: () =>
+        new Promise(resolve => {
+          setTimeout(resolve, 4000);
+        }),
+    };
+
+    const promise = testConnectionWithTimeOut(sequelize as unknown as Sequelize, 'a db URI', 3000);
+
+    jest.advanceTimersByTime(3000);
+
+    await expect(promise).rejects.toThrow(DatabaseConnectError);
+  });
+
+  it('should pass before 3s', async () => {
+    jest.useFakeTimers();
+
+    const sequelize = {
+      authenticate: () =>
+        new Promise(resolve => {
+          setTimeout(resolve, 2000);
+        }),
+    };
+
+    const promise = testConnectionWithTimeOut(sequelize as unknown as Sequelize, 'a db URI', 3000);
+
+    jest.advanceTimersByTime(2000);
+
+    await expect(promise).resolves.not.toThrow();
+  });
+});