From 5542442aa10367b76ba6717698baeeee580ec272 Mon Sep 17 00:00:00 2001
From: Gustavo Edinger <gbe@nect.com>
Date: Thu, 9 Jan 2025 14:51:55 +0100
Subject: [PATCH] fix: reset reference to get and set data

---
 src/__tests__/client/reset.test.ts | 3 +++
 src/lib/client.ts                  | 5 ++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/__tests__/client/reset.test.ts b/src/__tests__/client/reset.test.ts
index c1c46d9e..e020b340 100644
--- a/src/__tests__/client/reset.test.ts
+++ b/src/__tests__/client/reset.test.ts
@@ -30,5 +30,8 @@ describe('client (reset)', () => {
 
     const usersAfterReset = await userService.findMany();
     expect(usersAfterReset.length > 0).not.toBeTruthy();
+
+    const getDataUsers = prismock.getData().user;
+    expect(getDataUsers.length > 0).not.toBeTruthy();
   });
 });
diff --git a/src/lib/client.ts b/src/lib/client.ts
index c99c9856..cd4eceb6 100755
--- a/src/lib/client.ts
+++ b/src/lib/client.ts
@@ -66,10 +66,13 @@ export function createPrismock(instance: PrismaModule) {
     private generate() {
       const { delegates, setData, getData } = generateDelegates({ models: instance.dmmf.datamodel.models as DMMF.Model[] });
 
-      Object.entries({ ...delegates, setData, getData }).forEach(([key, value]) => {
+      Object.entries({ ...delegates }).forEach(([key, value]) => {
         if (key in this) Object.assign((this as unknown as Delegates)[key], value);
         else Object.assign(this, { [key]: value });
       });
+
+      (this as unknown as typeof PrismaClient & PrismockData).getData = getData;
+      (this as unknown as typeof PrismaClient & PrismockData).setData = setData;
     }
 
     async $connect() {