fix: test failing to exit

Author: MHassanTariq

src/tests/faucet-btc-tests.ts

@@ -10,7 +10,7 @@ import {
 } from '../btc-faucet';
 import { ApiServer, startApiServer } from '../api/init';
 import { ChainID } from '@stacks/transactions';
-import { cycleMigrations, PgDataStore } from '../datastore/postgres-store';
+import { cycleMigrations, PgDataStore, runMigrations } from '../datastore/postgres-store';
 
 async function getBalanceWithWalletImport(address: string): Promise<number> {
   const client = getRpcClient();
@@ -102,11 +102,11 @@ describe('btc faucet', () => {
 
   describe('faucet http API', () => {
     let apiServer: ApiServer;
-
+    let db: PgDataStore;
     beforeAll(async () => {
       process.env.PG_DATABASE = 'postgres';
       await cycleMigrations();
-      const db = await PgDataStore.connect();
+      db = await PgDataStore.connect();
       apiServer = await startApiServer({
         datastore: db,
         chainId: ChainID.Testnet,
@@ -144,6 +144,8 @@ describe('btc faucet', () => {
 
     afterAll(async () => {
       await apiServer.terminate();
+      await db?.close();
+      await runMigrations(undefined, 'down');
     });
   });
 });

src/tests/setup.ts

@@ -28,7 +28,7 @@ export default async (): Promise<void> => {
     },
     httpLogLevel: 'silly',
   });
-  Object.assign(global, { server: server });
+  Object.assign(global, { server: server, db: db });
   console.log('Waiting for RPC connection to core node..');
   await new StacksCoreRpcClient().waitForConnection(60000);
   console.log('Jest - setup done');

src/tests/teardown.ts

@@ -1,14 +1,16 @@
 import * as isCI from 'is-ci';
+import { PgDataStore } from '../datastore/postgres-store';
 
 // ts-unused-exports:disable-next-line
 export default async (): Promise<void> => {
   console.log('Jest - teardown..');
   const eventSocketServer: import('net').Server = (global as any).server;
-  await new Promise<void>(resolve => {
-    eventSocketServer.close(() => {
-      console.log('Jest - teardown done');
-      resolve();
-    });
+  const database: PgDataStore = (global as any).db;
+  await new Promise<void>(async resolve => {
+    eventSocketServer.close();
+    await database.close();
+    console.log('Jest - teardown done');
+    resolve();
   });
 
   // If running in CI setup the "why am I still running?" log to detect stuck Jest tests