forcedRollback for tests

src/index.test.ts

@@ -35,6 +35,17 @@ describe(`tx`, () => {
 		expect(committed.thing).toEqual('comitted')
 	})
 
+	it(`doesn't commit changes with forcedRollback`, async () => {
+		await tx(pg, async (db) => {
+			db.query(`INSERT INTO things (thing) VALUES ('comitted')`)
+		})
+
+		const { rowCount } = await pg.query(
+			`SELECT id, thing FROM things WHERE thing = 'node_error'`
+		)
+		expect(rowCount).toBe(0)
+	})
+
 	it(`doesn't commit changes when there's a Node exception`, async () => {
 		await expect(
 			tx(pg, async (db) => {

src/index.ts

@@ -1,21 +1,28 @@
 import { Pool, PoolClient } from 'pg'
 import { ProxyClient } from './proxy_client'
 
+/**
+ * @param pg node postgres pool
+ * @param callback callback that will use provided transaction client
+ * @param forceRollback force rollback even without errors - useful for integration tests
+ * @returns
+ */
 export default async function tx<T>(
 	pg: Pool,
-	callback: (db: PoolClient) => Promise<T>
+	callback: (db: PoolClient) => Promise<T>,
+	forceRollback?: boolean
 ): Promise<T> {
-  const client = await pg.connect()
+	const client = await pg.connect()
 	const proxyClient = new ProxyClient(client)
 	await proxyClient.query(`BEGIN`)
 
 	try {
 		const result = await callback(proxyClient)
-    proxyClient.proxyRelease()
-		await client.query(`COMMIT`)
+		proxyClient.proxyRelease()
+		await client.query(forceRollback ? `ROLLBACK` : `COMMIT`)
 		return result
 	} catch (e) {
-    proxyClient.proxyRelease()
+		proxyClient.proxyRelease()
 		await client.query(`ROLLBACK`)
 		throw e
 	} finally {