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

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

@@ -0,0 +1,215 @@
+import { beforeEach, describe, expect, it, mock } from 'bun:test'
+import { MockBookSchema } from '../models/mock-book-models'
+import { MockMemoryRepository } from './mock-memory-repository'
+
+describe('MockMemoryRepository - Assign Functionality', () => {
+    const mockSchema = MockBookSchema
+
+    let repository: MockMemoryRepository<typeof mockSchema>
+
+    beforeEach(() => {
+        repository = new MockMemoryRepository({ schema: mockSchema })
+    })
+
+    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.basic.test.ts

@@ -0,0 +1,129 @@
+import { beforeEach, describe, expect, it } from 'bun:test'
+import { MockBookSchema } from '../models/mock-book-models'
+import { MockMemoryRepository } from './mock-memory-repository'
+import { z } from 'zod/v4'
+import { ZodModel } from '@declaro/zod'
+
+describe('MockMemoryRepository - Basic Operations', () => {
+    const mockSchema = MockBookSchema
+
+    let repository: MockMemoryRepository<typeof mockSchema>
+
+    beforeEach(() => {
+        repository = new MockMemoryRepository({ schema: mockSchema })
+    })
+
+    it('should create an item', async () => {
+        const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+        const createdItem = await repository.create(input)
+
+        expect(createdItem).toEqual(input)
+        expect(await repository.load({ id: createdItem.id })).toEqual(createdItem)
+    })
+
+    it('should throw an error when creating an item with duplicate primary key', async () => {
+        const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+        const createdItem = await repository.create(input)
+
+        await expect(repository.create({ ...input, id: createdItem.id })).rejects.toThrow(
+            'Item with the same primary key already exists',
+        )
+    })
+
+    it('should update an existing item', async () => {
+        const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+        const createdItem = await repository.create(input)
+
+        const updatedInput = { title: 'Updated Book', author: 'Updated Author', publishedDate: new Date() }
+        const updatedItem = await repository.update({ id: createdItem.id }, updatedInput)
+
+        expect(updatedItem).toEqual({ id: createdItem.id, ...updatedInput })
+        expect(await repository.load({ id: createdItem.id })).toEqual({ id: createdItem.id, ...updatedInput })
+    })
+
+    it('should throw an error when updating a non-existent item', async () => {
+        const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+
+        await expect(repository.update({ id: 999 }, input)).rejects.toThrow('Item not found')
+    })
+
+    it('should remove an item', async () => {
+        const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+        const createdItem = await repository.create(input)
+
+        const removedItem = await repository.remove({ id: createdItem.id })
+
+        expect(removedItem).toEqual(input)
+        const loadedItem = await repository.load({ id: createdItem.id })
+        expect(loadedItem).toBeNull()
+    })
+
+    it('should restore a removed item', async () => {
+        const input = { id: 42, title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+        const createdItem = await repository.create(input)
+        await repository.remove({ id: createdItem.id })
+
+        const restoredItem = await repository.restore({ id: createdItem.id })
+
+        expect(restoredItem).toEqual(input)
+        expect(await repository.load({ id: createdItem.id })).toEqual(input)
+    })
+
+    it('should throw an error when restoring a non-existent item', async () => {
+        await expect(repository.restore({ id: 999 })).rejects.toThrow('Item not found in trash')
+    })
+
+    it('should allow me to create an item without a primary key', async () => {
+        const input = { title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+        const createdItem = await repository.create(input)
+
+        expect(createdItem.id).toBeDefined()
+        expect(createdItem.title).toBe(input.title)
+        expect(createdItem.author).toBe(input.author)
+        expect(await repository.load({ id: createdItem.id })).toEqual(createdItem)
+    })
+
+    it('should increment ids when creating items without a primary key', async () => {
+        const input1 = { title: 'Test Book 1', author: 'Author Name 1', publishedDate: new Date() }
+        const input2 = { title: 'Test Book 2', author: 'Author Name 2', publishedDate: new Date() }
+
+        const createdItem1 = await repository.create(input1)
+        const createdItem2 = await repository.create(input2)
+
+        expect(createdItem1.id).toBe(1)
+        expect(createdItem2.id).toBe(2)
+        expect(await repository.load({ id: createdItem1.id })).toEqual(createdItem1)
+        expect(await repository.load({ id: createdItem2.id })).toEqual(createdItem2)
+    })
+
+    it('should return null when loading a non-existent item', async () => {
+        const result = await repository.load({ id: 999 })
+        expect(result).toBeNull()
+    })
+
+    it('should be able to load items from a custom filter', async () => {
+        // Creating a hypothetical schema with a custom filter, and a title lookup attribute
+        const repository = new MockMemoryRepository({
+            schema: mockSchema.custom({
+                lookup: (h) =>
+                    new ZodModel(
+                        h.name,
+                        z.object({
+                            id: z.number().optional(),
+                            title: z.string().optional(),
+                        }),
+                    ),
+            }),
+            lookup: (data, lookup) =>
+                data.id === lookup.id || data.title?.toLowerCase() === lookup.title?.toLowerCase(),
+        })
+
+        const input = { title: 'Test Book', author: 'Author Name', publishedDate: new Date() }
+        const createdItem = await repository.create(input)
+
+        const loadedItem = await repository.load({ id: createdItem.id })
+        const titleLoadedItem = await repository.load({ title: createdItem.title })
+        expect(loadedItem).toEqual(createdItem)
+        expect(titleLoadedItem).toEqual(createdItem)
+    })
+})

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

@@ -0,0 +1,159 @@
+import { beforeEach, describe, expect, it } from 'bun:test'
+import { MockBookSchema } from '../models/mock-book-models'
+import { MockMemoryRepository } from './mock-memory-repository'
+
+describe('MockMemoryRepository - Bulk Upsert Functionality', () => {
+    const mockSchema = MockBookSchema
+
+    let repository: MockMemoryRepository<typeof mockSchema>
+
+    beforeEach(() => {
+        repository = new MockMemoryRepository({ schema: mockSchema })
+    })
+
+    it('should upsert multiple new items', async () => {
+        const inputs = [
+            { id: 1, title: 'Book 1', author: 'Author 1', publishedDate: new Date('2023-01-01') },
+            { id: 2, title: 'Book 2', author: 'Author 2', publishedDate: new Date('2023-02-01') },
+            { id: 3, title: 'Book 3', author: 'Author 3', publishedDate: new Date('2023-03-01') },
+        ]
+
+        const upsertedItems = await repository.bulkUpsert(inputs)
+
+        expect(upsertedItems).toHaveLength(3)
+        expect(upsertedItems).toEqual(inputs)
+
+        // Verify all items were created
+        for (const input of inputs) {
+            expect(await repository.load({ id: input.id })).toEqual(input)
+        }
+    })
+
+    it('should update existing items and create new ones in same operation', async () => {
+        // Create some initial items
+        const existing1 = {
+            id: 1,
+            title: 'Original Book 1',
+            author: 'Original Author 1',
+            publishedDate: new Date('2023-01-01'),
+        }
+        const existing2 = {
+            id: 2,
+            title: 'Original Book 2',
+            author: 'Original Author 2',
+            publishedDate: new Date('2023-02-01'),
+        }
+        await repository.create(existing1)
+        await repository.create(existing2)
+
+        // Bulk upsert with mix of updates and new items
+        const upsertInputs = [
+            { id: 1, title: 'Updated Book 1', author: 'Updated Author 1', publishedDate: new Date('2023-06-01') }, // Update
+            { id: 2, title: 'Updated Book 2', author: 'Updated Author 2', publishedDate: new Date('2023-07-01') }, // Update
+            { id: 3, title: 'New Book 3', author: 'New Author 3', publishedDate: new Date('2023-08-01') }, // Create
+            { id: 4, title: 'New Book 4', author: 'New Author 4', publishedDate: new Date('2023-09-01') }, // Create
+        ]
+
+        const upsertedItems = await repository.bulkUpsert(upsertInputs)
+
+        expect(upsertedItems).toHaveLength(4)
+        expect(upsertedItems).toEqual(upsertInputs)
+
+        // Verify all items have the updated/new values
+        for (const input of upsertInputs) {
+            expect(await repository.load({ id: input.id })).toEqual(input)
+        }
+    })
+
+    it('should handle bulk upsert with items without primary keys', async () => {
+        const inputs = [
+            { title: 'Book Without ID 1', author: 'Author 1', publishedDate: new Date() },
+            { title: 'Book Without ID 2', author: 'Author 2', publishedDate: new Date() },
+            { title: 'Book Without ID 3', author: 'Author 3', publishedDate: new Date() },
+        ]
+
+        const upsertedItems = await repository.bulkUpsert(inputs)
+
+        expect(upsertedItems).toHaveLength(3)
+
+        // All items should have generated IDs
+        expect(upsertedItems[0].id).toBe(1)
+        expect(upsertedItems[1].id).toBe(2)
+        expect(upsertedItems[2].id).toBe(3)
+
+        // Verify content is preserved
+        for (let i = 0; i < inputs.length; i++) {
+            expect(upsertedItems[i].title).toBe(inputs[i].title)
+            expect(upsertedItems[i].author).toBe(inputs[i].author)
+            expect(await repository.load({ id: upsertedItems[i].id })).toEqual(upsertedItems[i])
+        }
+    })
+
+    it('should handle empty bulk upsert', async () => {
+        const upsertedItems = await repository.bulkUpsert([])
+
+        expect(upsertedItems).toEqual([])
+    })
+
+    it('should handle bulk upsert with partial updates', async () => {
+        // Create initial items
+        const initial1 = {
+            id: 1,
+            title: 'Original Book 1',
+            author: 'Original Author 1',
+            publishedDate: new Date('2023-01-01'),
+        }
+        const initial2 = {
+            id: 2,
+            title: 'Original Book 2',
+            author: 'Original Author 2',
+            publishedDate: new Date('2023-02-01'),
+        }
+        await repository.create(initial1)
+        await repository.create(initial2)
+
+        // Updates with all required fields but only changing title
+        const partialUpdates = [
+            { id: 1, title: 'Updated Title 1', author: 'Original Author 1', publishedDate: new Date('2023-01-01') },
+            { id: 2, title: 'Updated Title 2', author: 'Original Author 2', publishedDate: new Date('2023-02-01') },
+        ]
+
+        const upsertedItems = await repository.bulkUpsert(partialUpdates)
+
+        expect(upsertedItems).toHaveLength(2)
+
+        // Should have updated titles but kept other properties
+        expect(upsertedItems[0].title).toBe('Updated Title 1')
+        expect(upsertedItems[0].author).toBe('Original Author 1')
+        expect(upsertedItems[0].publishedDate).toEqual(initial1.publishedDate)
+
+        expect(upsertedItems[1].title).toBe('Updated Title 2')
+        expect(upsertedItems[1].author).toBe('Original Author 2')
+        expect(upsertedItems[1].publishedDate).toEqual(initial2.publishedDate)
+    })
+
+    it('should handle large bulk operations efficiently', async () => {
+        // Create a large number of items to test performance
+        const inputs: Array<{ id: number; title: string; author: string; publishedDate: Date }> = []
+        for (let i = 1; i <= 100; i++) {
+            inputs.push({
+                id: i,
+                title: `Book ${i}`,
+                author: `Author ${i}`,
+                publishedDate: new Date(`2023-${String((i % 12) + 1).padStart(2, '0')}-01`),
+            })
+        }
+
+        const startTime = Date.now()
+        const upsertedItems = await repository.bulkUpsert(inputs)
+        const endTime = Date.now()
+
+        expect(upsertedItems).toHaveLength(100)
+        expect(endTime - startTime).toBeLessThan(1000) // Should complete in under 1 second
+
+        // Verify a few random items
+        expect(await repository.load({ id: 1 })).toEqual(inputs[0])
+        expect(await repository.load({ id: 50 })).toEqual(inputs[49])
+        expect(await repository.load({ id: 100 })).toEqual(inputs[99])
+    })
+})

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

@@ -0,0 +1,98 @@
+import { beforeEach, describe, expect, it } from 'bun:test'
+import { MockBookSchema } from '../models/mock-book-models'
+import { MockMemoryRepository } from './mock-memory-repository'
+
+describe('MockMemoryRepository - Count Functionality', () => {
+    const mockSchema = MockBookSchema
+
+    let repository: MockMemoryRepository<typeof mockSchema>
+
+    beforeEach(() => {
+        repository = new MockMemoryRepository({ schema: mockSchema })
+    })
+
+    it('should count all items when no filter is provided', async () => {
+        // Create test data
+        await repository.create({ title: 'Book 1', author: 'Author 1', publishedDate: new Date() })
+        await repository.create({ title: 'Book 2', author: 'Author 2', publishedDate: new Date() })
+        await repository.create({ title: 'Book 3', author: 'Author 3', publishedDate: new Date() })
+
+        const count = await repository.count({})
+        expect(count).toBe(3)
+    })
+
+    it('should count filtered items when filter function is provided', async () => {
+        const repositoryWithFilter = new MockMemoryRepository({
+            schema: mockSchema,
+            filter: (data, filters) => {
+                if (filters.text) {
+                    return data.title.toLowerCase().includes(filters.text.toLowerCase())
+                }
+                return true
+            },
+        })
+
+        // Create test data
+        await repositoryWithFilter.create({ title: 'Test Book A', author: 'Author 1', publishedDate: new Date() })
+        await repositoryWithFilter.create({ title: 'Test Book B', author: 'Author 2', publishedDate: new Date() })
+        await repositoryWithFilter.create({ title: 'Other Book', author: 'Author 3', publishedDate: new Date() })
+        await repositoryWithFilter.create({ title: 'Another Book', author: 'Author 4', publishedDate: new Date() })
+
+        // Count all items
+        const totalCount = await repositoryWithFilter.count({})
+        expect(totalCount).toBe(4)
+
+        // Count filtered items
+        const filteredCount = await repositoryWithFilter.count({ text: 'Test' })
+        expect(filteredCount).toBe(2)
+
+        // Count with no matches
+        const noMatchCount = await repositoryWithFilter.count({ text: 'Ruby' })
+        expect(noMatchCount).toBe(0)
+    })
+
+    it('should count items correctly after CRUD operations', async () => {
+        const repositoryWithFilter = new MockMemoryRepository({
+            schema: mockSchema,
+            filter: (data, filters) => {
+                if (filters.text) {
+                    return data.title.toLowerCase().includes(filters.text.toLowerCase())
+                }
+                return true
+            },
+        })
+
+        // Initial count should be 0
+        expect(await repositoryWithFilter.count({})).toBe(0)
+
+        // Create items
+        const book1 = await repositoryWithFilter.create({
+            title: 'Test Book 1',
+            author: 'Author 1',
+            publishedDate: new Date(),
+        })
+        const book2 = await repositoryWithFilter.create({
+            title: 'Other Book',
+            author: 'Author 2',
+            publishedDate: new Date(),
+        })
+
+        // Count after creation
+        expect(await repositoryWithFilter.count({})).toBe(2)
+        expect(await repositoryWithFilter.count({ text: 'Test' })).toBe(1)
+
+        // Remove an item
+        await repositoryWithFilter.remove({ id: book1.id })
+
+        // Count after removal
+        expect(await repositoryWithFilter.count({})).toBe(1)
+        expect(await repositoryWithFilter.count({ text: 'Test' })).toBe(0)
+
+        // Restore the item
+        await repositoryWithFilter.restore({ id: book1.id })
+
+        // Count after restore
+        expect(await repositoryWithFilter.count({})).toBe(2)
+        expect(await repositoryWithFilter.count({ text: 'Test' })).toBe(1)
+    })
+})