fix: add support for transaction isolation level commands, closes #55

felixmosh · Oct 7, 2024 · 74123a2 · 74123a2
1 parent b0a9f95
commit 74123a2
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 1 deletion.
diff --git a/src/Tracker.ts b/src/Tracker.ts
@@ -25,7 +25,7 @@ type ResponseTypes = {
   simulateErrorOnce: (error: Handler['error']) => Tracker;
 };
 
-type QueryMethodType = typeof queryMethods[number];
+type QueryMethodType = (typeof queryMethods)[number];
 
 type History = Record<QueryMethodType, RawQuery[]> & {
   transactions: TransactionState[];

diff --git a/src/constants.ts b/src/constants.ts
@@ -6,4 +6,6 @@ export const transactionCommands = [
   'ROLLBACK',
   'SAVEPOINT',
   'RELEASE SAVEPOINT',
+  'SET TRANSACTION ISOLATION LEVEL',
+  'BEGIN TRANSACTION ISOLATION LEVEL',
 ] as const;
diff --git a/tests/dialects.spec.ts b/tests/dialects.spec.ts
@@ -70,6 +70,19 @@ describe('specific dialect', () => {
       expect(tracker.history.insert[0].sql).toContain('returning');
       expect(tracker.history.delete).toHaveLength(1);
     });
+
+    it('should support transactions with an explicit isolation level', async () => {
+      tracker.on.insert('table_name').responseOnce(1);
+      tracker.on.delete('table_name').responseOnce(1);
+      tracker.on.select('foo').responseOnce([]);
+
+      const trx = await db.transaction({ isolationLevel: 'read uncommitted' });
+
+      await db('table_name').transacting(trx).insert({ name: 'Steve' }).transacting(trx);
+      await db('table_name').transacting(trx).delete().where({ name: 'Steve' }).transacting(trx);
+
+      await trx.commit();
+    });
   });
 
   describe('mysql', () => {

diff --git a/tests/transaction.spec.ts b/tests/transaction.spec.ts
@@ -233,4 +233,17 @@ describe('transaction', () => {
       },
     ]);
   });
+
+  it('should support transactions with an explicit isolation level', async () => {
+    tracker.on.insert('table_name').responseOnce(1);
+    tracker.on.delete('table_name').responseOnce(1);
+    tracker.on.select('foo').responseOnce([]);
+
+    const trx = await db.transaction({ isolationLevel: 'serializable' });
+
+    await db('table_name').transacting(trx).insert({ name: 'Steve' }).transacting(trx);
+    await db('table_name').transacting(trx).delete().where({ name: 'Steve' }).transacting(trx);
+
+    await trx.commit();
+  });
 });