@startsimpli/ui 0.1.0

package/src/components/unified-table/__tests__/hooks/useSelection.test.ts

@@ -0,0 +1,367 @@
+import { renderHook, act } from '@testing-library/react'
+import { useSelection } from '../../hooks/useSelection'
+
+interface TestData {
+  id: string
+  name: string
+}
+
+describe('useSelection', () => {
+  const mockData: TestData[] = [
+    { id: '1', name: 'Item 1' },
+    { id: '2', name: 'Item 2' },
+    { id: '3', name: 'Item 3' },
+  ]
+
+  const getRowId = (row: TestData) => row.id
+
+  const defaultProps = {
+    currentPageData: mockData,
+    totalCount: 10,
+    getRowId,
+  }
+
+  describe('Initialization', () => {
+    it('should initialize with no selection', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      expect(result.current.selectedIds.size).toBe(0)
+      expect(result.current.selectAllPages).toBe(false)
+      expect(result.current.getSelectedCount()).toBe(0)
+    })
+  })
+
+  describe('Single Row Selection', () => {
+    it('should toggle a single row selection', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleRow('1')
+      })
+
+      expect(result.current.selectedIds.has('1')).toBe(true)
+      expect(result.current.selectedIds.size).toBe(1)
+      expect(result.current.getSelectedCount()).toBe(1)
+    })
+
+    it('should deselect a selected row', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleRow('1')
+      })
+
+      expect(result.current.selectedIds.has('1')).toBe(true)
+
+      act(() => {
+        result.current.toggleRow('1')
+      })
+
+      expect(result.current.selectedIds.has('1')).toBe(false)
+      expect(result.current.selectedIds.size).toBe(0)
+    })
+
+    it('should select multiple rows individually', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleRow('1')
+        result.current.toggleRow('2')
+        result.current.toggleRow('3')
+      })
+
+      expect(result.current.selectedIds.size).toBe(3)
+      expect(result.current.selectedIds.has('1')).toBe(true)
+      expect(result.current.selectedIds.has('2')).toBe(true)
+      expect(result.current.selectedIds.has('3')).toBe(true)
+    })
+
+    it('should call onSelectionChange callback', () => {
+      const onSelectionChange = jest.fn()
+      const { result } = renderHook(() =>
+        useSelection({ ...defaultProps, onSelectionChange })
+      )
+
+      act(() => {
+        result.current.toggleRow('1')
+      })
+
+      expect(onSelectionChange).toHaveBeenCalledWith(expect.any(Set))
+      expect(onSelectionChange.mock.calls[0][0].has('1')).toBe(true)
+    })
+  })
+
+  describe('Toggle All (Current Page)', () => {
+    it('should select all rows on current page', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleAll()
+      })
+
+      expect(result.current.selectedIds.size).toBe(3)
+      expect(result.current.selectedIds.has('1')).toBe(true)
+      expect(result.current.selectedIds.has('2')).toBe(true)
+      expect(result.current.selectedIds.has('3')).toBe(true)
+    })
+
+    it('should deselect all rows on current page', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleAll()
+      })
+
+      expect(result.current.selectedIds.size).toBe(3)
+
+      act(() => {
+        result.current.toggleAll()
+      })
+
+      expect(result.current.selectedIds.size).toBe(0)
+    })
+
+    it('should select all when some are selected', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleRow('1')
+      })
+
+      expect(result.current.selectedIds.size).toBe(1)
+
+      act(() => {
+        result.current.toggleAll()
+      })
+
+      // Should select all since not all were selected
+      expect(result.current.selectedIds.size).toBe(3)
+    })
+
+    it('should deselect selectAllPages when toggling individual page', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(result.current.selectAllPages).toBe(true)
+
+      act(() => {
+        result.current.toggleAll()
+      })
+
+      expect(result.current.selectAllPages).toBe(false)
+    })
+  })
+
+  describe('Select All Pages', () => {
+    it('should toggle selectAllPages flag', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(result.current.selectAllPages).toBe(true)
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(result.current.selectAllPages).toBe(false)
+    })
+
+    it('should clear individual selections when selecting all pages', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleRow('1')
+        result.current.toggleRow('2')
+      })
+
+      expect(result.current.selectedIds.size).toBe(2)
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(result.current.selectAllPages).toBe(true)
+      expect(result.current.selectedIds.size).toBe(0)
+    })
+
+    it('should call onSelectAllPages callback', () => {
+      const onSelectAllPages = jest.fn()
+      const { result } = renderHook(() =>
+        useSelection({ ...defaultProps, onSelectAllPages })
+      )
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(onSelectAllPages).toHaveBeenCalledWith(true)
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(onSelectAllPages).toHaveBeenCalledWith(false)
+    })
+
+    it('should disable selectAllPages when toggling individual row', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(result.current.selectAllPages).toBe(true)
+
+      act(() => {
+        result.current.toggleRow('1')
+      })
+
+      expect(result.current.selectAllPages).toBe(false)
+    })
+  })
+
+  describe('Clear Selection', () => {
+    it('should clear all selections', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleRow('1')
+        result.current.toggleRow('2')
+      })
+
+      expect(result.current.selectedIds.size).toBe(2)
+
+      act(() => {
+        result.current.clearSelection()
+      })
+
+      expect(result.current.selectedIds.size).toBe(0)
+      expect(result.current.selectAllPages).toBe(false)
+    })
+
+    it('should clear selectAllPages flag', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(result.current.selectAllPages).toBe(true)
+
+      act(() => {
+        result.current.clearSelection()
+      })
+
+      expect(result.current.selectAllPages).toBe(false)
+    })
+
+    it('should call callbacks when clearing', () => {
+      const onSelectionChange = jest.fn()
+      const onSelectAllPages = jest.fn()
+      const { result } = renderHook(() =>
+        useSelection({ ...defaultProps, onSelectionChange, onSelectAllPages })
+      )
+
+      act(() => {
+        result.current.toggleRow('1')
+      })
+
+      onSelectionChange.mockClear()
+
+      act(() => {
+        result.current.clearSelection()
+      })
+
+      expect(onSelectionChange).toHaveBeenCalledWith(expect.any(Set))
+      expect(onSelectionChange.mock.calls[0][0].size).toBe(0)
+      expect(onSelectAllPages).toHaveBeenCalledWith(false)
+    })
+  })
+
+  describe('Get Selected Count', () => {
+    it('should return correct count for individual selections', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      expect(result.current.getSelectedCount()).toBe(0)
+
+      act(() => {
+        result.current.toggleRow('1')
+      })
+
+      expect(result.current.getSelectedCount()).toBe(1)
+
+      act(() => {
+        result.current.toggleRow('2')
+      })
+
+      expect(result.current.getSelectedCount()).toBe(2)
+    })
+
+    it('should return totalCount when selectAllPages is true', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.selectAllPagesToggle()
+      })
+
+      expect(result.current.getSelectedCount()).toBe(10) // totalCount
+    })
+  })
+
+  describe('Edge Cases', () => {
+    it('should handle empty current page data', () => {
+      const { result } = renderHook(() =>
+        useSelection({ ...defaultProps, currentPageData: [] })
+      )
+
+      act(() => {
+        result.current.toggleAll()
+      })
+
+      expect(result.current.selectedIds.size).toBe(0)
+    })
+
+    it('should handle selecting non-existent row', () => {
+      const { result } = renderHook(() => useSelection(defaultProps))
+
+      act(() => {
+        result.current.toggleRow('non-existent')
+      })
+
+      expect(result.current.selectedIds.has('non-existent')).toBe(true)
+      expect(result.current.selectedIds.size).toBe(1)
+    })
+
+    it('should maintain selection across page changes', () => {
+      const { result, rerender } = renderHook(
+        ({ data }) => useSelection({ ...defaultProps, currentPageData: data }),
+        { initialProps: { data: mockData } }
+      )
+
+      act(() => {
+        result.current.toggleRow('1')
+        result.current.toggleRow('2')
+      })
+
+      expect(result.current.selectedIds.size).toBe(2)
+
+      // Simulate page change
+      const newPageData = [
+        { id: '4', name: 'Item 4' },
+        { id: '5', name: 'Item 5' },
+      ]
+      rerender({ data: newPageData })
+
+      // Previous selections should be maintained
+      expect(result.current.selectedIds.size).toBe(2)
+      expect(result.current.selectedIds.has('1')).toBe(true)
+      expect(result.current.selectedIds.has('2')).toBe(true)
+    })
+  })
+})