src/factories/createMockPool.ts

@@ -8,6 +8,7 @@ import {
 } from '../types';
 import { createUid } from '../utilities/createUid';
 import { createClientConfiguration } from './createClientConfiguration';
+import { type Pool as PgPool, type PoolClient as PgClientPool } from 'pg';
 
 export const createMockPool = (
   overrides: MockPoolOverrides,
@@ -30,13 +31,12 @@ export const createMockPool = (
         on: () => {},
         query: overrides.query,
         release: () => {},
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      } as any;
+      } as unknown as PgClientPool;
 
       return connection;
     },
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  } as any;
+    end: async () => {},
+  } as unknown as PgPool;
 
   poolStateMap.set(pool, {
     ended: false,

test/slonik/factories/createMockPool.ts

@@ -4,6 +4,20 @@ import { createMockQueryResult } from '../../../src/factories/createMockQueryRes
 import test from 'ava';
 import * as sinon from 'sinon';
 
+test('ends pool', async (t) => {
+  const pool = createMockPool({
+    query: async () => {
+      return createMockQueryResult([]);
+    },
+  });
+
+  t.false(pool.getPoolState().ended);
+
+  await pool.end();
+
+  t.true(pool.getPoolState().ended);
+});
+
 test('executes a mock query (pool.query)', async (t) => {
   t.plan(4);