prisma-mock 1.1.0-alpha.1 → 1.1.0-alpha.3

package/README.md

@@ -14,9 +14,30 @@ yarn add prisma-mock --dev
 
 ## Usage
 
-### Basic Example
+### Prisma 7
 
-Simple example of how to create a prisma mock instance:
+With Prisma 7, you typically generate your client to a custom output directory (such as `./generated/client`), and import the generated `Prisma` namespace from your generated files. Use the `prisma-mock/client` entry point and pass the `Prisma` namespace when
+initializing your mock client.
+
+> **Note:** You will also need to configure and use the [DMMF Generator](#dmmf-generator) to produce the necessary metadata for mocking.
+
+```js
+import createPrismaMock from "prisma-mock/client"
+import { Prisma } from "./generated/client"
+import * as dmmf from "./generated/dmmf"
+
+let client
+
+beforeEach(() => {
+  client = createPrismaMock(Prisma, {
+    datamodel: dmmf,
+  })
+})
+```
+
+### Prisma 6 and below
+
+Simple example of how to create a prisma mock instance using the default (node_module) export:
 
 ```js
 import createPrismaMock from "prisma-mock"
@@ -97,7 +118,11 @@ createPrismaMock<P extends PrismaClient = PrismaClient>(
     caseInsensitive?: boolean
     enableIndexes?: boolean
   }
-): P & { $getInternalState: () => Required<PrismaMockData<P>> }
+): P & { 
+  $getInternalState: () => Required<PrismaMockData<P>>
+  $setInternalState: (state: Required<PrismaMockData<P>>) => void
+  $clear: () => void
+}
 ```
 
 ### Parameters
@@ -117,6 +142,8 @@ createPrismaMock<P extends PrismaClient = PrismaClient>(
 Returns a mock Prisma client with all standard model methods plus:
 
 - `$getInternalState()`: Method to access the internal data state for testing/debugging
+- `$setInternalState(state)`: Method to replace the entire internal data state. Useful for resetting state to a specific snapshot or setting up complex test scenarios. The state is deep copied to avoid reference issues.
+- `$clear()`: Method to reset the internal state back to the initial state
 
 ### Client Export (`prisma-mock/client`)
 
@@ -130,7 +157,11 @@ createPrismaMock<PClient extends PrismaClient, P extends typeof Prisma = typeof
     caseInsensitive?: boolean
     enableIndexes?: boolean
   }
-): PClient & { $getInternalState: () => Required<PrismaMockData<PClient>> }
+): PClient & { 
+  $getInternalState: () => Required<PrismaMockData<PClient>>
+  $setInternalState: (state: Required<PrismaMockData<PClient>>) => void
+  $clear: () => void
+}
 ```
 
 #### Parameters
@@ -284,6 +315,7 @@ const client = createPrismaMock(Prisma, {
 ### Attribute Functions ✅
 
 - `autoincrement()`
+- `auto()` (MongoDB ObjectId)
 - `cuid()`
 - `uuid()`
 - `now()`
@@ -331,10 +363,6 @@ The following features are planned but not yet implemented:
 - `isEmpty`
 - `equals`
 
-### Attributes
-
-- `auto()`
-
 ### Referential Actions
 
 - `onDelete: Restrict`

package/lib/client.d.ts

@@ -2,6 +2,7 @@ import type { Prisma, PrismaClient } from "@prisma/client";
 import { MockPrismaOptions, PrismaMockData } from "./types";
 declare function createPrismaMock<PClient extends PrismaClient, P extends typeof Prisma = typeof Prisma>(prisma: P, options?: MockPrismaOptions<PClient, P>): PClient & {
     $getInternalState: () => Required<PrismaMockData<PClient>>;
+    $setInternalState: (state: Required<PrismaMockData<PClient>>) => void;
     $clear: () => void;
 };
 export default createPrismaMock;

package/lib/client.js

@@ -22,9 +22,10 @@ function createPrismaMock(prisma, options = {
     enableIndexes: true,
     data: {}
 }) {
+    let internalState = options.data ? (0, deepCopy_1.deepCopy)(options.data) : {};
     // Reference object to hold the mock data state
     let ref = {
-        data: options.data ? (0, deepCopy_1.deepCopy)(options.data) : {},
+        data: internalState,
     };
     // Initialize the mock client (either use provided one or create new)
     let client = options.mockClient ? options.mockClient : {};
@@ -126,8 +127,12 @@ function createPrismaMock(prisma, options = {
     });
     // Add method to access internal state for testing/debugging
     client['$getInternalState'] = () => ref.data;
+    client['$setInternalState'] = (state) => {
+        internalState = (0, deepCopy_1.deepCopy)(state);
+        ref.data = internalState;
+    };
     client['$clear'] = () => {
-        ref.data = options.data ? (0, deepCopy_1.deepCopy)(options.data) : {};
+        ref.data = internalState;
     };
     // @ts-ignore
     return client;

package/lib/defaults/index.js

@@ -6,11 +6,13 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const autoincrement_1 = __importDefault(require("./autoincrement"));
 const cuid_1 = __importDefault(require("./cuid"));
 const now_1 = __importDefault(require("./now"));
+const objectId_1 = __importDefault(require("./objectId"));
 const uuid_1 = __importDefault(require("./uuid"));
 function createHandleDefault() {
     // const registry = new Map<string, (string, Prisma.DMMF.Field, PrismaMockData) => any>();
     const registry = new Map();
     registry.set("autoincrement", (0, autoincrement_1.default)());
+    registry.set("auto", (0, objectId_1.default)());
     registry.set("cuid", (0, cuid_1.default)());
     registry.set("uuid", (0, uuid_1.default)());
     registry.set("now", now_1.default);

package/lib/defaults/objectId.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Generates MongoDB ObjectId-like strings (24 hex chars).
+ * Used for @default(auto()) with @db.ObjectId
+ */
+declare const createObjectId: () => () => string;
+export default createObjectId;

package/lib/defaults/objectId.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+/**
+ * Generates MongoDB ObjectId-like strings (24 hex chars).
+ * Used for @default(auto()) with @db.ObjectId
+ */
+const createObjectId = () => {
+    let counter = 0;
+    return () => {
+        const hex = "0123456789abcdef";
+        let result = "";
+        for (let i = 0; i < 22; i++) {
+            result += hex[Math.floor(Math.random() * 16)];
+        }
+        result += (++counter).toString(16).padStart(2, "0");
+        return result;
+    };
+};
+exports.default = createObjectId;

package/lib/delegate.js

@@ -133,7 +133,7 @@ const createDelegate = ({ ref, prisma, datamodel = prisma.dmmf.datamodel, caseIn
                     if (isCreating && (field.isUnique || field.isId)) {
                         const existing = findOne({ where: { [field.name]: inputFieldData } });
                         if (existing) {
-                            (0, errors_1.throwKnownError)(prisma, `Unique constraint failed on the fields: (\`${field.name}\`)`, { code: 'P2002', meta: { target: [field.name] } });
+                            (0, errors_1.throwKnownError)(prisma, `Unique constraint failed on the fields: (\`${field.name}\`)`, { code: 'P2002', meta: { modelName: model.name, target: [field.name] } });
                         }
                     }
                     // Handle relation fields (object type)
