feat: add support for nested transactions (#9)

src/Tracker.ts

@@ -44,7 +44,10 @@ export class Tracker {
   public _handle(rawQuery: RawQuery): Promise<any> {
     return new Promise((resolve, reject) => {
       setTimeout(async () => {
-        if (typeof rawQuery.method === 'undefined' && transactionCommands.includes(rawQuery.sql)) {
+        if (
+          typeof rawQuery.method === 'undefined' &&
+          transactionCommands.some((trxCommand) => rawQuery.sql.startsWith(trxCommand))
+        ) {
           return resolve(undefined);
         }
 

src/constants.ts

@@ -1,3 +1,9 @@
 export const queryMethods = ['select', 'insert', 'update', 'delete', 'any'] as const;
 
-export const transactionCommands = ['BEGIN;', 'COMMIT;', 'ROLLBACK'];
+export const transactionCommands = [
+  'BEGIN;',
+  'COMMIT;',
+  'ROLLBACK',
+  'SAVEPOINT',
+  'RELEASE SAVEPOINT',
+];

tests/insert.spec.ts

@@ -222,4 +222,22 @@ describe('mock Insert statement', () => {
     expect(tracker.history.insert).toHaveLength(1);
     expect(tracker.history.delete).toHaveLength(1);
   });
+
+  it('should support nested transactions', async () => {
+    tracker.on.insert('table_name').responseOnce(1);
+    tracker.on.delete('table_name').responseOnce(1);
+
+    await db.transaction(async (trx) => {
+      await db('table_name').insert({ name: faker.name.firstName() }).transacting(trx);
+      await trx.transaction(async (innerTrx) => {
+        await db('table_name')
+          .delete()
+          .where({ name: faker.name.firstName() })
+          .transacting(innerTrx);
+      });
+    });
+
+    expect(tracker.history.insert).toHaveLength(1);
+    expect(tracker.history.delete).toHaveLength(1);
+  });
 });