Add test cases for all changes

Author: j-sanaa

packages/agentflow/src/atoms/Dropdown.test.tsx

@@ -0,0 +1,77 @@
+import { fireEvent, render, screen, waitFor } from '@testing-library/react'
+
+import { Dropdown, DropdownOption } from './Dropdown'
+
+const mockOnSelect = jest.fn()
+
+const options: DropdownOption[] = [
+    { label: 'GPT-4o', name: 'gpt-4o' },
+    { label: 'Claude 3', name: 'claude-3' },
+    { label: 'Gemini', name: 'gemini', description: 'Google model', imageSrc: 'http://test/gemini.png' }
+]
+
+beforeEach(() => {
+    jest.clearAllMocks()
+})
+
+describe('Dropdown', () => {
+    it('renders without crashing', () => {
+        render(<Dropdown value='' options={options} onSelect={mockOnSelect} />)
+
+        expect(screen.getByRole('combobox')).toBeInTheDocument()
+    })
+
+    it('displays options when opened', async () => {
+        render(<Dropdown value='' options={options} onSelect={mockOnSelect} />)
+
+        fireEvent.mouseDown(screen.getByRole('combobox'))
+        await waitFor(() => {
+            expect(screen.getByText('GPT-4o')).toBeInTheDocument()
+            expect(screen.getByText('Claude 3')).toBeInTheDocument()
+            expect(screen.getByText('Gemini')).toBeInTheDocument()
+        })
+    })
+
+    it('calls onSelect with option name when option is clicked', async () => {
+        render(<Dropdown value='' options={options} onSelect={mockOnSelect} />)
+
+        const input = screen.getByRole('combobox')
+        fireEvent.change(input, { target: { value: 'GPT' } })
+        await waitFor(() => screen.getByText('GPT-4o'))
+        fireEvent.click(screen.getByText('GPT-4o'))
+
+        expect(mockOnSelect).toHaveBeenCalledWith('gpt-4o')
+    })
+
+    it('shows image in option when imageSrc is present', async () => {
+        render(<Dropdown value='' options={options} onSelect={mockOnSelect} />)
+
+        fireEvent.mouseDown(screen.getByRole('combobox'))
+        await waitFor(() => {
+            const img = screen.getByAltText('Google model')
+            expect(img).toBeInTheDocument()
+            expect(img).toHaveAttribute('src', 'http://test/gemini.png')
+        })
+    })
+
+    it('shows description in option when present', async () => {
+        render(<Dropdown value='' options={options} onSelect={mockOnSelect} />)
+
+        fireEvent.mouseDown(screen.getByRole('combobox'))
+        await waitFor(() => {
+            expect(screen.getByText('Google model')).toBeInTheDocument()
+        })
+    })
+
+    it('renders in disabled state', () => {
+        render(<Dropdown value='' options={options} onSelect={mockOnSelect} disabled />)
+
+        expect(screen.getByRole('combobox')).toBeDisabled()
+    })
+
+    it('shows selected value label', () => {
+        render(<Dropdown value='gpt-4o' options={options} onSelect={mockOnSelect} />)
+
+        expect(screen.getByRole('combobox')).toHaveValue('GPT-4o')
+    })
+})

packages/agentflow/src/atoms/ScenariosInput.test.tsx

@@ -25,11 +25,14 @@ jest.mock('./NodeInputHandler', () => ({
 }))
 
 jest.mock('@tabler/icons-react', () => ({
-    IconInfoCircle: () => <span data-testid='icon-info' />,
     IconPlus: () => <span data-testid='icon-plus' />,
     IconTrash: () => <span data-testid='icon-trash' />
 }))
 
