77 lines
2.9 KiB
TypeScript
77 lines
2.9 KiB
TypeScript
|
// @vitest-environment jsdom
|
||
|
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
|
||
|
|
import { mount, flushPromises } from '@vue/test-utils'
|
||
|
|
import { ref } from 'vue'
|
||
|
|
|
||
|
|
const getAgentQueueItems = vi.fn()
|
||
|
|
const getKnowledgeByAgent = vi.fn()
|
||
|
|
|
||
|
|
vi.mock('@/services/knowledge', () => ({
|
||
|
|
knowledgeService: {
|
||
|
|
getAgentQueueItems: (...a: unknown[]) => getAgentQueueItems(...a),
|
||
|
|
getKnowledgeByAgent: (...a: unknown[]) => getKnowledgeByAgent(...a)
|
||
|
|
}
|
||
|
|
}))
|
||
|
|
vi.mock('@/services/widget', () => ({ widgetService: { createAgentWidget: vi.fn() } }))
|
||
|
|
vi.mock('@/composables/useOnboardingTestChat', () => ({
|
||
|
|
useOnboardingTestChat: () => ({
|
||
|
|
messages: ref([]), loading: ref(false), connecting: ref(false), connectionError: ref(''),
|
||
|
|
start: vi.fn(), send: vi.fn(), cleanup: vi.fn()
|
||
|
|
})
|
||
|
|
}))
|
||
|
|
|
||
|
|
import StepTest from '@/components/aiagent/onboarding/steps/StepTest.vue'
|
||
|
|
|
||
|
|
const SITE = 'https://teeckbox.com'
|
||
|
|
const REASON = 'This site could not be found — the domain does not resolve.'
|
||
|
|
const queue = (status: string) => ({
|
||
|
|
queue_items: [{ id: 1, source: SITE, source_type: 'website', status, created_at: '',
|
||
|
|
reason: status === 'failed' ? REASON : null }]
|
||
|
|
})
|
||
|
|
const knowledgePage = (count: number) => ({ knowledge: [], pagination: { total_count: count } })
|
||
|
|
|
||
|
|
const mountStep = async () => {
|
||
|
|
const wrapper = mount(StepTest, { props: { agentId: 'agent-1', agentName: 'Rex', widgetId: 'w-1' } })
|
||
|
|
await flushPromises()
|
||
|
|
return wrapper
|
||
|
|
}
|
||
|
|
|
||
|
|
describe('StepTest - what the agent knows', () => {
|
||
|
|
beforeEach(() => {
|
||
|
|
vi.clearAllMocks()
|
||
|
|
vi.useFakeTimers()
|
||
|
|
})
|
||
|
|
afterEach(() => vi.useRealTimers())
|
||
|
|
|
||
|
|
it('a crawl that fails after Teach says why, not "Knowledge indexed"', async () => {
|
||
|
|
getAgentQueueItems.mockResolvedValueOnce(queue('pending')).mockResolvedValue(queue('failed'))
|
||
|
|
getKnowledgeByAgent.mockResolvedValue(knowledgePage(0))
|
||
|
|
const wrapper = await mountStep()
|
||
|
|
expect(wrapper.text()).toContain('知识库仍在后台建立索引')
|
||
|
|
|
||
|
|
await vi.advanceTimersByTimeAsync(5000)
|
||
|
|
await flushPromises()
|
||
|
|
|
||
|
|
expect(wrapper.text()).not.toContain('知识库已索引')
|
||
|
|
expect(wrapper.text()).toContain(`无法读取 ${SITE}。${REASON}`)
|
||
|
|
expect(wrapper.text()).toContain('它还完全不了解你的业务')
|
||
|
|
})
|
||
|
|
|
||
|
|
it('a crawl that finishes says it is indexed', async () => {
|
||
|
|
getAgentQueueItems.mockResolvedValueOnce(queue('pending')).mockResolvedValue(queue('completed'))
|
||
|
|
getKnowledgeByAgent.mockResolvedValue(knowledgePage(1))
|
||
|
|
const wrapper = await mountStep()
|
||
|
|
|
||
|
|
await vi.advanceTimersByTimeAsync(5000)
|
||
|
|
await flushPromises()
|
||
|
|
|
||
|
|
expect(wrapper.text()).toContain('知识库已索引')
|
||
|
|
})
|
||
|
|
|
||
|
|
it('arriving after the failure shows it too', async () => {
|
||
|
|
getAgentQueueItems.mockResolvedValue(queue('failed'))
|
||
|
|
getKnowledgeByAgent.mockResolvedValue(knowledgePage(0))
|
||
|
|
const wrapper = await mountStep()
|
||
|
|
expect(wrapper.text()).toContain(`无法读取 ${SITE}。`)
|
||
|
|
})
|
||
|
|
})
|