prisma-mock 0.0.25 → 0.0.26

package/lib/index.js

@@ -24,31 +24,29 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
         return name.substr(0, 1).toLowerCase() + name.substr(1);
     };
     const shallowCompare = (a, b) => {
-        for (let key in a) {
+        for (let key in b) {
             if (a[key] !== b[key])
                 return false;
         }
         return true;
     };
-    const checkIds = (model, data) => {
+    const removeMultiFieldIds = (model, data) => {
         var _a, _b;
         const c = getCamelCase(model.name);
         const idFields = model.idFields || ((_a = model.primaryKey) === null || _a === void 0 ? void 0 : _a.fields);
-        const checkId = (ids) => {
+        const removeId = (ids) => {
             const id = ids.join('_');
             data = Object.assign(Object.assign({}, data), { [c]: data[c].map(item => {
                     const _a = item, _b = id, idVal = _a[_b], rest = __rest(_a, [typeof _b === "symbol" ? _b : _b + ""]);
-                    return Object.assign(Object.assign({ [id]: ids.reduce((prev, field) => {
-                            return Object.assign(Object.assign({}, prev), { [field]: item[field] });
-                        }, {}) }, item), idVal);
+                    return Object.assign(Object.assign({}, rest), idVal);
                 }) });
         };
         if ((idFields === null || idFields === void 0 ? void 0 : idFields.length) > 1) {
-            checkId(idFields);
+            removeId(idFields);
         }
         if (((_b = model.uniqueFields) === null || _b === void 0 ? void 0 : _b.length) > 0) {
             for (const uniqueField of model.uniqueFields) {
-                checkId(uniqueField);
+                removeId(uniqueField);
             }
         }
         return data;
@@ -162,6 +160,8 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
                                     delegate.update({ data: c.update, where: getFieldRelationshipWhere(item, field) });
                                 }
                             }
+                            const _e = d, _f = field.name, _update = _e[_f], rest = __rest(_e, [typeof _f === "symbol" ? _f : _f + ""]);
+                            d = rest; //?
                         }
                     }
                     if (c.increment) {
@@ -275,13 +275,13 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
                     const idFields = model.idFields || ((_a = model.primaryKey) === null || _a === void 0 ? void 0 : _a.fields);
                     if ((idFields === null || idFields === void 0 ? void 0 : idFields.length) > 1) {
                         if (child === idFields.join('_')) {
-                            return shallowCompare(val, filter);
+                            return shallowCompare(item, filter);
                         }
                     }
                     if (((_b = model.uniqueFields) === null || _b === void 0 ? void 0 : _b.length) > 0) {
                         for (const uniqueField of model.uniqueFields) {
                             if (child === uniqueField.join('_')) {
-                                return shallowCompare(val, filter);
+                                return shallowCompare(item, filter);
                             }
                         }
                     }
@@ -402,12 +402,13 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
                 return e;
             });
             data = Object.assign(Object.assign({}, data), { [prop]: newItems });
+            data = removeMultiFieldIds(model, data);
             return data;
         };
         const create = args => {
             const d = nestedUpdate(args, true, null);
             data = Object.assign(Object.assign({}, data), { [prop]: [...data[prop], d] });
-            data = checkIds(model, data);
+            data = removeMultiFieldIds(model, data);
             // TODO: multi field ids
             return findOne(Object.assign({ where: { id: d.id } }, args));
         };
@@ -484,8 +485,18 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
             return newItem;
         };
         const update = (args) => {
-            updateMany(args);
-            return findOne(args);
+            let updatedItem;
+            const newItems = data[prop].map(e => {
+                if (matchFnc(args.where)(e)) {
+                    let data = nestedUpdate(args, false, e);
+                    updatedItem = Object.assign(Object.assign({}, e), data);
+                    return updatedItem;
+                }
+                return e;
+            });
+            data = Object.assign(Object.assign({}, data), { [prop]: newItems });
+            data = removeMultiFieldIds(model, data);
+            return findOne(Object.assign(Object.assign({}, args), { where: updatedItem }));
         };
         return {
             findOne,
@@ -529,7 +540,7 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
         if (!data[c]) {
             data = Object.assign(Object.assign({}, (data || {})), { [c]: [] });
         }
-        data = checkIds(model, data);
+        data = removeMultiFieldIds(model, data);
         const objs = Delegate(c, model);
         Object.keys(objs).forEach(fncName => {
             client[c][fncName].mockImplementation(async (...params) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prisma-mock",
-  "version": "0.0.25",
+  "version": "0.0.26",
   "description": "Mock prisma for unit testing database",
   "main": "lib/index.js",
   "license": "MIT",