opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-09 21:28:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(plugins): mock stop all task hook

Browse Source 
Update the new plugin task hook spec so CI web tests include the Stop All mutation mock.
This commit is contained in:
sawyer-shi 2026-05-08 15:30:00 +08:00
parent a4c68ee1fa
commit 6acafc3f52
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
web/app/components/plugins/plugin-page/plugin-tasks/__tests__/hooks.spec.ts
View File

@ -4,6 +4,7 @@ import { TaskStatus } from '@/app/components/plugins/types'
import { usePluginTaskStatus } from '../hooks'
const mockClearTask = vi.fn().mockResolvedValue({})
const mockStopAllTask = vi.fn().mockResolvedValue({})
const mockRefetch = vi.fn()
vi.mock('@/service/use-plugins', () => ({
@ -28,6 +29,9 @@ vi.mock('@/service/use-plugins', () => ({
useMutationClearTaskPlugin: () => ({
mutateAsync: mockClearTask,
}),
useMutationStopAllTaskPlugins: () => ({
mutateAsync: mockStopAllTask,
}),
}))
describe('usePluginTaskStatus', () => {
@ -74,4 +78,13 @@ describe('usePluginTaskStatus', () => {
})
expect(mockRefetch).toHaveBeenCalled()
})
it('should handle stop all plugins', async () => {
const { result } = renderHook(() => usePluginTaskStatus())
await result.current.handleStopAllPlugins()
expect(mockStopAllTask).toHaveBeenCalled()
expect(mockRefetch).toHaveBeenCalled()
})
})