@@ -509,7 +509,9 @@ const createDelegate = ({ ref, prisma, datamodel = prisma.dmmf.datamodel, caseIn
         const findOrThrow = (args) => {
             const found = findOne(args);
             if (!found) {
-                (0, errors_1.throwKnownError)(prisma, `No ${prop.slice(0, 1).toUpperCase()}${prop.slice(1)} found`);
+                (0, errors_1.throwKnownError)(prisma, `No ${prop.slice(0, 1).toUpperCase()}${prop.slice(1)} found`, {
+                    meta: { cause: "No record was found for a query.", modelName: model.name },
+                });
             }
             return found;
         };
@@ -824,7 +826,7 @@ const createDelegate = ({ ref, prisma, datamodel = prisma.dmmf.datamodel, caseIn
             if (!hasMatch) {
                 if (args.skipForeignKeysChecks)
                     return;
-                (0, errors_1.throwKnownError)(prisma, "An operation failed because it depends on one or more records that were required but not found. Record to update not found.", { meta: { cause: "Record to update not found." } });
+                (0, errors_1.throwKnownError)(prisma, "An operation failed because it depends on one or more records that were required but not found. Record to update not found.", { meta: { cause: "No record was found for an update.", modelName: model.name } });
             }
             ref.data = {
                 ...ref.data,
@@ -1063,7 +1065,7 @@ const createDelegate = ({ ref, prisma, datamodel = prisma.dmmf.datamodel, caseIn
             delete: (args) => {
                 const item = findOne(args);
                 if (!item) {
-                    (0, errors_1.throwKnownError)(prisma, "An operation failed because it depends on one or more records that were required but not found. Record to delete does not exist.", { meta: { cause: "Record to delete does not exist." } });
+                    (0, errors_1.throwKnownError)(prisma, "An operation failed because it depends on one or more records that were required but not found. Record to delete does not exist.", { meta: { cause: "No record was found for a delete.", modelName: model.name } });
                 }
                 const deleted = deleteMany(args);
                 if (deleted.length) {

package/lib/index.d.ts

@@ -5,6 +5,7 @@ type MockPrismaOptions<P extends PrismaClient = PrismaClient> = Omit<MockPrismaO
 };
 export default function createPrismaClient<P extends PrismaClient = PrismaClient>(options?: MockPrismaOptions<P>): P & {
     $getInternalState: () => Required<Partial<{ [key in import("./types").IsTable<Uncapitalize<import("./types").IsString<keyof P>>>]: import("./types").PrismaList<P, key>; }>>;
+    $setInternalState: (state: Required<Partial<{ [key in import("./types").IsTable<Uncapitalize<import("./types").IsString<keyof P>>>]: import("./types").PrismaList<P, key>; }>>) => void;
     $clear: () => void;
 };
 export {};

package/lib/legacy.d.ts

@@ -73,5 +73,74 @@ export default function createPrismaClient(data?: PrismaMockData<PrismaClient>,
             id: string;
         }>[];
     }>>;
+    $setInternalState: (state: Required<Partial<{
+        account: Partial<{
+            name: string;
+            id: number;
+            sort: number;
+        }>[];
+        user: Partial<{
+            name: string;
+            id: number;
+            sort: number;
+            accountId: number;
+            role: import(".prisma/client").$Enums.Role;
+            clicks: number;
+            deleted: boolean;
+            uniqueField: string;
+            age: number;
+        }>[];
+        answers: Partial<{
+            id: number;
+            title: string;
+        }>[];
+        element: Partial<{
+            value: string;
+            userId: number;
+            e_id: number;
+            json: Prisma.JsonValue;
+        }>[];
+        stripe: Partial<{
+            id: number;
+            sort: number;
+            accountId: number;
+            customerId: string;
+            active: boolean;
+        }>[];
+        userAnswers: Partial<{
+            value: string;
+            answerId: number;
+            userId: number;
+        }>[];
+        document: Partial<{
+            name: string;
+            id: string;
+        }>[];
+        post: Partial<{
+            id: number;
+            title: string;
+            published: boolean;
+            authorId: number;
+            updated: Date;
+            created: Date;
+        }>[];
+        pet: Partial<{
+            name: string;
+            id: number;
+            ownerId: number;
+        }>[];
+        toy: Partial<{
+            name: string;
+            id: number;
+            ownerId: number;
+        }>[];
+        transaction: Partial<{
+            id: string;
+            initiator: string;
+        }>[];
+        dbGeneratedId: Partial<{
+            id: string;
+        }>[];
+    }>>) => void;
     $clear: () => void;
 };

package/lib/utils/queryMatching.js

@@ -40,6 +40,8 @@ function createMatch({ prisma, getFieldRelationshipWhere, getDelegateForFieldNam
                 if (info?.relationName) {
                     const childName = (0, fieldHelpers_1.getCamelCase)(info.type);
                     let childWhere = {};
+                    let useIsFilter = false;
+                    let useIsNotFilter = false;
                     if (filter.every) {
                         childWhere = filter.every;
                     }
@@ -49,6 +51,14 @@ function createMatch({ prisma, getFieldRelationshipWhere, getDelegateForFieldNam
                     else if (filter.none) {
                         childWhere = filter.none;
                     }
+                    else if ("is" in filter) {
+                        useIsFilter = true;
+                        childWhere = filter.is === null ? {} : filter.is;
+                    }
+                    else if ("isNot" in filter) {
+                        useIsNotFilter = true;
+                        childWhere = filter.isNot === null ? {} : filter.isNot;
+                    }
                     else {
                         childWhere = filter;
                     }
@@ -57,10 +67,37 @@ function createMatch({ prisma, getFieldRelationshipWhere, getDelegateForFieldNam
                     });
                     const delegate = getDelegateForFieldName(childName);
                     const joinWhere = getFieldRelationshipWhere(item, info, submodel);
+                    if (useIsFilter) {
+                        if (filter.is === null) {
+                            if (!joinWhere)
+                                return true;
+                            const res = delegate.findMany({ where: joinWhere });
+                            return res.length === 0;
+                        }
+                        if (!joinWhere)
+                            return false;
+                        const res = delegate.findMany({
+                            where: { AND: [childWhere, joinWhere] },
+                        });
+                        return res.length > 0;
+                    }
+                    if (useIsNotFilter) {
+                        if (filter.isNot === null) {
+                            if (!joinWhere)
+                                return false;
+                            const res = delegate.findMany({ where: joinWhere });
+                            return res.length > 0;
+                        }
+                        if (!joinWhere)
+                            return true;
+                        const res = delegate.findMany({
+                            where: { AND: [childWhere, joinWhere] },
+                        });
+                        return res.length === 0;
+                    }
                     if (!joinWhere) {
                         return false;
                     }
-                    // return true
                     const res = delegate.findMany({
                         where: {
                             AND: [
@@ -70,17 +107,12 @@ function createMatch({ prisma, getFieldRelationshipWhere, getDelegateForFieldNam
                         }
                     });
                     if (filter.every) {
-                        // const all = data[childName].filter(
-                        //   matchFnc(getFieldRelationshipWhere(item, info)),
-                        // )
                         const where = getFieldRelationshipWhere(item, info, model);
                         if (!where)
                             return false;
                         const all = delegate.findMany({
                             where,
                         });
-                        // For "every": all related records must match the condition
-                        // If no related records exist, "every" is vacuously true
                         if (all.length === 0)
                             return true;
                         return res.length === all.length;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prisma-mock",
-  "version": "1.1.0-alpha.1",
+  "version": "1.1.0-alpha.3",
   "description": "Mock prisma for unit testing database",
   "main": "lib/index.js",
   "repository": {
@@ -45,11 +45,13 @@
     "preversion": "jest && tsc",
     "lint": "tsc --noEmit",
     "build": "tsc",
-    "test": "jest",
+    "test": "prisma generate && npm run generate:mongodb && jest",
     "generate": "prisma generate",
+    "generate:mongodb": "DATABASE_URL_MONGODB=mongodb://localhost:27017/test prisma generate --schema=prisma/schema-mongodb.prisma",
     "release": "yarn build && changeset publish && git push --follow-tags",
     "watch": "tsc --watch",
     "test:postgres": "env-cmd -e postgres jest --maxWorkers=1",
+    "test:mongodb": "npm run generate:mongodb && jest __tests__/mongodb.test.ts",
     "changeset": "changeset",
     "changeset:add": "changeset add",
     "changeset:validate:ci": "changeset status --since=origin/main --verbose"