fix: add missing events

gajus · Nov 14, 2024 · c60cc6a · c60cc6a
1 parent 6aa21ad
commit c60cc6a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/packages/slonik/src/helpers.test/createPoolWithMockedQuery.ts b/packages/slonik/src/helpers.test/createPoolWithMockedQuery.ts
@@ -1,6 +1,7 @@
 import { type ConnectionPoolClient } from '../factories/createConnectionPool';
 import { createPool } from '../factories/createPool';
 import { type ClientConfigurationInput } from '../types';
+import EventEmitter from 'node:events';
 import * as sinon from 'sinon';
 
 export const createPoolWithMockedQuery = async (
@@ -28,7 +29,12 @@ export const createPoolWithMockedQuery = async (
           // We are re-using the same connection for all queries
           // as it makes it easier to spy on the connection.
           // eslint-disable-next-line require-atomic-updates
-          connection = await driver.createClient();
+          connection = await driver.createClient().then((nextConnection) => {
+            return {
+              ...nextConnection,
+              events: new EventEmitter(),
+            };
+          });
 
           // eslint-disable-next-line require-atomic-updates
           connection.query = query;

diff --git a/packages/slonik/src/helpers.test/createPoolWithSpy.ts b/packages/slonik/src/helpers.test/createPoolWithSpy.ts
@@ -1,6 +1,7 @@
 import { type ConnectionPoolClient } from '../factories/createConnectionPool';
 import { createPool } from '../factories/createPool';
 import { type ClientConfigurationInput } from '../types';
+import EventEmitter from 'node:events';
 import * as sinon from 'sinon';
 
 export const createPoolWithSpy = async (
@@ -33,7 +34,12 @@ export const createPoolWithSpy = async (
           // We are re-using the same connection for all queries
           // as it makes it easier to spy on the connection.
           // eslint-disable-next-line require-atomic-updates
-          connection = await driver.createClient();
+          connection = await driver.createClient().then((nextConnection) => {
+            return {
+              ...nextConnection,
+              events: new EventEmitter(),
+            };
+          });
 
           spy.acquire = sinon.spy(connection, 'acquire');
           spy.destroy = sinon.spy(connection, 'destroy');