fix: enhance delegate model interaction with afterEntityMutation plugin support in child updates

Author: genu

packages/orm/src/client/crud/operations/base.ts

@@ -1218,8 +1218,10 @@ export abstract class BaseOperationHandler<Schema extends SchemaDef> {
             );
             // only fields not consumed by base update will be used for this model
             finalData = baseUpdateResult.remainingFields;
-            // base update may change entity ids, update the filter
-            combinedWhere = baseUpdateResult.baseEntity;
+            // trim to id fields only to avoid "missing FROM-clause" errors in the child update
+            combinedWhere = baseUpdateResult.baseEntity
+                ? getIdValues(this.schema, modelDef.baseModel!, baseUpdateResult.baseEntity)
+                : baseUpdateResult.baseEntity;
 
             // update this entity with fields in updated base
             if (baseUpdateResult.baseEntity) {

tests/e2e/orm/plugin-infra/entity-mutation-hooks.test.ts

@@ -691,4 +691,45 @@ describe('Entity mutation hooks tests', () => {
             await expect(client.user.findMany()).toResolveWithLength(0);
         });
     });
+
+    describe('Delegate model interaction', () => {
+        it('update on child model succeeds with afterEntityMutation plugin', async () => {
+            const client = await createTestClient(
+                `
+model Base {
+    id Int @id @default(autoincrement())
+    orgId Int
+    type String
+    @@delegate(type)
+}
+
+model Child extends Base {
+    childField Int
+}
+`,
+            );
+
+            const withPlugin = client.$use({
+                id: 'test',
+                onEntityMutation: {
+                    afterEntityMutation() {},
+                },
+            });
+
+            const created = await withPlugin.child.create({
+                data: { orgId: 1, childField: 10 },
+            });
+
+            const updated = await withPlugin.child.update({
+                where: { id: created.id },
+                data: { orgId: 2, childField: 20 },
+            });
+
+            expect(updated).toBeTruthy();
+            expect(updated.orgId).toBe(2);
+            expect(updated.childField).toBe(20);
+
+            await client.$disconnect();
+        });
+    });
 });