mirror of
https://gitee.com/mateos/mateclaw.git
synced 2026-11-04 06:55:32 +08:00
51 lines
2.5 KiB
TypeScript
51 lines
2.5 KiB
TypeScript
import { createApp, defineComponent, nextTick } from 'vue'
|
|
import { createI18n } from 'vue-i18n'
|
|
import { afterEach, expect, it, vi } from 'vitest'
|
|
import zhCN from '@/i18n/locales/zh-CN'
|
|
import { agentApi, agentBindingApi } from '@/api'
|
|
import AgentCreateWizard from '../AgentCreateWizard.vue'
|
|
|
|
vi.mock('vue-router', () => ({ useRouter: () => ({ push: vi.fn() }) }))
|
|
vi.mock('@/api', () => ({
|
|
agentApi: { generate: vi.fn(), create: vi.fn() },
|
|
agentBindingApi: { setSkills: vi.fn(), setTools: vi.fn() },
|
|
toolApi: { listAvailable: vi.fn().mockResolvedValue({ data: [] }) },
|
|
skillApi: { page: vi.fn().mockResolvedValue({ data: [] }) },
|
|
wikiApi: { listBindableKBs: vi.fn().mockResolvedValue({ data: [] }) },
|
|
}))
|
|
vi.mock('@/composables/useMcToast', () => ({ mcToast: { error: vi.fn(), success: vi.fn() } }))
|
|
vi.mock('@/components/common/SkillIcon.vue', () => ({ default: defineComponent({ template: '<span />' }) }))
|
|
vi.mock('@/components/common/SkillIconPicker.vue', () => ({ default: defineComponent({ template: '<span />' }) }))
|
|
vi.mock('@/components/agent/WizardCapabilityPicker.vue', () => ({ default: defineComponent({ template: '<span />' }) }))
|
|
|
|
let app: ReturnType<typeof createApp> | undefined
|
|
afterEach(() => {
|
|
app?.unmount()
|
|
document.body.innerHTML = ''
|
|
vi.clearAllMocks()
|
|
})
|
|
|
|
it('creates the generated agent with a 100-round budget so complex tasks can exceed ten rounds', async () => {
|
|
vi.mocked(agentApi.generate).mockResolvedValue({ data: { name: 'Research assistant' } } as never)
|
|
vi.mocked(agentApi.create).mockResolvedValue({ data: { id: '123', name: 'Research assistant' } } as never)
|
|
vi.mocked(agentBindingApi.setSkills).mockResolvedValue({} as never)
|
|
vi.mocked(agentBindingApi.setTools).mockResolvedValue({} as never)
|
|
const host = document.createElement('div')
|
|
document.body.appendChild(host)
|
|
app = createApp(AgentCreateWizard)
|
|
app.use(createI18n({ legacy: false, locale: 'zh-CN', messages: { 'zh-CN': zhCN } }))
|
|
app.mount(host)
|
|
|
|
const input = host.querySelector('textarea')!
|
|
input.value = 'Research a complex task'
|
|
input.dispatchEvent(new Event('input', { bubbles: true }))
|
|
await nextTick()
|
|
host.querySelector<HTMLButtonElement>('.wiz-btn-primary')!.click()
|
|
await vi.waitFor(() => expect(host.querySelector('.wiz-notice')).not.toBeNull())
|
|
host.querySelector<HTMLButtonElement>('.wiz-actions .wiz-btn-primary')!.click()
|
|
|
|
await vi.waitFor(() => expect(agentApi.create).toHaveBeenCalledWith(expect.objectContaining({
|
|
name: 'Research assistant', maxIterations: 100,
|
|
})))
|
|
})
|