one more fix

electrovir · Sep 12, 2024 · 92d4c2e · 92d4c2e
1 parent d0ff4a2
commit 92d4c2e
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/packages/node/src/prisma/model-data.test.ts b/packages/node/src/prisma/model-data.test.ts
@@ -34,19 +34,21 @@ describe(
             data: IsAny<PrismaClient> extends true ? any : PrismaAddDataData<PrismaClient>,
         ) {
             const prismaClient = await setupPrismaClient();
+            try {
+                await prisma.client.addData(prismaClient, data);
 
-            await prisma.client.addData(prismaClient, data);
-
-            const dumpedData = await prisma.client.dumpData(prismaClient, {
-                omitFields: [
-                    'createdAt',
-                    'updatedAt',
-                    'id',
-                ],
-            });
-            await prismaClient.$disconnect();
+                const dumpedData = await prisma.client.dumpData(prismaClient, {
+                    omitFields: [
+                        'createdAt',
+                        'updatedAt',
+                        'id',
+                    ],
+                });
 
-            return dumpedData;
+                return dumpedData;
+            } finally {
+                await prismaClient.$disconnect();
+            }
         }
 
         it('includes all fields by default', async () => {