orchid-orm 1.3.0 → 1.3.3

package/src/relations/hasOne.test.ts

@@ -5,6 +5,7 @@ import {
   expectSql,
   profileData,
   userData,
+  useRelationCallback,
   useTestDatabase,
 } from '../test-utils/test-utils';
 import { User, Profile, Model } from '../test-utils/test-models';
@@ -414,6 +415,46 @@ describe('hasOne', () => {
             bio: 'profile 2',
           });
         });
+
+        describe('relation callbacks', () => {
+          const { beforeCreate, afterCreate, resetMocks } = useRelationCallback(
+            db.user.relations.profile,
+          );
+
+          it('should invoke callbacks', async () => {
+            await db.user.create({
+              ...userData,
+              profile: {
+                create: profileData,
+              },
+            });
+
+            expect(beforeCreate).toHaveBeenCalledTimes(1);
+            expect(afterCreate).toHaveBeenCalledTimes(1);
+          });
+
+          it('should invoke callbacks in a batch create', async () => {
+            resetMocks();
+
+            await db.user.createMany([
+              {
+                ...userData,
+                profile: {
+                  create: profileData,
+                },
+              },
+              {
+                ...userData,
+                profile: {
+                  create: profileData,
+                },
+              },
+            ]);
+
+            expect(beforeCreate).toHaveBeenCalledTimes(1);
+            expect(afterCreate).toHaveBeenCalledTimes(1);
+          });
+        });
       });
 
       describe('nested connect', () => {
@@ -502,6 +543,52 @@ describe('hasOne', () => {
             bio: 'profile 2',
           });
         });
+
+        describe('relation callbacks', () => {
+          const { beforeUpdate, afterUpdate, resetMocks } = useRelationCallback(
+            db.user.relations.profile,
+          );
+
+          it('should invoke callbacks', async () => {
+            const profileId = await db.profile.get('id').create(profileData);
+
+            await db.user.create({
+              ...userData,
+              profile: {
+                connect: { id: profileId },
+              },
+            });
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(1);
+            expect(afterUpdate).toHaveBeenCalledTimes(1);
+          });
+
+          it('should invoke callbacks in a batch create', async () => {
+            resetMocks();
+
+            const ids = await db.profile
+              .pluck('id')
+              .createMany([profileData, profileData]);
+
+            await db.user.createMany([
+              {
+                ...userData,
+                profile: {
+                  connect: { id: ids[0] },
+                },
+              },
+              {
+                ...userData,
+                profile: {
+                  connect: { id: ids[1] },
+                },
+              },
+            ]);
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(2);
+            expect(afterUpdate).toHaveBeenCalledTimes(2);
+          });
+        });
       });
 
       describe('connect or create', () => {
@@ -610,6 +697,80 @@ describe('hasOne', () => {
           });
         });
       });
