42 lines
1.7 KiB
TypeScript
42 lines
1.7 KiB
TypeScript
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
|
|
import { DatabaseSync } from 'node:sqlite'
|
|
import { migrate } from '../../src/main/db/migrate'
|
|
import { VolumeRepository } from '../../src/main/db/repositories/volume.repo'
|
|
import { AiSessionRepository } from '../../src/main/db/repositories/ai-session.repo'
|
|
import { InteractiveRepository } from '../../src/main/db/repositories/interactive.repo'
|
|
import { ChapterRepository } from '../../src/main/db/repositories/chapter.repo'
|
|
import { InteractiveWritingService } from '../../src/main/services/interactive-writing.service'
|
|
import type { SqliteDb } from '../../src/main/db/types'
|
|
|
|
describe('InteractiveWritingService.finishChapter', () => {
|
|
let db: SqliteDb
|
|
let service: InteractiveWritingService
|
|
let volId: string
|
|
let flowId: string
|
|
|
|
beforeEach(() => {
|
|
db = new DatabaseSync(':memory:')
|
|
migrate(db)
|
|
volId = new VolumeRepository(db).create('卷一', 0).id
|
|
const session = new AiSessionRepository(db).create('测试', 'test-model')
|
|
const flow = new InteractiveRepository(db).create(session.id)
|
|
flowId = flow.id
|
|
const repo = new InteractiveRepository(db)
|
|
repo.addScene(flowId, '<p>场景一</p>', 'A')
|
|
repo.addScene(flowId, '<p>场景二</p>', 'B')
|
|
service = new InteractiveWritingService(db, {
|
|
chat: async () => ''
|
|
} as never)
|
|
})
|
|
|
|
afterEach(() => db.close())
|
|
|
|
it('IT-INT-04: merges scenes into new interactive chapter', () => {
|
|
const { chapterId } = service.finishChapter(flowId, volId, '交互初稿')
|
|
const ch = new ChapterRepository(db).get(chapterId)!
|
|
expect(ch.origin).toBe('interactive')
|
|
expect(ch.content).toContain('场景一')
|
|
expect(ch.content).toContain('场景二')
|
|
})
|
|
})
|