more tests

Author: ymc9

tests/integration/tests/enhancements/with-policy/unique-as-id.test.ts

@@ -171,4 +171,127 @@ describe('With Policy: unique as id', () => {
             })
         ).toResolveTruthy();
     });
+
+    it('unique fields with to-many nested update', async () => {
+        const { enhance } = await loadSchema(
+            `
+        model A {
+            id Int @id @default(autoincrement())
+            x Int
+            y Int
+            value Int
+            bs B[]
+            @@unique([x, y])
+
+            @@allow('read,create', true)
+            @@allow('update,delete', value > 0)
+        }
+
+        model B {
+            id Int @id @default(autoincrement())
+            value Int
+            a A @relation(fields: [aId], references: [id])
+            aId Int
+
+            @@allow('all', value > 0)
+        }
+        `,
+            { logPrismaQuery: true }
+        );
+
+        const db = enhance();
+
+        await db.a.create({
+            data: { x: 1, y: 1, value: 1, bs: { create: [{ id: 1, value: 1 }] } },
+        });
+
+        await db.a.create({
+            data: { x: 2, y: 2, value: 2, bs: { create: [{ id: 2, value: 2 }] } },
+        });
+
+        await db.a.update({
+            where: { x_y: { x: 1, y: 1 } },
+            data: { bs: { updateMany: { data: { value: 3 } } } },
+        });
+
+        // check b#1 is updated
+        await expect(db.b.findUnique({ where: { id: 1 } })).resolves.toMatchObject({ value: 3 });
+
+        // check b#2 is not affected
+        await expect(db.b.findUnique({ where: { id: 2 } })).resolves.toMatchObject({ value: 2 });
+
+        await db.a.update({
+            where: { x_y: { x: 1, y: 1 } },
+            data: { bs: { deleteMany: {} } },
+        });
+
+        // check b#1 is deleted
+        await expect(db.b.findUnique({ where: { id: 1 } })).resolves.toBeNull();
+
+        // check b#2 is not affected
+        await expect(db.b.findUnique({ where: { id: 2 } })).resolves.toMatchObject({ value: 2 });
+    });
+
+    it('unique fields with to-one nested update', async () => {
+        const { enhance } = await loadSchema(
+            `
+        model A {
+            id Int @id @default(autoincrement())
+            x Int
+            y Int
+            value Int
+            b B?
+            @@unique([x, y])
+
+            @@allow('read,create', true)
+            @@allow('update,delete', value > 0)
+        }
+
+        model B {
+            id Int @id @default(autoincrement())
+            value Int
+            a A @relation(fields: [aId], references: [id])
+            aId Int @unique
+
+            @@allow('all', value > 0)
+        }
+        `
+        );
+
+        const db = enhance();
+
+        await db.a.create({
+            data: { x: 1, y: 1, value: 1, b: { create: { id: 1, value: 1 } } },
+        });
+
+        await db.a.create({
+            data: { x: 2, y: 2, value: 2, b: { create: { id: 2, value: 2 } } },
+        });
+
+        await db.a.update({
+            where: { x_y: { x: 1, y: 1 } },
+            data: { b: { update: { data: { value: 3 } } } },
+        });
+
+        // check b#1 is updated
+        await expect(db.b.findUnique({ where: { id: 1 } })).resolves.toMatchObject({ value: 3 });
+
+        // check b#2 is not affected
+        await expect(db.b.findUnique({ where: { id: 2 } })).resolves.toMatchObject({ value: 2 });
+
+        await db.a.update({
+            where: { x_y: { x: 1, y: 1 } },
+            data: { b: { delete: true } },
+        });
+
+        // check b#1 is deleted
+        await expect(db.b.findUnique({ where: { id: 1 } })).resolves.toBeNull();
+        await expect(db.a.findUnique({ where: { x_y: { x: 1, y: 1 } }, include: { b: true } })).resolves.toMatchObject({
+            b: null,
+        });
+
+        // check b#2 is not affected
+        await expect(db.b.findUnique({ where: { id: 2 } })).resolves.toMatchObject({ value: 2 });
+        await expect(db.a.findUnique({ where: { x_y: { x: 2, y: 2 } }, include: { b: true } })).resolves.toBeTruthy();
+    });
 });