+
+      describe('relation callbacks', () => {
+        const {
+          beforeUpdate,
+          afterUpdate,
+          beforeCreate,
+          afterCreate,
+          resetMocks,
+        } = useRelationCallback(db.user.relations.profile);
+
+        it('should invoke callbacks when connecting', async () => {
+          const id = await db.profile.get('id').create(profileData);
+
+          await db.user.create({
+            ...userData,
+            profile: {
+              connectOrCreate: {
+                where: { id },
+                create: profileData,
+              },
+            },
+          });
+
+          expect(beforeUpdate).toHaveBeenCalledTimes(1);
+          expect(afterUpdate).toHaveBeenCalledTimes(1);
+        });
+
+        it('should invoke callbacks when creating', async () => {
+          await db.user.create({
+            ...userData,
+            profile: {
+              connectOrCreate: {
+                where: { id: 0 },
+                create: profileData,
+              },
+            },
+          });
+
+          expect(beforeCreate).toHaveBeenCalledTimes(1);
+          expect(afterCreate).toHaveBeenCalledTimes(1);
+        });
+
+        it('should invoke callbacks in a batch create', async () => {
+          resetMocks();
+
+          const id = await db.profile.get('id').create(profileData);
+
+          await db.user.createMany([
+            {
+              ...userData,
+              profile: {
+                connectOrCreate: {
+                  where: { id: 0 },
+                  create: profileData,
+                },
+              },
+            },
+            {
+              ...userData,
+              profile: {
+                connectOrCreate: {
+                  where: { id },
+                  create: profileData,
+                },
+              },
+            },
+          ]);
+
+          expect(beforeUpdate).toHaveBeenCalledTimes(2);
+          expect(afterUpdate).toHaveBeenCalledTimes(2);
+          expect(beforeCreate).toHaveBeenCalledTimes(1);
+          expect(afterCreate).toHaveBeenCalledTimes(1);
+        });
+      });
     });
 
     describe('update', () => {
@@ -656,6 +817,52 @@ describe('hasOne', () => {
             .where({ id: { in: profileIds } });
           expect(updatedUserIds).toEqual([null, null]);
         });
+
+        describe('relation callbacks', () => {
+          const { beforeUpdate, afterUpdate, resetMocks } = useRelationCallback(
+            db.user.relations.profile,
+          );
+
+          it('should invoke callbacks', async () => {
+            const id = await db.user.get('id').create({
+              ...userData,
+              profile: { create: profileData },
+            });
+
+            await db.user.find(id).update({
+              profile: {
+                disconnect: true,
+              },
+            });
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(1);
+            expect(afterUpdate).toHaveBeenCalledTimes(1);
+          });
+
+          it('should invoke callbacks in a batch update', async () => {
+            resetMocks();
+
+            const ids = await db.user.pluck('id').createMany([
+              {
+                ...userData,
+                profile: { create: profileData },
+              },
+              {
+                ...userData,
+                profile: { create: profileData },
+              },
+            ]);
+
+            await db.user.where({ id: { in: ids } }).update({
+              profile: {
+                disconnect: true,
+              },
+            });
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(1);
+            expect(afterUpdate).toHaveBeenCalledTimes(1);
+          });
+        });
       });
 
       describe('set', () => {
@@ -689,6 +896,28 @@ describe('hasOne', () => {
 
           await expect(query).rejects.toThrow();
         });
+
+        describe('relation callbacks', () => {
+          const { beforeUpdate, afterUpdate } = useRelationCallback(
+            db.user.relations.profile,
+          );
+
+          it('should invoke callbacks', async () => {
+            const userId = await db.user
+              .get('id')
+              .create({ ...userData, profile: { create: profileData } });
+            const profileId = await db.profile.get('id').create(profileData);
+
+            await db.user.find(userId).update({
+              profile: {
+                set: { id: profileId },
+              },
+            });
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(2);
+            expect(afterUpdate).toHaveBeenCalledTimes(2);
+          });
+        });
       });
 
       describe('delete', () => {
@@ -727,6 +956,45 @@ describe('hasOne', () => {
           const count = await db.profile.count();
           expect(count).toBe(0);
         });
+
+        describe('relation callbacks', () => {
+          const { beforeDelete, afterDelete, resetMocks } = useRelationCallback(
+            db.user.relations.profile,
+          );
+
+          it('should invoke callbacks', async () => {
+            const id = await db.user
+              .get('id')
+              .create({ ...userData, profile: { create: profileData } });
+
+            await db.user.find(id).update({
+              profile: {
+                delete: true,
+              },
+            });
+
+            expect(beforeDelete).toHaveBeenCalledTimes(1);
+            expect(afterDelete).toHaveBeenCalledTimes(1);
+          });
+
+          it('should invoke callbacks in a batch update', async () => {
+            resetMocks();
+
+            const ids = await db.user.pluck('id').createMany([
+              { ...userData, profile: { create: profileData } },
+              { ...userData, profile: { create: profileData } },
+            ]);
+
+            await db.user.where({ id: { in: ids } }).update({
+              profile: {
+                delete: true,
+              },
+            });
+
+            expect(beforeDelete).toHaveBeenCalledTimes(1);
+            expect(afterDelete).toHaveBeenCalledTimes(1);
+          });
+        });
       });
 
       describe('nested update', () => {
@@ -764,6 +1032,49 @@ describe('hasOne', () => {
           const bios = await db.profile.pluck('bio');
           expect(bios).toEqual(['updated', 'updated']);
         });
+
+        describe('relation callbacks', () => {
+          const { beforeUpdate, afterUpdate, resetMocks } = useRelationCallback(
+            db.user.relations.profile,
+          );
+
+          it('should invoke callbacks', async () => {
+            const id = await db.user
+              .get('id')
+              .create({ ...userData, profile: { create: profileData } });
+
+            await db.user.find(id).update({
+              profile: {
+                update: {
+                  bio: 'updated',
+                },
+              },
+            });
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(1);
+            expect(afterUpdate).toHaveBeenCalledTimes(1);
+          });
+
+          it('should invoke callbacks in a batch update', async () => {
+            resetMocks();
+
+            const ids = await db.user.pluck('id').createMany([
+              { ...userData, profile: { create: profileData } },
+              { ...userData, profile: { create: profileData } },
+            ]);
+
+            await db.user.where({ id: { in: ids } }).update({
+              profile: {
+                update: {
+                  bio: 'updated',
+                },
+              },
+            });
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(1);
+            expect(afterUpdate).toHaveBeenCalledTimes(1);
+          });
+        });
       });
 
       describe('nested upsert', () => {
@@ -827,6 +1138,56 @@ describe('hasOne', () => {
 
           await expect(query).rejects.toThrow();
         });
+
+        describe('relation callbacks', () => {
+          const {
+            beforeUpdate,
+            afterUpdate,
+            beforeCreate,
+            afterCreate,
+            resetMocks,
+          } = useRelationCallback(db.user.relations.profile);
+
+          it('should invoke callbacks when connecting', async () => {
+            const userId = await db.user
+              .get('id')
+              .create({ ...userData, profile: { create: profileData } });
+
+            await db.user.find(userId).update({
+              profile: {
+                upsert: {
+                  update: {
+                    bio: 'updated',
+                  },
+                  create: profileData,
+                },
+              },
+            });
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(1);
+            expect(afterUpdate).toHaveBeenCalledTimes(1);
+          });
+
+          it('should invoke callbacks when creating', async () => {
+            resetMocks();
+
+            const userId = await db.user.get('id').create(userData);
+
+            await db.user.find(userId).update({
+              profile: {
+                upsert: {
+                  update: {
+                    bio: 'updated',
+                  },
+                  create: profileData,
+                },
+              },
+            });
+
+            expect(beforeCreate).toHaveBeenCalledTimes(1);
+            expect(afterCreate).toHaveBeenCalledTimes(1);
+          });
+        });
       });
 
       describe('nested create', () => {
@@ -868,6 +1229,35 @@ describe('hasOne', () => {
 
           await expect(query).rejects.toThrow();
         });
+
+        describe('relation callbacks', () => {
+          const {
+            beforeUpdate,
+            afterUpdate,
+            beforeCreate,
+            afterCreate,
+            resetMocks,
+          } = useRelationCallback(db.user.relations.profile);
+
+          it('should invoke callbacks to disconnect previous and create new', async () => {
+            const id = await db.user
+              .get('id')
+              .create({ ...userData, profile: { create: profileData } });
+
+            resetMocks();
+
+            await db.user.find(id).update({
+              profile: {
+                create: profileData,
+              },
+            });
+
+            expect(beforeUpdate).toHaveBeenCalledTimes(1);
+            expect(afterUpdate).toHaveBeenCalledTimes(1);
+            expect(beforeCreate).toHaveBeenCalledTimes(1);
+            expect(afterCreate).toHaveBeenCalledTimes(1);
+          });
+        });
       });
     });
   });
