@declaro/data 2.0.0-beta.95 → 2.0.0-beta.97

package/dist/ts/test/mock/repositories/mock-memory-repository.d.ts

@@ -7,6 +7,7 @@ export interface IMockMemoryRepositoryArgs<TSchema extends AnyModelSchema> {
     schema: TSchema;
     lookup?: (data: InferDetail<TSchema>, lookup: InferLookup<TSchema>) => boolean;
     filter?: (data: InferSummary<TSchema>, filters: InferFilters<TSchema>) => boolean;
+    assign?: (data: InferDetail<TSchema>, input: InferInput<TSchema>) => InferDetail<TSchema>;
 }
 export declare class MockMemoryRepository<TSchema extends AnyModelSchema> implements IRepository<TSchema> {
     protected args: IMockMemoryRepositoryArgs<TSchema>;
@@ -31,6 +32,13 @@ export declare class MockMemoryRepository<TSchema extends AnyModelSchema> implem
      * @returns Filtered array of items
      */
     protected applyFilters(input: InferFilters<TSchema>): InferDetail<TSchema>[];
+    /**
+     * Assign input data to existing data using the provided assign function or default Object.assign
+     * @param existingData - The existing data to merge with
+     * @param input - The input data to assign
+     * @returns The merged data
+     */
+    protected assignInput(existingData: InferDetail<TSchema>, input: InferInput<TSchema>): InferDetail<TSchema>;
     protected generatePrimaryKey(): Promise<string | number>;
 }
 //# sourceMappingURL=mock-memory-repository.d.ts.map

package/dist/ts/test/mock/repositories/mock-memory-repository.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mock-memory-repository.d.ts","sourceRoot":"","sources":["../../../../../src/test/mock/repositories/mock-memory-repository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,oBAAoB,EAAqB,MAAM,eAAe,CAAA;AAC5F,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uCAAuC,CAAA;AAExE,OAAO,KAAK,EACR,WAAW,EACX,YAAY,EACZ,UAAU,EACV,WAAW,EACX,kBAAkB,EAClB,YAAY,EACf,MAAM,wCAAwC,CAAA;AAE/C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kDAAkD,CAAA;AACtF,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAA;AAE5F,MAAM,WAAW,yBAAyB,CAAC,OAAO,SAAS,cAAc;IACrE,MAAM,EAAE,OAAO,CAAA;IACf,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,OAAO,CAAA;IAC9E,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,KAAK,OAAO,CAAA;CACpF;AAED,qBAAa,oBAAoB,CAAC,OAAO,SAAS,cAAc,CAAE,YAAW,WAAW,CAAC,OAAO,CAAC;IAMjF,SAAS,CAAC,IAAI,EAAE,yBAAyB,CAAC,OAAO,CAAC;IAL9D,SAAS,CAAC,IAAI,oCAA0C;IACxD,SAAS,CAAC,KAAK,oCAA0C;IACzD,SAAS,CAAC,cAAc,EAAE,oBAAoB,CAAA;IAC9C,SAAS,CAAC,MAAM,EAAE,MAAM,CAAI;gBAEN,IAAI,EAAE,yBAAyB,CAAC,OAAO,CAAC;IAOxD,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;IAevE,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;IAQzE,MAAM,CACR,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,EAC5B,OAAO,CAAC,EAAE,cAAc,CAAC,OAAO,CAAC,GAClC,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;IA2CjC,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IAepE,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IAcrE,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAuBjE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAmB/F,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,EAAE,cAAc,CAAC,OAAO,CAAC,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC;IAKpG,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,cAAc,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAkB5G,UAAU,CACZ,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC,EAAE,EAC7B,OAAO,CAAC,EAAE,cAAc,GAAG,cAAc,GAC1C,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;IAIlC;;;;OAIG;IACH,SAAS,CAAC,YAAY,CAAC,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,EAAE;cAW5D,kBAAkB;CAcrC"}
+{"version":3,"file":"mock-memory-repository.d.ts","sourceRoot":"","sources":["../../../../../src/test/mock/repositories/mock-memory-repository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,oBAAoB,EAAqB,MAAM,eAAe,CAAA;AAC5F,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uCAAuC,CAAA;AAExE,OAAO,KAAK,EACR,WAAW,EACX,YAAY,EACZ,UAAU,EACV,WAAW,EACX,kBAAkB,EAClB,YAAY,EACf,MAAM,wCAAwC,CAAA;AAE/C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kDAAkD,CAAA;AACtF,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAA;AAE5F,MAAM,WAAW,yBAAyB,CAAC,OAAO,SAAS,cAAc;IACrE,MAAM,EAAE,OAAO,CAAA;IACf,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,OAAO,CAAA;IAC9E,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,KAAK,OAAO,CAAA;IACjF,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,WAAW,CAAC,OAAO,CAAC,CAAA;CAC5F;AAED,qBAAa,oBAAoB,CAAC,OAAO,SAAS,cAAc,CAAE,YAAW,WAAW,CAAC,OAAO,CAAC;IAMjF,SAAS,CAAC,IAAI,EAAE,yBAAyB,CAAC,OAAO,CAAC;IAL9D,SAAS,CAAC,IAAI,oCAA0C;IACxD,SAAS,CAAC,KAAK,oCAA0C;IACzD,SAAS,CAAC,cAAc,EAAE,oBAAoB,CAAA;IAC9C,SAAS,CAAC,MAAM,EAAE,MAAM,CAAI;gBAEN,IAAI,EAAE,yBAAyB,CAAC,OAAO,CAAC;IAOxD,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;IAevE,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;IAQzE,MAAM,CACR,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,EAC5B,OAAO,CAAC,EAAE,cAAc,CAAC,OAAO,CAAC,GAClC,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;IA2CjC,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IAepE,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IAcrE,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAsBjE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAmB/F,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,EAAE,cAAc,CAAC,OAAO,CAAC,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC;IAKpG,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,cAAc,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAkB5G,UAAU,CACZ,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC,EAAE,EAC7B,OAAO,CAAC,EAAE,cAAc,GAAG,cAAc,GAC1C,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;IAIlC;;;;OAIG;IACH,SAAS,CAAC,YAAY,CAAC,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,EAAE;IAW5E;;;;;OAKG;IACH,SAAS,CAAC,WAAW,CAAC,YAAY,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC;cAS3F,kBAAkB;CAcrC"}

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@declaro/data",
-    "version": "2.0.0-beta.95",
+    "version": "2.0.0-beta.97",
     "description": "A data-mapper framework for managing application data across integrated systems.",
     "main": "dist/node/index.cjs",
     "module": "dist/node/index.js",
@@ -22,9 +22,9 @@
         "@declaro/zod": "^2.0.0-beta.51"
     },
     "devDependencies": {
-        "@declaro/auth": "^2.0.0-beta.95",
-        "@declaro/core": "^2.0.0-beta.95",
-        "@declaro/zod": "^2.0.0-beta.95",
+        "@declaro/auth": "^2.0.0-beta.97",
+        "@declaro/core": "^2.0.0-beta.97",
+        "@declaro/zod": "^2.0.0-beta.97",
         "crypto-browserify": "^3.12.1",
         "typescript": "^5.8.3",
         "uuid": "^11.1.0",
@@ -43,5 +43,5 @@
         "require": "./dist/node/index.cjs",
         "browser": "./dist/browser/index.js"
     },
