100 lines
3.8 KiB
TypeScript
100 lines
3.8 KiB
TypeScript
import { describe, it, expect } from 'vitest'
|
|
import fs from 'fs'
|
|
import os from 'os'
|
|
import path from 'path'
|
|
import { ErrorCode } from '@d3ro/core/errors'
|
|
import { getRAGService } from '../../src/main/services/RAGService'
|
|
import { getFileTranscriptionService } from '../../src/main/services/FileTranscriptionService'
|
|
import { useRedHarness } from './harness'
|
|
import { USER_TEXT } from './fixtures'
|
|
|
|
useRedHarness()
|
|
|
|
function writeTemp(name: string, body: string): string {
|
|
const p = path.join(os.tmpdir(), `d3ro-red-${Date.now()}-${name}`)
|
|
fs.writeFileSync(p, body, 'utf-8')
|
|
return p
|
|
}
|
|
|
|
describe('유스케이스: 지식베이스 문서 추가/삭제/질의 실패', () => {
|
|
it('초기 문서 목록은 비어 있다', () => {
|
|
expect(getRAGService().getDocuments()).toEqual([])
|
|
expect(getRAGService().getStateInfo().documentCount).toBe(0)
|
|
})
|
|
|
|
it('지원하지 않는 확장자는 RAGUnsupportedFormat 이다', async () => {
|
|
const p = writeTemp('x.bin', 'aaaa')
|
|
await expect(getRAGService().addDocument(p)).rejects.toMatchObject({
|
|
code: ErrorCode.RAGUnsupportedFormat,
|
|
})
|
|
})
|
|
|
|
it('너무 짧은 텍스트는 RAGIndexingFailed 다', async () => {
|
|
const p = writeTemp('short.txt', 'hi')
|
|
await expect(getRAGService().addDocument(p)).rejects.toMatchObject({
|
|
code: ErrorCode.RAGIndexingFailed,
|
|
})
|
|
})
|
|
|
|
it('충분한 txt 문서를 추가하면 목록에 나타난다', async () => {
|
|
const p = writeTemp('doc.txt', `${USER_TEXT.KO}\n`.repeat(5))
|
|
const doc = await getRAGService().addDocument(p)
|
|
expect(doc.fileType).toBe('txt')
|
|
expect(getRAGService().getDocuments().some((d) => d.id === doc.id)).toBe(true)
|
|
})
|
|
|
|
it('문서를 제거하면 목록에서 사라진다', async () => {
|
|
const p = writeTemp('rm.txt', `${USER_TEXT.EN}\n`.repeat(8))
|
|
const doc = await getRAGService().addDocument(p)
|
|
getRAGService().removeDocument(doc.id)
|
|
expect(getRAGService().getDocuments().some((d) => d.id === doc.id)).toBe(false)
|
|
})
|
|
|
|
it('없는 문서 재인덱스는 RAGDocumentNotFound 다', async () => {
|
|
await expect(getRAGService().reindex('missing')).rejects.toMatchObject({
|
|
code: ErrorCode.RAGDocumentNotFound,
|
|
})
|
|
})
|
|
|
|
it('청크가 없을 때 query 는 빈 답을 성공으로 위장하지 않는다', async () => {
|
|
await expect(getRAGService().query('무엇이든')).rejects.toMatchObject({
|
|
code: ErrorCode.RAGQueryFailed,
|
|
})
|
|
})
|
|
|
|
it('임베딩 서버가 없으면 인덱싱이 indexed=true + 0 chunks 로 성공 위장하지 않는다', async () => {
|
|
const p = writeTemp('emb.txt', `${USER_TEXT.KO}\n`.repeat(6))
|
|
const doc = await getRAGService().addDocument(p)
|
|
await new Promise((r) => setTimeout(r, 80))
|
|
const stored = getRAGService().getDocuments().find((d) => d.id === doc.id)
|
|
if (stored?.indexed) {
|
|
expect(stored.chunkCount).toBeGreaterThan(0)
|
|
} else {
|
|
expect(stored?.indexed).toBe(false)
|
|
}
|
|
})
|
|
})
|
|
|
|
describe('유스케이스: 파일 전사 입력 검증 / 취소', () => {
|
|
it('초기 상태는 idle 이다', () => {
|
|
expect(getFileTranscriptionService().state).toBe('idle')
|
|
expect(getFileTranscriptionService().getStateInfo().jobId).toBeNull()
|
|
})
|
|
|
|
it('지원하지 않는 포맷은 FileTranscriptionInvalidFormat 이다', async () => {
|
|
const p = writeTemp('x.txt', 'not audio')
|
|
await expect(getFileTranscriptionService().startTranscription(p)).rejects.toMatchObject({
|
|
code: ErrorCode.FileTranscriptionInvalidFormat,
|
|
})
|
|
})
|
|
|
|
it('없는 파일은 숨기지 않고 던진다', async () => {
|
|
await expect(
|
|
getFileTranscriptionService().startTranscription(path.join(os.tmpdir(), 'nope.wav')),
|
|
).rejects.toBeTruthy()
|
|
})
|
|
|
|
it('취소는 idle 이 아니어도 안전하다', () => {
|
|
expect(() => getFileTranscriptionService().cancel()).not.toThrow()
|
|
})
|
|
})
|