test(SegmentedControl): add test cases

web/app/components/base/segmented-control/index.css

@@ -52,8 +52,11 @@
   }
 
   .active {
-    @apply border-components-segmented-control-item-active-border bg-components-segmented-control-item-active-bg shadow-xs shadow-shadow-shadow-3
-    text-text-secondary
+    @apply border-components-segmented-control-item-active-border bg-components-segmented-control-item-active-bg shadow-xs shadow-shadow-shadow-3 text-text-secondary
+  }
+
+  .disabled {
+    @apply cursor-not-allowed text-text-disabled hover:text-text-disabled bg-transparent hover:bg-transparent
   }
 
   .active.accent {

web/app/components/base/segmented-control/index.spec.tsx

@@ -0,0 +1,98 @@
+import { fireEvent, render, screen } from '@testing-library/react'
+import '@testing-library/jest-dom'
+import SegmentedControl from './index'
+
+describe('SegmentedControl', () => {
+  const options = [
+    { value: 'option1', text: 'Option 1' },
+    { value: 'option2', text: 'Option 2' },
+    { value: 'option3', text: 'Option 3' },
+  ]
+
+  const optionsWithDisabled = [
+    { value: 'option1', text: 'Option 1' },
+    { value: 'option2', text: 'Option 2', disabled: true },
+    { value: 'option3', text: 'Option 3' },
+  ]
+
+  const onSelectMock = jest.fn((value: string | number | symbol) => value)
+
+  beforeEach(() => {
+    onSelectMock.mockClear()
+  })
+
+  it('renders all options correctly', () => {
+    render(<SegmentedControl options={options} value='option1' onChange={onSelectMock} />)
+
+    options.forEach((option) => {
+      expect(screen.getByText(option.text)).toBeInTheDocument()
+    })
+
+    const divider = screen.getByTestId('segmented-control-divider-1')
+    expect(divider).toBeInTheDocument()
+  })
+
+  it('renders with custom activeClassName when provided', () => {
+    render(
+      <SegmentedControl
+        options={options}
+        value='option1'
+        onChange={onSelectMock}
+        activeClassName='custom-active-class'
+      />,
+    )
+
+    const selectedOption = screen.getByText('Option 1').closest('button')
+    expect(selectedOption).toHaveClass('custom-active-class')
+  })
+
+  it('highlights the selected option', () => {
+    render(<SegmentedControl options={options} value='option2' onChange={onSelectMock} />)
+
+    const selectedOption = screen.getByText('Option 2').closest('button')
+    expect(selectedOption).toHaveClass('active')
+  })
+
+  it('calls onChange when an option is clicked', () => {
+    render(<SegmentedControl options={options} value='option1' onChange={onSelectMock} />)
+
+    fireEvent.click(screen.getByText('Option 3'))
+    expect(onSelectMock).toHaveBeenCalledWith('option3')
+  })
+
+  it('does not call onChange when clicking the already selected option', () => {
+    render(<SegmentedControl options={options} value='option1' onChange={onSelectMock} />)
+
+    fireEvent.click(screen.getByText('Option 1'))
+    expect(onSelectMock).not.toHaveBeenCalled()
+  })
+
+  it('handles disabled state correctly', () => {
+    render(<SegmentedControl options={optionsWithDisabled} value='option1' onChange={onSelectMock} />)
+
+    fireEvent.click(screen.getByText('Option 2'))
+    expect(onSelectMock).not.toHaveBeenCalled()
+
+    const optionElement = screen.getByText('Option 2').closest('button')
+    expect(optionElement).toHaveAttribute('disabled')
+    expect(optionElement).toHaveClass('disabled')
+
+    fireEvent.click(screen.getByText('Option 3'))
+    expect(onSelectMock).toHaveBeenCalledWith('option3')
+  })
+
+  it('renders with custom className when provided', () => {
+    const customClass = 'my-custom-class'
+    render(
+      <SegmentedControl
+        options={options}
+        value='option1'
+        onChange={onSelectMock}
+        className={customClass}
+      />,
+    )
+
+    const selectedOption = screen.getByText('Option 1').closest('button')?.closest('div')
+    expect(selectedOption).toHaveClass(customClass)
+  })
+})

web/app/components/base/segmented-control/index.tsx

@@ -9,8 +9,9 @@ import './index.css'
 type SegmentedControlOption<T> = {
   value: T
   text?: string
-  Icon: RemixiconComponentType
+  Icon?: RemixiconComponentType
   count?: number
+  disabled?: boolean
 }
 
 type SegmentedControlProps<T extends string | number | symbol> = {
@@ -90,9 +91,9 @@ export const SegmentedControl = <T extends string | number | symbol>({
   activeState,
   activeClassName,
 }: SegmentedControlProps<T>
-& VariantProps<typeof SegmentedControlVariants>
-& VariantProps<typeof SegmentedControlItemVariants>
-& VariantProps<typeof ItemTextWrapperVariants>) => {
+  & VariantProps<typeof SegmentedControlVariants>
+  & VariantProps<typeof SegmentedControlItemVariants>
+  & VariantProps<typeof ItemTextWrapperVariants>) => {
   const selectedOptionIndex = options.findIndex(option => option.value === value)
 
   return (
@@ -101,7 +102,7 @@ export const SegmentedControl = <T extends string | number | symbol>({
       className,
     )}>
       {options.map((option, index) => {
-        const { Icon, text, count } = option
+        const { Icon, text, count, disabled } = option
         const isSelected = index === selectedOptionIndex
         const isNextSelected = index === selectedOptionIndex - 1
         const isLast = index === options.length - 1
@@ -113,10 +114,15 @@ export const SegmentedControl = <T extends string | number | symbol>({
               isSelected ? 'active' : 'default',
               SegmentedControlItemVariants({ size, activeState: isSelected ? activeState : 'default' }),
               isSelected && activeClassName,
+              disabled && 'disabled',
             )}
-            onClick={() => onChange(option.value)}
+            onClick={() => {
+              if (!isSelected)
+                onChange(option.value)
+            }}
+            disabled={disabled}
           >
-            <Icon className='size-4 shrink-0' />
+            {Icon && <Icon className='size-4 shrink-0' />}
             {text && (
               <div className={cn('inline-flex items-center gap-x-1', ItemTextWrapperVariants({ size }))}>
                 <span>{text}</span>
@@ -128,7 +134,7 @@ export const SegmentedControl = <T extends string | number | symbol>({
               </div>
             )}
             {!isLast && !isSelected && !isNextSelected && (
-              <div className='absolute right-[-1px] top-0 flex h-full items-center'>
+              <div data-testid={`segmented-control-divider-${index}`} className='absolute right-[-1px] top-0 flex h-full items-center'>
                 <Divider type='vertical' className='mx-0 h-3.5' />
               </div>
             )}