graphile-test 4.12.1 → 4.13.0

package/context.js

@@ -170,8 +170,27 @@ const runGraphQLInContext = async ({ input, conn, schema, resolvedPreset, pgServ
     // instead of getting a new connection from the pool
     const withPgClientKey = pgService.withPgClientKey ?? 'withPgClient';
     contextValue[withPgClientKey] = async (_pgSettings, callback) => {
-        // Simply use the test client - it's already in a transaction
-        // The pgSettings have already been applied above via setContextOnClient
+        // Augment the client with withTransaction if it doesn't already have it.
+        // In production, @dataplan/pg provides this method. In tests the raw pg
+        // Client lacks it, so we implement it via savepoints (which nest cleanly
+        // inside the test harness's outer transaction).
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const client = pgClient;
+        if (typeof client.withTransaction !== 'function') {
+            client.withTransaction = async (cb) => {
+                const sp = `wt_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
+                await client.query(`SAVEPOINT ${sp}`);
+                try {
+                    const result = await cb(client);
+                    await client.query(`RELEASE SAVEPOINT ${sp}`);
+                    return result;
+                }
+                catch (err) {
+                    await client.query(`ROLLBACK TO SAVEPOINT ${sp}`);
+                    throw err;
+                }
+            };
+        }
         return callback(pgClient);
     };
     // Check if we're in a transaction by looking at the test client's transaction state

package/esm/context.js

@@ -166,8 +166,27 @@ export const runGraphQLInContext = async ({ input, conn, schema, resolvedPreset,
     // instead of getting a new connection from the pool
     const withPgClientKey = pgService.withPgClientKey ?? 'withPgClient';
     contextValue[withPgClientKey] = async (_pgSettings, callback) => {
-        // Simply use the test client - it's already in a transaction
-        // The pgSettings have already been applied above via setContextOnClient
+        // Augment the client with withTransaction if it doesn't already have it.
+        // In production, @dataplan/pg provides this method. In tests the raw pg
+        // Client lacks it, so we implement it via savepoints (which nest cleanly
+        // inside the test harness's outer transaction).
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const client = pgClient;
+        if (typeof client.withTransaction !== 'function') {
+            client.withTransaction = async (cb) => {
+                const sp = `wt_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
+                await client.query(`SAVEPOINT ${sp}`);
+                try {
+                    const result = await cb(client);
+                    await client.query(`RELEASE SAVEPOINT ${sp}`);
+                    return result;
+                }
+                catch (err) {
+                    await client.query(`ROLLBACK TO SAVEPOINT ${sp}`);
+                    throw err;
+                }
+            };
+        }
         return callback(pgClient);
     };
     // Check if we're in a transaction by looking at the test client's transaction state

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphile-test",
-  "version": "4.12.1",
+  "version": "4.13.0",
   "author": "Constructive <developers@constructive.io>",
   "description": "PostGraphile Testing",
   "main": "index.js",
@@ -55,5 +55,5 @@
     "pgpm",
     "test"
   ],
-  "gitHead": "0fcb26c8f3379de5c2bf486e7ad78bb61a76e497"
+  "gitHead": "90016935b53d6fb84e0b83879377f0c2eb9abce6"
 }