prisma-mock 0.5.3 → 0.5.5

package/README.md

@@ -205,7 +205,7 @@ TODO (path, string_contains, string_starts_with, string_ends_with, array_contain
 ## Prisma Client methods
 
 - $transaction
-- TODO: $transaction (interactive)
+- $transaction (interactive)
 - TODO: $transaction (isolation)
 
 # Contribution

package/lib/index.js

@@ -29,6 +29,7 @@ const jest_mock_extended_1 = require("jest-mock-extended");
 const defaults_1 = __importStar(require("./defaults"));
 const shallowCompare_1 = require("./utils/shallowCompare");
 const deepEqual_1 = require("./utils/deepEqual");
+const deepCopy_1 = require("./utils/deepCopy");
 function IsFieldDefault(f) {
     return f.name !== undefined;
 }
@@ -135,10 +136,21 @@ const createPrismaMock = (data = {}, datamodel = client_1.Prisma.dmmf.datamodel,
     };
     client["$transaction"].mockImplementation(async (actions) => {
         const res = [];
-        for (const action of actions) {
-            res.push(await action);
+        if (Array.isArray(actions)) {
+            for (const action of actions) {
+                res.push(await action);
+            }
+            return res;
+        }
+        else {
+            const snapshot = (0, deepCopy_1.deepCopy)(data);
+            try {
+                await actions(client);
+            }
+            catch {
+                data = snapshot;
+            }
         }
-        return res;
     });
     client["$connect"].mockImplementation(async () => { });
     client["$disconnect"].mockImplementation(async () => { });

package/lib/utils/deepCopy.d.ts

@@ -0,0 +1 @@
+export declare function deepCopy<T>(source: T): T;

package/lib/utils/deepCopy.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.deepCopy = void 0;
+function deepCopy(source) {
+    if (Object(source) !== source) {
+        // primitive
+        return source;
+    }
+    else if (Array.isArray(source)) {
+        return source.map(deepCopy);
+    }
+    return Object.fromEntries(Object.entries(source).map(([k, v]) => ([k, deepCopy(v)])));
+}
+exports.deepCopy = deepCopy;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prisma-mock",
-  "version": "0.5.3",
+  "version": "0.5.5",
   "description": "Mock prisma for unit testing database",
   "main": "lib/index.js",
   "repository": "https://github.com/demonsters/prisma-mock",