youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 02:18:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: tests

Browse Source
This commit is contained in:
yyh
2026-03-11 20:28:37 +08:00
parent 5d07ccce59
commit 715a0fabfc
4 changed files with 90 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
web/app/components/base/prompt-editor/plugins/hitl-input-block/__tests__/variable-block.spec.tsx
View File

@ -33,7 +33,7 @@ const createWorkflowNodesMap = (title = 'Node One'): WorkflowNodesMap => ({
}) })
const hasErrorIcon = (container: HTMLElement) => { const hasErrorIcon = (container: HTMLElement) => {
return container.querySelector('svg.text-text-destructive') !== null return container.querySelector('svg.text-text-warning') !== null
} }
const renderVariableBlock = (props: { const renderVariableBlock = (props: {

8
web/app/components/datasets/settings/form/__tests__/index.spec.tsx
View File

@ -127,6 +127,14 @@ vi.mock('@/service/use-common', () => ({
], ],
}, },
}), }),
useCurrentWorkspace: () => ({
data: {
trial_credits: 1000,
trial_credits_used: 100,
next_credit_reset_date: undefined,
},
isPending: false,
}),
})) }))
vi.mock('@/app/components/header/account-setting/model-provider-page/hooks', () => ({ vi.mock('@/app/components/header/account-setting/model-provider-page/hooks', () => ({

15
web/app/components/datasets/settings/form/components/__tests__/indexing-section.spec.tsx
View File

@ -30,6 +30,21 @@ vi.mock('@/context/app-context', () => ({
}, },
})) }))
vi.mock('@/service/use-common', async () => {
const actual = await vi.importActual<typeof import('@/service/use-common')>('@/service/use-common')
return {
...actual,
useCurrentWorkspace: () => ({
data: {
trial_credits: 1000,
trial_credits_used: 100,
next_credit_reset_date: undefined,
},
isPending: false,
}),
}
})
// Mock model-provider-page hooks // Mock model-provider-page hooks
vi.mock('@/app/components/header/account-setting/model-provider-page/hooks', () => ({ vi.mock('@/app/components/header/account-setting/model-provider-page/hooks', () => ({
useModelList: () => ({ data: [], mutate: vi.fn(), isLoading: false }), useModelList: () => ({ data: [], mutate: vi.fn(), isLoading: false }),

165
web/app/components/plugins/plugin-detail-panel/model-selector/__tests__/index.spec.tsx
View File

@ -9,40 +9,6 @@ import ModelParameterModal from '../index'
// ==================== Mock Setup ==================== // ==================== Mock Setup ====================
// Mock shared state for portal
let mockPortalOpenState = false
vi.mock('@/app/components/base/portal-to-follow-elem', () => ({
PortalToFollowElem: ({ children, open }: { children: React.ReactNode, open: boolean }) => {
mockPortalOpenState = open || false
return (
<div data-testid="portal-elem" data-open={open}>
{children}
</div>
)
},
PortalToFollowElemTrigger: ({ children, onClick, className }: { children: React.ReactNode, onClick: () => void, className?: string }) => (
<div data-testid="portal-trigger" onClick={onClick} className={className}>
{children}
</div>
),
PortalToFollowElemContent: ({ children, className }: { children: React.ReactNode, className?: string }) => {
if (!mockPortalOpenState)
return null
return (
<div data-testid="portal-content" className={className}>
{children}
</div>
)
},
}))
vi.mock('@/app/components/base/toast', () => ({
default: {
notify: vi.fn(),
},
}))
// Mock provider context // Mock provider context
const mockProviderContextValue = { const mockProviderContextValue = {
isAPIKeySet: true, isAPIKeySet: true,
@ -87,6 +53,8 @@ vi.mock('@/utils/completion-params', () => ({
fetchAndMergeValidCompletionParams: (...args: unknown[]) => mockFetchAndMergeValidCompletionParams(...args), fetchAndMergeValidCompletionParams: (...args: unknown[]) => mockFetchAndMergeValidCompletionParams(...args),
})) }))
const mockToastNotify = vi.spyOn(Toast, 'notify')
// Mock child components // Mock child components
vi.mock('@/app/components/header/account-setting/model-provider-page/model-selector', () => ({ vi.mock('@/app/components/header/account-setting/model-provider-page/model-selector', () => ({
default: ({ defaultModel, modelList, scopeFeatures, onSelect }: { default: ({ defaultModel, modelList, scopeFeatures, onSelect }: {
@ -108,30 +76,33 @@ vi.mock('@/app/components/header/account-setting/model-provider-page/model-selec
})) }))
vi.mock('@/app/components/header/account-setting/model-provider-page/model-parameter-modal/trigger', () => ({ vi.mock('@/app/components/header/account-setting/model-provider-page/model-parameter-modal/trigger', () => ({
default: ({ disabled, hasDeprecated, modelDisabled, currentProvider, currentModel, providerName, modelId, isInWorkflow }: { default: ({ currentProvider, currentModel, providerName, modelId, isInWorkflow }: {
disabled?: boolean
hasDeprecated?: boolean
modelDisabled?: boolean
currentProvider?: Model currentProvider?: Model
currentModel?: ModelItem currentModel?: ModelItem
providerName?: string providerName?: string
modelId?: string modelId?: string
isInWorkflow?: boolean isInWorkflow?: boolean
}) => ( }) => {
<div const hasDeprecated = !currentProvider || !currentModel
data-testid="trigger" const modelDisabled = currentModel?.status !== ModelStatusEnum.active
data-disabled={disabled} const disabled = !mockProviderContextValue.isAPIKeySet || hasDeprecated || modelDisabled
data-has-deprecated={hasDeprecated}
data-model-disabled={modelDisabled} return (
data-provider={providerName} <div
data-model={modelId} data-testid="trigger"
data-in-workflow={isInWorkflow} data-disabled={disabled}
data-has-current-provider={!!currentProvider} data-has-deprecated={hasDeprecated}
data-has-current-model={!!currentModel} data-model-disabled={modelDisabled}
> data-provider={providerName}
Trigger data-model={modelId}
</div> data-in-workflow={isInWorkflow}
), data-has-current-provider={!!currentProvider}
data-has-current-model={!!currentModel}
>
Trigger
</div>
)
},
})) }))
vi.mock('@/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger', () => ({ vi.mock('@/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger', () => ({
@ -273,7 +244,7 @@ const setupModelLists = (config: {
describe('ModelParameterModal', () => { describe('ModelParameterModal', () => {
beforeEach(() => { beforeEach(() => {
vi.clearAllMocks() vi.clearAllMocks()
mockPortalOpenState = false mockToastNotify.mockReturnValue({})
mockProviderContextValue.isAPIKeySet = true mockProviderContextValue.isAPIKeySet = true
mockProviderContextValue.modelProviders = [] mockProviderContextValue.modelProviders = []
setupModelLists() setupModelLists()
@ -356,7 +327,7 @@ describe('ModelParameterModal', () => {
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
// Assert // Assert
expect(screen.queryByTestId('portal-content')).not.toBeInTheDocument() expect(screen.queryByTestId('model-selector')).not.toBeInTheDocument()
}) })
it('should render model selector inside portal content when open', async () => { it('should render model selector inside portal content when open', async () => {
@ -365,13 +336,12 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
expect(screen.getByTestId('portal-content')).toBeInTheDocument() expect(screen.getByTestId('model-selector')).toBeInTheDocument()
}) })
expect(screen.getByTestId('model-selector')).toBeInTheDocument()
}) })
}) })
@ -405,12 +375,11 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
const content = screen.getByTestId('portal-content') expect(document.querySelector('.custom-popup-class')).toBeInTheDocument()
expect(content.querySelector('.custom-popup-class')).toBeInTheDocument()
}) })
}) })
@ -422,7 +391,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
const selector = screen.getByTestId('model-selector') const selector = screen.getByTestId('model-selector')
@ -438,13 +407,13 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
expect(screen.queryByTestId('portal-content')).not.toBeInTheDocument() expect(screen.queryByTestId('model-selector')).not.toBeInTheDocument()
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
expect(screen.getByTestId('portal-content')).toBeInTheDocument() expect(screen.getByTestId('model-selector')).toBeInTheDocument()
}) })
}) })
@ -454,15 +423,15 @@ describe('ModelParameterModal', () => {
// Act // Act
const { rerender } = render(<ModelParameterModal {...props} />) const { rerender } = render(<ModelParameterModal {...props} />)
expect(screen.getByTestId('portal-elem')).toHaveAttribute('data-open', 'false') expect(screen.queryByTestId('model-selector')).not.toBeInTheDocument()
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Force a re-render to ensure state is stable // Force a re-render to ensure state is stable
rerender(<ModelParameterModal {...props} />) rerender(<ModelParameterModal {...props} />)
// Assert - open state should remain false due to readonly // Assert - open state should remain false due to readonly
expect(screen.getByTestId('portal-elem')).toHaveAttribute('data-open', 'false') expect(screen.queryByTestId('model-selector')).not.toBeInTheDocument()
}) })
}) })
@ -474,7 +443,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -489,7 +458,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -512,7 +481,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -530,7 +499,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -547,7 +516,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -564,7 +533,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -581,7 +550,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -598,7 +567,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -615,7 +584,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -632,7 +601,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -831,7 +800,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
await waitFor(() => { await waitFor(() => {
fireEvent.click(screen.getByTestId('model-selector')) fireEvent.click(screen.getByTestId('model-selector'))
@ -856,7 +825,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
await waitFor(() => { await waitFor(() => {
fireEvent.click(screen.getByTestId('model-selector')) fireEvent.click(screen.getByTestId('model-selector'))
@ -888,7 +857,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
await waitFor(() => { await waitFor(() => {
fireEvent.click(screen.getByTestId('model-selector')) fireEvent.click(screen.getByTestId('model-selector'))
@ -915,7 +884,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
await waitFor(() => { await waitFor(() => {
fireEvent.click(screen.getByTestId('model-selector')) fireEvent.click(screen.getByTestId('model-selector'))
@ -951,7 +920,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
await waitFor(() => { await waitFor(() => {
const panel = screen.getByTestId('llm-params-panel') const panel = screen.getByTestId('llm-params-panel')
@ -988,7 +957,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
await waitFor(() => { await waitFor(() => {
const panel = screen.getByTestId('tts-params-panel') const panel = screen.getByTestId('tts-params-panel')
@ -1025,7 +994,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -1051,7 +1020,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -1077,7 +1046,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -1104,12 +1073,11 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
const content = screen.getByTestId('portal-content') expect(document.querySelector('.bg-divider-subtle')).toBeInTheDocument()
expect(content.querySelector('.bg-divider-subtle')).toBeInTheDocument()
}) })
}) })
}) })
@ -1146,7 +1114,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -1185,7 +1153,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -1264,7 +1232,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert // Assert
await waitFor(() => { await waitFor(() => {
@ -1280,7 +1248,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert - defaultModel is created with undefined provider // Assert - defaultModel is created with undefined provider
await waitFor(() => { await waitFor(() => {
@ -1297,7 +1265,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert - defaultModel is created with undefined model // Assert - defaultModel is created with undefined model
await waitFor(() => { await waitFor(() => {
@ -1314,7 +1282,7 @@ describe('ModelParameterModal', () => {
// Act // Act
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
// Assert - when defaultModel is undefined, attribute is not set (returns null) // Assert - when defaultModel is undefined, attribute is not set (returns null)
await waitFor(() => { await waitFor(() => {
@ -1350,14 +1318,13 @@ describe('ModelParameterModal', () => {
// Act // Act
const { rerender } = render(<ModelParameterModal {...props} />) const { rerender } = render(<ModelParameterModal {...props} />)
fireEvent.click(screen.getByTestId('portal-trigger')) fireEvent.click(screen.getByTestId('trigger'))
await waitFor(() => { await waitFor(() => {
expect(screen.getByTestId('model-selector')).toHaveAttribute('data-model-list-count', '1') expect(screen.getByTestId('model-selector')).toHaveAttribute('data-model-list-count', '1')
}) })
// Rerender with different scope // Rerender with different scope
mockPortalOpenState = true
rerender(<ModelParameterModal {...props} scope={ModelTypeEnum.textEmbedding} />) rerender(<ModelParameterModal {...props} scope={ModelTypeEnum.textEmbedding} />)
// Assert // Assert
@ -1398,7 +1365,7 @@ describe('ModelParameterModal', () => {
render(<ModelParameterModal {...props} />) render(<ModelParameterModal {...props} />)
// Assert // Assert
const trigger = screen.getByTestId('portal-trigger') const trigger = screen.getByTestId('trigger')
expect(trigger).toBeInTheDocument() expect(trigger).toBeInTheDocument()
}) })
}) })