-    "gitHead": "9b00e9de1fd859c26610126a123636e5211502a6"
+    "gitHead": "f5735a57bfd6cc342c60e9974395c389f43c3f93"
 }

package/src/application/model-controller.test.ts

@@ -23,18 +23,21 @@ describe('ModelController', () => {
             getMockAuthSession({
                 claims: ['books::book.write:all'],
             }),
+            null,
             authService,
         )
         invalidAuthValidator = new AuthValidator(
             getMockAuthSession({
                 claims: ['authors::author.write:all'],
             }),
+            null,
             authService,
         )
         readAuthValidator = new AuthValidator(
             getMockAuthSession({
                 claims: ['books::book.read:all'],
             }),
+            null,
             authService,
         )
         service = new ModelService({
@@ -274,6 +277,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.create:all', 'books::book.update:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, createUpdateValidator)
@@ -289,6 +293,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.create:all', 'books::book.update:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, createUpdateValidator)
@@ -308,6 +313,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.create:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, createOnlyValidator)
@@ -322,6 +328,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.update:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, updateOnlyValidator)
@@ -336,6 +343,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.create:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, createOnlyValidator)
@@ -353,6 +361,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.update:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, updateOnlyValidator)
@@ -390,6 +399,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.create:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, createOnlyValidator)
@@ -405,6 +415,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.update:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, updateOnlyValidator)
@@ -423,6 +434,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.remove:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, removeOnlyValidator)
@@ -440,6 +452,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.restore:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, restoreOnlyValidator)
@@ -458,6 +471,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['users::user.write:all'], // Wrong namespace
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, wrongNamespaceValidator)
@@ -474,6 +488,7 @@ describe('ModelController', () => {
                 getMockAuthSession({
                     claims: ['books::author.write:all'], // Wrong resource
                 }),
+                null,
                 authService,
             )
             const controller = new ModelController(service, wrongResourceValidator)

package/src/application/read-only-model-controller.test.ts

@@ -22,12 +22,14 @@ describe('ReadOnlyModelController', () => {
             getMockAuthSession({
                 claims: ['books::book.read:all'],
             }),
+            null,
             authService,
         )
         invalidAuthValidator = new AuthValidator(
             getMockAuthSession({
                 claims: ['authors::author.read:all'],
             }),
+            null,
             authService,
         )
         service = new ReadOnlyModelService({
@@ -203,6 +205,7 @@ describe('ReadOnlyModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.load:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ReadOnlyModelController(service, loadOnlyValidator)
@@ -220,6 +223,7 @@ describe('ReadOnlyModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.loadMany:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ReadOnlyModelController(service, loadManyOnlyValidator)
@@ -239,6 +243,7 @@ describe('ReadOnlyModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.search:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ReadOnlyModelController(service, searchOnlyValidator)
@@ -259,6 +264,7 @@ describe('ReadOnlyModelController', () => {
                 getMockAuthSession({
                     claims: ['books::book.count:all'],
                 }),
+                null,
                 authService,
             )
             const controller = new ReadOnlyModelController(service, countOnlyValidator)
@@ -278,6 +284,7 @@ describe('ReadOnlyModelController', () => {
                 getMockAuthSession({
                     claims: ['users::user.read:all'], // Wrong namespace
                 }),
+                null,
                 authService,
             )
             const controller = new ReadOnlyModelController(service, wrongNamespaceValidator)
@@ -293,6 +300,7 @@ describe('ReadOnlyModelController', () => {
                 getMockAuthSession({
                     claims: ['books::author.read:all'], // Wrong resource
                 }),
+                null,
                 authService,
             )
             const controller = new ReadOnlyModelController(service, wrongResourceValidator)

package/src/test/mock/repositories/mock-memory-repository.test.ts

@@ -712,4 +712,208 @@ describe('MockMemoryRepository', () => {
             expect(await repository.load({ id: 100 })).toEqual(inputs[99])
         })
     })
+
+    describe('assign functionality', () => {
+        it('should use default Object.assign for create when no custom assign function is provided', async () => {
+            const input = { title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+            const createdItem = await repository.create(input)
+
+            expect(createdItem).toMatchObject(input)
+            expect(createdItem.id).toBeDefined()
+        })
+
+        it('should use default Object.assign for update when no custom assign function is provided', async () => {
+            const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+            const createdItem = await repository.create(input)
+
+            const updateInput = { title: 'Updated Book', author: 'Updated Author', publishedDate: new Date() }
+            const updatedItem = await repository.update({ id: createdItem.id }, updateInput)
+
+            expect(updatedItem).toEqual({
+                id: createdItem.id,
+                title: 'Updated Book',
+                author: 'Updated Author',
+                publishedDate: updateInput.publishedDate,
+            })
+        })
+
+        it('should use default Object.assign for upsert when no custom assign function is provided', async () => {
+            const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+            const upsertedItem = await repository.upsert(input)
+
+            expect(upsertedItem).toEqual(input)
+
+            const updateInput = { id: 42, title: 'Updated Book', author: 'Updated Author', publishedDate: new Date() }
+            const updatedItem = await repository.upsert(updateInput)
+
+            expect(updatedItem).toEqual(updateInput)
+        })
+
+        it('should use custom assign function for create operation', async () => {
+            const customAssignMock = mock((existing: any, input: any) => ({
+                ...existing,
+                ...input,
+                customField: 'custom_create_value',
+            }))
+
+            const customRepository = new MockMemoryRepository({
+                schema: mockSchema,
+                assign: customAssignMock,
+            })
+
+            const input = { title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+            const createdItem = await customRepository.create(input)
+
+            expect(customAssignMock).toHaveBeenCalledWith({}, input)
+            expect((createdItem as any).customField).toBe('custom_create_value')
+            expect(createdItem).toMatchObject(input)
+        })
+
+        it('should use custom assign function for update operation', async () => {
+            const customAssignMock = mock((existing: any, input: any) => ({
+                ...existing,
+                ...input,
+                customField: 'custom_update_value',
+                lastModified: new Date('2023-01-01'),
+            }))
+
+            const customRepository = new MockMemoryRepository({
+                schema: mockSchema,
+                assign: customAssignMock,
+            })
+
+            const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+            const createdItem = await customRepository.create(input)
+
+            const updateInput = { title: 'Updated Book', author: 'Updated Author', publishedDate: new Date() }
+            const updatedItem = await customRepository.update({ id: createdItem.id }, updateInput)
+
+            expect(customAssignMock).toHaveBeenCalledWith(createdItem, updateInput)
+            expect((updatedItem as any).customField).toBe('custom_update_value')
+            expect((updatedItem as any).lastModified).toEqual(new Date('2023-01-01'))
+            expect(updatedItem.title).toBe('Updated Book')
+        })
+
+        it('should use custom assign function for upsert operation on existing item', async () => {
+            const customAssignMock = mock((existing: any, input: any) => ({
+                ...existing,
+                ...input,
+                mergeTimestamp: new Date('2023-01-01'),
+                isUpserted: true,
+            }))
+
+            const customRepository = new MockMemoryRepository({
+                schema: mockSchema,
+                assign: customAssignMock,
+            })
+
+            const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+            const createdItem = await customRepository.create(input)
+
+            const upsertInput = { id: 42, title: 'Upserted Book', author: 'Upserted Author', publishedDate: new Date() }
+            const upsertedItem = await customRepository.upsert(upsertInput)
+
+            // Should have been called twice: once for create, once for upsert
+            expect(customAssignMock).toHaveBeenCalledTimes(2)
+            expect(customAssignMock).toHaveBeenLastCalledWith(createdItem, upsertInput)
+            expect((upsertedItem as any).mergeTimestamp).toEqual(new Date('2023-01-01'))
+            expect((upsertedItem as any).isUpserted).toBe(true)
+            expect(upsertedItem.title).toBe('Upserted Book')
+        })
+
+        it('should use custom assign function for upsert operation on new item', async () => {
+            const customAssignMock = mock((existing: any, input: any) => ({
+                ...existing,
+                ...input,
+                createdViaUpsert: true,
+            }))
+
+            const customRepository = new MockMemoryRepository({
+                schema: mockSchema,
+                assign: customAssignMock,
+            })
+
+            const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+            const upsertedItem = await customRepository.upsert(input)
+
+            expect(customAssignMock).toHaveBeenCalledWith({}, input)
+            expect((upsertedItem as any).createdViaUpsert).toBe(true)
+            expect(upsertedItem).toMatchObject(input)
+        })
+
+        it('should use custom assign function for bulkUpsert operation', async () => {
+            const customAssignMock = mock((existing: any, input: any) => ({
+                ...existing,
+                ...input,
+                bulkProcessed: true,
+            }))
+
+            const customRepository = new MockMemoryRepository({
+                schema: mockSchema,
+                assign: customAssignMock,
+            })
+
+            const inputs = [
+                { id: 1, title: 'Book 1', author: 'Author 1', publishedDate: new Date() },
+                { id: 2, title: 'Book 2', author: 'Author 2', publishedDate: new Date() },
+            ]
+
+            const results = await customRepository.bulkUpsert(inputs)
+
+            expect(customAssignMock).toHaveBeenCalledTimes(2)
+            expect((results[0] as any).bulkProcessed).toBe(true)
+            expect((results[1] as any).bulkProcessed).toBe(true)
+            expect(results[0]).toMatchObject(inputs[0])
+            expect(results[1]).toMatchObject(inputs[1])
+        })
+
+        it('should handle complex custom assign logic with conditional merging', async () => {
+            const customAssign = (existing: any, input: any) => {
+                const result = { ...existing, ...input }
+
+                // Custom logic: preserve original author if input doesn't have one
+                if (!input.author && existing.author) {
+                    result.author = existing.author
+                }
+
+                // Custom logic: track modification count
+                result.modificationCount = (existing.modificationCount || 0) + 1
+
+                return result
+            }
+
+            const customRepository = new MockMemoryRepository({
+                schema: mockSchema,
+                assign: customAssign,
+            })
+
+            const input = { id: 42, title: 'Test Book', author: 'Original Author', publishedDate: new Date() }
+            const createdItem = await customRepository.create(input)
+
+            expect((createdItem as any).modificationCount).toBe(1)
+
+            // Update with partial data - use existing values for required fields
+            const updateInput = {
+                title: 'Updated Book',
+                author: createdItem.author,
+                publishedDate: createdItem.publishedDate,
+            }
+            const updatedItem = await customRepository.update({ id: createdItem.id }, updateInput)
+
+            expect(updatedItem.author).toBe('Original Author')
+            expect(updatedItem.title).toBe('Updated Book')
+            expect((updatedItem as any).modificationCount).toBe(2)
+
+            // Update with new author
+            const updateWithAuthor = {
+                title: updatedItem.title,
+                author: 'New Author',
+                publishedDate: updatedItem.publishedDate,
+            }
+            const finalItem = await customRepository.update({ id: createdItem.id }, updateWithAuthor)
+
+            expect(finalItem.author).toBe('New Author')
+            expect((finalItem as any).modificationCount).toBe(3)
+        })
+    })
 })

package/src/test/mock/repositories/mock-memory-repository.ts

@@ -17,6 +17,7 @@ export interface IMockMemoryRepositoryArgs<TSchema extends AnyModelSchema> {
     schema: TSchema
     lookup?: (data: InferDetail<TSchema>, lookup: InferLookup<TSchema>) => boolean
     filter?: (data: InferSummary<TSchema>, filters: InferFilters<TSchema>) => boolean
+    assign?: (data: InferDetail<TSchema>, input: InferInput<TSchema>) => InferDetail<TSchema>
 }
 
 export class MockMemoryRepository<TSchema extends AnyModelSchema> implements IRepository<TSchema> {
@@ -140,17 +141,16 @@ export class MockMemoryRepository<TSchema extends AnyModelSchema> implements IRe
             throw new Error('Item with the same primary key already exists')
         }
 
-        const payload = {
-            ...(input as any),
-        }
+        const baseData = {} as InferDetail<TSchema>
+        const payload = this.assignInput(baseData, input)
 
         if (!payload[this.entityMetadata.primaryKey]) {
             // Generate a new primary key if not provided
             payload[this.entityMetadata.primaryKey!] = await this.generatePrimaryKey()
         }
 
-        this.data.set(payload[this.entityMetadata.primaryKey!], payload as InferDetail<TSchema>)
-        return payload as InferDetail<TSchema>
+        this.data.set(payload[this.entityMetadata.primaryKey!], payload)
+        return payload
     }
 
     async update(lookup: InferLookup<TSchema>, input: InferInput<TSchema>): Promise<InferDetail<TSchema>> {
@@ -167,7 +167,7 @@ export class MockMemoryRepository<TSchema extends AnyModelSchema> implements IRe
             throw new Error('Item not found')
         }
 
-        const updatedItem = Object.assign({}, existingItem, input) as InferDetail<TSchema>
+        const updatedItem = this.assignInput(existingItem, input)
         this.data.set(primaryKeyValue, updatedItem)
         return updatedItem
     }
@@ -179,13 +179,13 @@ export class MockMemoryRepository<TSchema extends AnyModelSchema> implements IRe
 
     async upsert(input: InferInput<TSchema>, options?: ICreateOptions | IUpdateOptions): Promise<InferDetail<TSchema>> {
         const primaryKeyValue = input[this.entityMetadata.primaryKey]
-        let existingItem: InferDetail<TSchema> | undefined = undefined
+        let existingItem: InferDetail<TSchema> = {} as InferDetail<TSchema>
 
         if (primaryKeyValue) {
-            existingItem = this.data.get(primaryKeyValue) ?? {}
+            existingItem = this.data.get(primaryKeyValue) ?? ({} as InferDetail<TSchema>)
         }
 
-        const updatedItem = Object.assign({}, existingItem, input) as InferDetail<TSchema>
+        const updatedItem = this.assignInput(existingItem, input)
         if (!updatedItem[this.entityMetadata.primaryKey!]) {
             updatedItem[this.entityMetadata.primaryKey!] = await this.generatePrimaryKey()
         }
@@ -218,6 +218,21 @@ export class MockMemoryRepository<TSchema extends AnyModelSchema> implements IRe
         })
     }
 
+    /**
+     * Assign input data to existing data using the provided assign function or default Object.assign
+     * @param existingData - The existing data to merge with
+     * @param input - The input data to assign
+     * @returns The merged data
+     */
+    protected assignInput(existingData: InferDetail<TSchema>, input: InferInput<TSchema>): InferDetail<TSchema> {
+        if (typeof this.args.assign === 'function') {
+            return this.args.assign(existingData, input)
+        } else {
+            // Default implementation using Object.assign
+            return Object.assign({}, existingData, input) as InferDetail<TSchema>
+        }
+    }
+
     protected async generatePrimaryKey() {
         const lookupModel: Model<any, any> = this.args.schema.definition.lookup
         const lookupMeta = await lookupModel.toJSONSchema()