mirror of
https://github.com/langgenius/dify.git
synced 2025-12-19 22:28:46 +00:00
174 lines
4.8 KiB
TypeScript
174 lines
4.8 KiB
TypeScript
import React from 'react'
|
|
import { act, render, screen } from '@testing-library/react'
|
|
import userEvent from '@testing-library/user-event'
|
|
import GotoAnything from './index'
|
|
import type { ActionItem, SearchResult } from './actions/types'
|
|
|
|
const routerPush = jest.fn()
|
|
jest.mock('next/navigation', () => ({
|
|
useRouter: () => ({
|
|
push: routerPush,
|
|
}),
|
|
usePathname: () => '/',
|
|
}))
|
|
|
|
const keyPressHandlers: Record<string, (event: any) => void> = {}
|
|
jest.mock('ahooks', () => ({
|
|
useDebounce: (value: any) => value,
|
|
useKeyPress: (keys: string | string[], handler: (event: any) => void) => {
|
|
const keyList = Array.isArray(keys) ? keys : [keys]
|
|
keyList.forEach((key) => {
|
|
keyPressHandlers[key] = handler
|
|
})
|
|
},
|
|
}))
|
|
|
|
const triggerKeyPress = (combo: string) => {
|
|
const handler = keyPressHandlers[combo]
|
|
if (handler) {
|
|
act(() => {
|
|
handler({ preventDefault: jest.fn(), target: document.body })
|
|
})
|
|
}
|
|
}
|
|
|
|
let mockQueryResult = { data: [] as SearchResult[], isLoading: false, isError: false, error: null as Error | null }
|
|
jest.mock('@tanstack/react-query', () => ({
|
|
useQuery: () => mockQueryResult,
|
|
}))
|
|
|
|
jest.mock('@/context/i18n', () => ({
|
|
useGetLanguage: () => 'en_US',
|
|
}))
|
|
|
|
const contextValue = { isWorkflowPage: false, isRagPipelinePage: false }
|
|
jest.mock('./context', () => ({
|
|
useGotoAnythingContext: () => contextValue,
|
|
GotoAnythingProvider: ({ children }: { children: React.ReactNode }) => <>{children}</>,
|
|
}))
|
|
|
|
const createActionItem = (key: ActionItem['key'], shortcut: string): ActionItem => ({
|
|
key,
|
|
shortcut,
|
|
title: `${key} title`,
|
|
description: `${key} desc`,
|
|
action: jest.fn(),
|
|
search: jest.fn(),
|
|
})
|
|
|
|
const actionsMock = {
|
|
slash: createActionItem('/', '/'),
|
|
app: createActionItem('@app', '@app'),
|
|
plugin: createActionItem('@plugin', '@plugin'),
|
|
}
|
|
|
|
const createActionsMock = jest.fn(() => actionsMock)
|
|
const matchActionMock = jest.fn(() => undefined)
|
|
const searchAnythingMock = jest.fn(async () => mockQueryResult.data)
|
|
|
|
jest.mock('./actions', () => ({
|
|
__esModule: true,
|
|
createActions: () => createActionsMock(),
|
|
matchAction: () => matchActionMock(),
|
|
searchAnything: () => searchAnythingMock(),
|
|
}))
|
|
|
|
jest.mock('./actions/commands', () => ({
|
|
SlashCommandProvider: () => null,
|
|
}))
|
|
|
|
jest.mock('./actions/commands/registry', () => ({
|
|
slashCommandRegistry: {
|
|
findCommand: () => null,
|
|
getAvailableCommands: () => [],
|
|
getAllCommands: () => [],
|
|
},
|
|
}))
|
|
|
|
jest.mock('@/app/components/workflow/utils/common', () => ({
|
|
getKeyboardKeyCodeBySystem: () => 'ctrl',
|
|
isEventTargetInputArea: () => false,
|
|
isMac: () => false,
|
|
}))
|
|
|
|
jest.mock('@/app/components/workflow/utils/node-navigation', () => ({
|
|
selectWorkflowNode: jest.fn(),
|
|
}))
|
|
|
|
jest.mock('../plugins/install-plugin/install-from-marketplace', () => (props: { manifest?: { name?: string }, onClose: () => void }) => (
|
|
<div data-testid="install-modal">
|
|
<span>{props.manifest?.name}</span>
|
|
<button onClick={props.onClose}>close</button>
|
|
</div>
|
|
))
|
|
|
|
describe('GotoAnything', () => {
|
|
beforeEach(() => {
|
|
routerPush.mockClear()
|
|
Object.keys(keyPressHandlers).forEach(key => delete keyPressHandlers[key])
|
|
mockQueryResult = { data: [], isLoading: false, isError: false, error: null }
|
|
matchActionMock.mockReset()
|
|
searchAnythingMock.mockClear()
|
|
})
|
|
|
|
it('should open modal via shortcut and navigate to selected result', async () => {
|
|
mockQueryResult = {
|
|
data: [{
|
|
id: 'app-1',
|
|
type: 'app',
|
|
title: 'Sample App',
|
|
description: 'desc',
|
|
path: '/apps/1',
|
|
icon: <div data-testid="icon">🧩</div>,
|
|
data: {},
|
|
} as any],
|
|
isLoading: false,
|
|
isError: false,
|
|
error: null,
|
|
}
|
|
|
|
render(<GotoAnything />)
|
|
|
|
triggerKeyPress('ctrl.k')
|
|
|
|
const input = await screen.findByPlaceholderText('app.gotoAnything.searchPlaceholder')
|
|
await userEvent.type(input, 'app')
|
|
|
|
const result = await screen.findByText('Sample App')
|
|
await userEvent.click(result)
|
|
|
|
expect(routerPush).toHaveBeenCalledWith('/apps/1')
|
|
})
|
|
|
|
it('should open plugin installer when selecting plugin result', async () => {
|
|
mockQueryResult = {
|
|
data: [{
|
|
id: 'plugin-1',
|
|
type: 'plugin',
|
|
title: 'Plugin Item',
|
|
description: 'desc',
|
|
path: '',
|
|
icon: <div />,
|
|
data: {
|
|
name: 'Plugin Item',
|
|
latest_package_identifier: 'pkg',
|
|
},
|
|
} as any],
|
|
isLoading: false,
|
|
isError: false,
|
|
error: null,
|
|
}
|
|
|
|
render(<GotoAnything />)
|
|
|
|
triggerKeyPress('ctrl.k')
|
|
const input = await screen.findByPlaceholderText('app.gotoAnything.searchPlaceholder')
|
|
await userEvent.type(input, 'plugin')
|
|
|
|
const pluginItem = await screen.findByText('Plugin Item')
|
|
await userEvent.click(pluginItem)
|
|
|
|
expect(await screen.findByTestId('install-modal')).toHaveTextContent('Plugin Item')
|
|
})
|
|
})
|