@@ -918,6 +1308,7 @@ describe('hasOne through', () => {
       columns = this.setColumns((t) => ({
         postId: t.integer().foreignKey(() => Post, 'id'),
         tagId: t.integer().foreignKey(() => Tag, 'id'),
+        ...t.primaryKey(['postId', 'tagId']),
       }));
 
       relations = {

package/src/test-utils/test-utils.ts

@@ -4,6 +4,7 @@ import {
   startTransaction,
 } from 'pg-transactional-tests';
 import { db } from './test-db';
+import { DeleteQueryData, InsertQueryData, Query, UpdateQueryData } from 'pqb';
 
 type AssertEqual<T, Expected> = [T] extends [Expected]
   ? [Expected] extends [T]
@@ -67,3 +68,52 @@ export const useTestDatabase = () => {
     await db.$close();
   });
 };
+
+export const useRelationCallback = (rel: { query: Query }) => {
+  const beforeCreate = jest.fn();
+  const afterCreate = jest.fn();
+  const beforeUpdate = jest.fn();
+  const afterUpdate = jest.fn();
+  const beforeDelete = jest.fn();
+  const afterDelete = jest.fn();
+  const relQuery = rel.query;
+
+  beforeAll(() => {
+    relQuery._beforeCreate(beforeCreate);
+    relQuery._afterCreate(afterCreate);
+    relQuery._beforeUpdate(beforeUpdate);
+    relQuery._afterUpdate(afterUpdate);
+    relQuery._beforeDelete(beforeDelete);
+    relQuery._afterDelete(afterDelete);
+  });
+
+  afterAll(() => {
+    let q;
+    q = relQuery.query as InsertQueryData;
+    q.beforeCreate?.pop();
+    q.afterCreate?.pop();
+    q = relQuery.query as UpdateQueryData;
+    q.beforeUpdate?.pop();
+    q.afterUpdate?.pop();
+    q = relQuery.query as DeleteQueryData;
+    q.beforeDelete?.pop();
+    q.afterDelete?.pop();
+  });
+
+  return {
+    beforeCreate,
+    afterCreate,
+    beforeUpdate,
+    afterUpdate,
+    beforeDelete,
+    afterDelete,
+    resetMocks() {
+      beforeCreate.mockReset();
+      afterCreate.mockReset();
+      beforeUpdate.mockReset();
+      afterUpdate.mockReset();
+      beforeDelete.mockReset();
+      afterDelete.mockReset();
+    },
+  };
+};