124 lines
3.9 KiB
TypeScript
124 lines
3.9 KiB
TypeScript
import { render, screen } from '@testing-library/react'
|
|
import userEvent from '@testing-library/user-event'
|
|
import { describe, it, expect, vi, beforeEach } from 'vitest'
|
|
import { ModelTable } from '@/pages/Providers/ModelTable'
|
|
import type { Model } from '@/types'
|
|
|
|
const mockModels: Model[] = [
|
|
{
|
|
id: 'model-1',
|
|
providerId: 'openai',
|
|
modelName: 'gpt-4o',
|
|
enabled: true,
|
|
createdAt: '2024-01-01T00:00:00Z',
|
|
unifiedId: 'openai/gpt-4o',
|
|
},
|
|
{
|
|
id: 'model-2',
|
|
providerId: 'openai',
|
|
modelName: 'gpt-3.5-turbo',
|
|
enabled: false,
|
|
createdAt: '2024-01-02T00:00:00Z',
|
|
unifiedId: 'openai/gpt-3.5-turbo',
|
|
},
|
|
]
|
|
|
|
const mockMutate = vi.fn()
|
|
|
|
vi.mock('@/hooks/useModels', () => ({
|
|
useModels: vi.fn((providerId: string) => {
|
|
if (providerId === 'openai') {
|
|
return { data: mockModels, isLoading: false }
|
|
}
|
|
return { data: [], isLoading: false }
|
|
}),
|
|
useDeleteModel: vi.fn(() => ({ mutate: mockMutate })),
|
|
}))
|
|
|
|
const defaultProps = {
|
|
providerId: 'openai',
|
|
onAdd: vi.fn(),
|
|
onEdit: vi.fn(),
|
|
}
|
|
|
|
describe('ModelTable', () => {
|
|
beforeEach(() => {
|
|
mockMutate.mockClear()
|
|
})
|
|
|
|
it('renders model list with unified ID and model name', () => {
|
|
render(<ModelTable {...defaultProps} />)
|
|
|
|
expect(screen.getByText(/关联模型/)).toBeInTheDocument()
|
|
expect(screen.getByText('openai/gpt-4o')).toBeInTheDocument()
|
|
expect(screen.getByText('openai/gpt-3.5-turbo')).toBeInTheDocument()
|
|
expect(screen.getByText('gpt-4o')).toBeInTheDocument()
|
|
expect(screen.getByText('gpt-3.5-turbo')).toBeInTheDocument()
|
|
})
|
|
|
|
it('renders status tags correctly', () => {
|
|
render(<ModelTable {...defaultProps} />)
|
|
|
|
const enabledTags = screen.getAllByText('启用')
|
|
const disabledTags = screen.getAllByText('禁用')
|
|
expect(enabledTags.length).toBeGreaterThanOrEqual(1)
|
|
expect(disabledTags.length).toBeGreaterThanOrEqual(1)
|
|
})
|
|
|
|
it('calls onAdd when clicking "添加模型" button', async () => {
|
|
const user = userEvent.setup()
|
|
const onAdd = vi.fn()
|
|
render(<ModelTable {...defaultProps} onAdd={onAdd} />)
|
|
|
|
await user.click(screen.getByRole('button', { name: '添加模型' }))
|
|
expect(onAdd).toHaveBeenCalledTimes(1)
|
|
})
|
|
|
|
it('calls onEdit with correct model when clicking "编辑"', async () => {
|
|
const user = userEvent.setup()
|
|
const onEdit = vi.fn()
|
|
render(<ModelTable {...defaultProps} onEdit={onEdit} />)
|
|
|
|
const editButtons = screen.getAllByRole('button', { name: /编 ?辑/ })
|
|
await user.click(editButtons[0])
|
|
|
|
expect(onEdit).toHaveBeenCalledTimes(1)
|
|
expect(onEdit).toHaveBeenCalledWith(mockModels[0])
|
|
})
|
|
|
|
it('calls deleteModel.mutate with correct model ID when delete is confirmed', async () => {
|
|
const user = userEvent.setup()
|
|
|
|
render(<ModelTable {...defaultProps} />)
|
|
|
|
// Find and click the delete button for the first row
|
|
const deleteButtons = screen.getAllByRole('button', { name: '删除' })
|
|
await user.click(deleteButtons[0])
|
|
|
|
// TDesign Popconfirm renders confirmation popup with "确定" button
|
|
const confirmButton = await screen.findByRole('button', { name: '确定' })
|
|
await user.click(confirmButton)
|
|
|
|
// Assert that deleteModel.mutate was called with the correct model ID
|
|
expect(mockMutate).toHaveBeenCalledTimes(1)
|
|
expect(mockMutate).toHaveBeenCalledWith('model-1')
|
|
}, 10000)
|
|
|
|
it('shows custom empty text when models list is empty', () => {
|
|
render(<ModelTable providerId='anthropic' />)
|
|
expect(screen.getByText('暂无模型，点击上方按钮添加')).toBeInTheDocument()
|
|
})
|
|
|
|
it('does not render add button when onAdd is not provided', () => {
|
|
render(<ModelTable providerId='openai' />)
|
|
|
|
expect(screen.queryByRole('button', { name: '添加模型' })).not.toBeInTheDocument()
|
|
})
|
|
|
|
it('does not render edit button when onEdit is not provided', () => {
|
|
render(<ModelTable providerId='openai' onAdd={vi.fn()} />)
|
|
|
|
expect(screen.queryByRole('button', { name: /编 ?辑/ })).not.toBeInTheDocument()
|
|
})
|
|
})
|