Update partner details mutation to user schema

GTBitsOfGood · Feb 12, 2025 · 091623b · 091623b
1 parent 50e1431
commit 091623b
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 20 deletions.
diff --git a/src/app/api/partnerDetails/[userId]/route.test.ts b/src/app/api/partnerDetails/[userId]/route.test.ts
@@ -215,22 +215,21 @@ describe("POST /api/partnerDetails/[userId]", () => {
       expires: "",
     });
 
-    const existingPartnerDetails = {
-      userId: 1,
-      id: 1,
-      numberOfPatients: 8,
-      organizationType: "FOR_PROFIT" as OrganizationType,
-    };
-    dbMock.partnerDetails.findUnique.mockResolvedValueOnce(
-      existingPartnerDetails
-    );
-
     const updatedPartnerDetails = {
-      ...existingPartnerDetails,
       numberOfPatients: 5,
       organizationType: "NON_PROFIT" as OrganizationType,
     };
-    dbMock.partnerDetails.update.mockResolvedValueOnce(updatedPartnerDetails);
+
+    const updatedUser = {
+      id: 1,
+      email: "test_email",
+      name: "test_name",
+      passwordHash: "test_hash",
+      type: UserType.SUPER_ADMIN,
+      partnerDetails: updatedPartnerDetails
+    }
+
+    dbMock.user.update.mockResolvedValueOnce(updatedUser);
 
     const formData = new FormData();
     formData.append("numberOfPatients", "5");

diff --git a/src/app/api/partnerDetails/[userId]/route.ts b/src/app/api/partnerDetails/[userId]/route.ts
@@ -96,17 +96,14 @@ export async function POST(
     return argumentError("Invalid form data");
   }
 
-  const { numberOfPatients, organizationType } = parsedData.data;
-
   // update PartnerDetails record
   const userIdNumber = Number(userId); //db schema accepts a number
-  const updatedPartnerDetails = await db.partnerDetails.update({
-    where: { userId: userIdNumber },
+  const updatedUser = await db.user.update({
+    where: { id: userIdNumber },
     data: {
-      numberOfPatients,
-      organizationType,
-    },
+      partnerDetails: parsedData.data
+    }
   });
 
-  return NextResponse.json(updatedPartnerDetails);
+  return NextResponse.json(updatedUser.partnerDetails);
 }