prisma-mock 0.0.12 → 0.0.13

package/lib/index.js

@@ -265,7 +265,11 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
         const findOne = args => {
             if (!data[prop])
                 throw new Error(`${prop} not found in data`);
-            return includes(args)(data[prop].find(matchFnc(args.where)));
+            const items = findMany(args);
+            if (items.length === 0) {
+                return null;
+            }
+            return items[0];
         };
         const findMany = args => {
             const res = data[prop].filter(matchFnc(args === null || args === void 0 ? void 0 : args.where)).map(includes(args));
@@ -286,14 +290,20 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
                     return shouldInclude;
                 });
             }
+            if (args === null || args === void 0 ? void 0 : args.select) {
+                return res.map(item => {
+                    const newItem = {};
+                    Object.keys(args.select).forEach(key => newItem[key] = item[key]);
+                    return newItem;
+                });
+            }
             return res;
         };
-        const findFirst = args => {
-            const item = data[prop].find(matchFnc(args === null || args === void 0 ? void 0 : args.where));
-            if (item)
-                return includes(args)(item);
-            return null;
-        };
+        // const findFirst = args => {
+        //   const item = data[prop].find(matchFnc(args?.where));
+        //   if (item) return includes(args)(item);
+        //   return null;
+        // };
         const updateMany = args => {
             // if (!Array.isArray(data[prop])) {
             //   throw new Error(`${prop} not found in data`)
@@ -318,10 +328,11 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
             data = Object.assign(Object.assign({}, data), { [prop]: data[prop].filter(e => !matchFnc(args === null || args === void 0 ? void 0 : args.where)(e)) });
         };
         const includes = args => item => {
-            if (!(args === null || args === void 0 ? void 0 : args.include) || !item)
+            if ((!(args === null || args === void 0 ? void 0 : args.include) && !(args === null || args === void 0 ? void 0 : args.select)) || !item)
                 return item;
             let newItem = item;
-            const keys = Object.keys(args.include);
+            const obj = (args === null || args === void 0 ? void 0 : args.select) || (args === null || args === void 0 ? void 0 : args.include);
+            const keys = Object.keys(obj);
             keys.forEach(key => {
                 // Get field schema for relation info
                 const model = cachedSchema.datamodel.models.find(model => {
@@ -330,10 +341,14 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
                 const schema = model.fields.find(field => {
                     return field.name === key;
                 });
+                switch (schema.type) {
+                    case "Int":
+                        return;
+                }
                 // Get delegate for relation
                 const delegate = Delegate(getCamelCase(schema.type), model);
                 // Construct arg for relation query
-                let subArgs = args.include[key] === true ? {} : args.include[key];
+                let subArgs = obj[key] === true ? {} : obj[key];
                 subArgs = Object.assign(Object.assign({}, subArgs), { where: Object.assign(Object.assign({}, subArgs.where), getFieldRelationshipWhere(item, schema)) });
                 if (schema.isList) {
                     // Add relation
@@ -349,7 +364,7 @@ const createPrismaMock = async (data = {}, pathToSchema, client = (0, jest_mock_
             findOne,
             findUnique: findOne,
             findMany,
-            findFirst,
+            findFirst: findOne,
             create,
             createMany: (args) => {
                 args.data.forEach((data) => {

package/package.json

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