+jest.mock('./TooltipWithParser', () => ({
+    TooltipWithParser: ({ title }: { title: string }) => <span data-testid='tooltip-with-parser'>{title}</span>
+}))
+
 const scenarioInputParam: InputParam = {
     id: 'conditionAgentScenarios',
     name: 'conditionAgentScenarios',
@@ -61,7 +64,7 @@ describe('ScenariosInput', () => {
         const paramWithDesc: InputParam = { ...scenarioInputParam, description: 'Define scenarios for splitting' }
         render(<ScenariosInput inputParam={paramWithDesc} data={mockNodeData} onDataChange={mockOnDataChange} />)
 
-        expect(screen.getByTestId('icon-info')).toBeInTheDocument()
+        expect(screen.getByTestId('tooltip-with-parser')).toBeInTheDocument()
     })
 
     it('should render scenario items with "Scenario N" labels', () => {

packages/agentflow/src/atoms/StructuredOutputBuilder.test.tsx

@@ -9,11 +9,14 @@ const mockOnDataChange = jest.fn()
 
 jest.mock('@tabler/icons-react', () => ({
     IconArrowsMaximize: () => <span data-testid='icon-arrows-maximize' />,
-    IconInfoCircle: () => <span data-testid='icon-info-circle' />,
     IconPlus: () => <span data-testid='icon-plus' />,
     IconTrash: () => <span data-testid='icon-trash' />
 }))
 
+jest.mock('./TooltipWithParser', () => ({
+    TooltipWithParser: ({ title }: { title: string }) => <span data-testid='tooltip-with-parser'>{title}</span>
+}))
+
 jest.mock('./CodeInput', () => ({
     CodeInput: ({ value, onChange, language }: { value: string; onChange: (v: string) => void; language?: string }) => (
         <textarea data-testid='code-input' data-language={language} value={value} onChange={(e) => onChange(e.target.value)} />
@@ -366,7 +369,7 @@ describe('StructuredOutputBuilder', () => {
         render(<StructuredOutputBuilder inputParam={mockInputParam} data={dataWithEntries} onDataChange={mockOnDataChange} />)
 
         expect(screen.getByText('Enum Values')).toBeInTheDocument()
-        expect(screen.getByTestId('icon-info-circle')).toBeInTheDocument()
+        expect(screen.getByTestId('tooltip-with-parser')).toBeInTheDocument()
     })
 
     it('should render info icon and expand icon next to JSON Schema label', () => {
@@ -380,7 +383,7 @@ describe('StructuredOutputBuilder', () => {
         render(<StructuredOutputBuilder inputParam={mockInputParam} data={dataWithEntries} onDataChange={mockOnDataChange} />)
 
         expect(screen.getByText('JSON Schema')).toBeInTheDocument()
-        expect(screen.getByTestId('icon-info-circle')).toBeInTheDocument()
+        expect(screen.getByTestId('tooltip-with-parser')).toBeInTheDocument()
         expect(screen.getByTitle('Expand')).toBeInTheDocument()
     })
 

packages/agentflow/src/atoms/SwitchInput.test.tsx

@@ -0,0 +1,67 @@
+import { fireEvent, render, screen } from '@testing-library/react'
+
+import { SwitchInput } from './SwitchInput'
+
+const mockOnChange = jest.fn()
+
+beforeEach(() => {
+    jest.clearAllMocks()
+})
+
+describe('SwitchInput', () => {
+    it('renders switch element', () => {
+        render(<SwitchInput value={false} onChange={mockOnChange} />)
+
+        expect(screen.getByRole('checkbox')).toBeInTheDocument()
+    })
+
+    it('shows label when provided', () => {
+        render(<SwitchInput label='Enable feature' value={false} onChange={mockOnChange} />)
+
+        expect(screen.getByText('Enable feature')).toBeInTheDocument()
+    })
+
+    it('does not show label when not provided', () => {
+        const { container } = render(<SwitchInput value={false} onChange={mockOnChange} />)
+
+        expect(container.querySelector('.MuiTypography-root')).not.toBeInTheDocument()
+    })
+
+    it('switch is checked when value is true', () => {
+        render(<SwitchInput value={true} onChange={mockOnChange} />)
+
+        expect(screen.getByRole('checkbox')).toBeChecked()
+    })
+
+    it('switch is unchecked when value is false', () => {
+        render(<SwitchInput value={false} onChange={mockOnChange} />)
+
+        expect(screen.getByRole('checkbox')).not.toBeChecked()
+    })
+
+    it('calls onChange when toggled', () => {
+        render(<SwitchInput value={false} onChange={mockOnChange} />)
+
+        fireEvent.click(screen.getByRole('checkbox'))
+
+        expect(mockOnChange).toHaveBeenCalledWith(true)
+    })
+
+    it('switch is disabled when disabled prop is true', () => {
+        render(<SwitchInput value={false} onChange={mockOnChange} disabled={true} />)
+
+        expect(screen.getByRole('checkbox')).toBeDisabled()
+    })
+
+    it('handles string value coercion (truthy string → true)', () => {
+        render(<SwitchInput value='hello' onChange={mockOnChange} />)
+
+        expect(screen.getByRole('checkbox')).toBeChecked()
+    })
+
+    it('handles undefined value (defaults to false)', () => {
+        render(<SwitchInput value={undefined} onChange={mockOnChange} />)
+
+        expect(screen.getByRole('checkbox')).not.toBeChecked()
+    })
+})

packages/agentflow/src/atoms/TooltipWithParser.test.tsx

@@ -0,0 +1,34 @@
+import { render, screen } from '@testing-library/react'
+
+import { TooltipWithParser } from './TooltipWithParser'
+
+jest.mock('html-react-parser', () => {
+    const mockParser = (html: string) => html
+    mockParser.displayName = 'mockParser'
+    return { __esModule: true, default: mockParser }
+})
+
+describe('TooltipWithParser', () => {
+    it('renders info icon button', () => {
+        render(<TooltipWithParser title='Some tooltip' />)
+
+        expect(screen.getByRole('button')).toBeInTheDocument()
+        expect(screen.getByTestId('InfoIcon')).toBeInTheDocument()
+    })
+
+    it('renders tooltip with parsed HTML content', async () => {
+        render(<TooltipWithParser title='<b>Bold</b> text' />)
+
+        // Hover over the button to trigger the tooltip
+        const button = screen.getByRole('button')
+        button.focus()
+
+        expect(await screen.findByRole('tooltip')).toHaveTextContent('<b>Bold</b> text')
+    })
+
+    it('applies custom sx props', () => {
+        render(<TooltipWithParser title='Tooltip' sx={{ color: 'red' }} />)
+
+        expect(screen.getByTestId('InfoIcon')).toBeInTheDocument()
+    })
+})

packages/agentflow/src/core/utils/credentialDefaults.test.ts

@@ -0,0 +1,62 @@
+import type { CredentialSchemaInput } from '@/core/types'
+
+import { getDefaultValueForType } from './credentialDefaults'
+
+const makeInput = (overrides: Partial<CredentialSchemaInput> = {}): CredentialSchemaInput => ({
+    label: 'Test',
+    name: 'test',
+    type: 'string',
+    ...overrides
+})
+
+describe('getDefaultValueForType', () => {
+    it('returns input.default when defined', () => {
+        expect(getDefaultValueForType(makeInput({ type: 'string', default: 'custom' }))).toBe('custom')
+    })
+
+    it('returns input.default even when it is a falsy value', () => {
+        expect(getDefaultValueForType(makeInput({ type: 'number', default: 0 }))).toBe(0)
+        expect(getDefaultValueForType(makeInput({ type: 'boolean', default: false }))).toBe(false)
+        expect(getDefaultValueForType(makeInput({ type: 'string', default: '' }))).toBe('')
+    })
+
+    it('returns false for boolean', () => {
+        expect(getDefaultValueForType(makeInput({ type: 'boolean' }))).toBe(false)
+    })
+
+    it('returns 0 for number', () => {
+        expect(getDefaultValueForType(makeInput({ type: 'number' }))).toBe(0)
+    })
+
+    it("returns '{}' for json", () => {
+        expect(getDefaultValueForType(makeInput({ type: 'json' }))).toBe('{}')
+    })
+
+    it('returns first option name for options with options present', () => {
+        const input = makeInput({
+            type: 'options',
+            options: [
+                { label: 'First', name: 'first' },
+                { label: 'Second', name: 'second' }
+            ]
+        })
+        expect(getDefaultValueForType(input)).toBe('first')
+    })
+
+    it("returns '' for options with no options", () => {
+        expect(getDefaultValueForType(makeInput({ type: 'options' }))).toBe('')
+        expect(getDefaultValueForType(makeInput({ type: 'options', options: [] }))).toBe('')
+    })
+
+    it("returns '' for string", () => {
+        expect(getDefaultValueForType(makeInput({ type: 'string' }))).toBe('')
+    })
+
+    it("returns '' for password", () => {
+        expect(getDefaultValueForType(makeInput({ type: 'password' }))).toBe('')
+    })
+
+    it("returns '' for unknown type", () => {
+        expect(getDefaultValueForType(makeInput({ type: 'unknown' as CredentialSchemaInput['type'] }))).toBe('')
+    })
+})