pqb 0.17.6 → 0.17.7

package/dist/index.d.ts

@@ -7135,8 +7135,25 @@ type Arg = {
     $queryBuilder: Query;
 } | Query;
 declare const testTransaction: {
+    /**
+     * Start a test transaction.
+     * The returned promise is resolved immediately when transaction starts, not waiting for it to end.
+     *
+     * @param arg - ORM instance or a queryable instance (such as db.someTable).
+     */
     start(arg: Arg): Promise<void>;
+    /**
+     * Rollback a test transaction.
+     *
+     * @param arg - the same ORM or query argument passed into the `testTransaction.start`.
+     */
     rollback(arg: Arg): Promise<void> | undefined;
+    /**
+     * Will roll back the current `testTransaction` (won't have any effect if it was rolled back already),
+     * and if there's no nested test transactions left, it will close the db connection.
+     *
+     * @param arg - the same ORM or query argument passed into the `testTransaction.start`.
+     */
     close(arg: Arg): Promise<void>;
 };
 

package/dist/index.js

@@ -10195,12 +10195,18 @@ class Rollback extends Error {
 const trxForTest = Symbol("trxForTest");
 const argToDb = (arg) => "$queryBuilder" in arg ? arg.$queryBuilder : arg;
 const testTransaction = {
+  /**
+   * Start a test transaction.
+   * The returned promise is resolved immediately when transaction starts, not waiting for it to end.
+   *
+   * @param arg - ORM instance or a queryable instance (such as db.someTable).
+   */
   start(arg) {
     var _a, _b;
     const db = argToDb(arg);
     const { transactionStorage } = db.internal;
     const { getStore } = transactionStorage;
-    const adapter = db.baseQuery.q;
+    const { adapter } = db.baseQuery.q;
     const data = {
       adapter: {
         query: adapter.query,
@@ -10232,17 +10238,30 @@ const testTransaction = {
       });
     });
   },
+  /**
+   * Rollback a test transaction.
+   *
+   * @param arg - the same ORM or query argument passed into the `testTransaction.start`.
+   */
   rollback(arg) {
-    var _a, _b;
+    var _a;
     const db = argToDb(arg);
-    const data = (_a = db.internal[trxForTest]) == null ? void 0 : _a.pop();
-    if (data) {
-      (_b = data.reject) == null ? void 0 : _b.call(data, new Rollback());
-      Object.assign(db.baseQuery.q.adapter, data.adapter);
-      return data.promise;
+    const data = db.internal[trxForTest];
+    const last = data == null ? void 0 : data.pop();
+    if (!last)
+      return;
+    if ((data == null ? void 0 : data.length) === 1) {
+      Object.assign(db.baseQuery.q.adapter, data[0].adapter);
     }
-    return;
+    (_a = last.reject) == null ? void 0 : _a.call(last, new Rollback());
+    return last.promise;
   },
+  /**
+   * Will roll back the current `testTransaction` (won't have any effect if it was rolled back already),
+   * and if there's no nested test transactions left, it will close the db connection.
+   *
+   * @param arg - the same ORM or query argument passed into the `testTransaction.start`.
+   */
   async close(arg) {
     var _a;
     const db = argToDb(arg);