webnovel-writer/v7/test/state-machine/persist.test.js

import { test } from 'node:test'
import assert from 'node:assert/strict'
import os from 'node:os'
import path from 'node:path'
import { promises as fs } from 'node:fs'
import {
  persistRepair,
  persistCreateBook,
  persistVolumeReview,
  persistDraftOutline,
} from '../../src/state-machine/persist.js'

async function tmpRepo() {
  const root = await fs.mkdtemp(path.join(os.tmpdir(), 'wnw-persist-'))
  return { ctx: { repoPath: root }, root, cleanup: () => fs.rm(root, { recursive: true, force: true }) }
}
const read = (root, rel) => fs.readFile(path.join(root, rel), 'utf8')

test('persistDraftOutline（序6）→ 写 工作区/细纲.md', async () => {
  const { ctx, root, cleanup } = await tmpRepo()
  try {
    const r = await persistDraftOutline(ctx, { 细纲: '## 本章要写到的事\n林晚突破。\n' })
    assert.equal(r.ok, true)
    assert.match(await read(root, '工作区/细纲.md'), /林晚突破/)
  } finally { await cleanup() }
})

test('persistCreateBook（序1）→ 写 book.yaml + 总纲 + 第一卷卷纲', async () => {
  const { ctx, root, cleanup } = await tmpRepo()
  try {
    const r = await persistCreateBook(ctx, {
      book: { spec_version: '7.0', 书名: '剑起青云', 题材: '仙侠' },
      总纲: '# 总纲\n主角逆袭。',
      卷纲: '# 第1卷\n入门。',
    })
    assert.equal(r.ok, true)
    assert.match(await read(root, 'book.yaml'), /剑起青云/)
    assert.match(await read(root, '大纲/总纲.md'), /逆袭/)
    assert.match(await read(root, '大纲/第01卷.md'), /入门/)
  } finally { await cleanup() }
})

test('persistVolumeReview（序4）→ 写卷摘要 + 下卷卷纲', async () => {
  const { ctx, root, cleanup } = await tmpRepo()
  try {
    const r = await persistVolumeReview(ctx, { 卷号: 1, 卷摘要: '第一卷收束。', 下卷卷纲: '# 第2卷\n新地图。' })
    assert.equal(r.ok, true)
    assert.match(await read(root, '定稿/摘要/卷摘要/01.md'), /收束/)
    assert.match(await read(root, '大纲/第02卷.md'), /新地图/)
  } finally { await cleanup() }
})

test('persistRepair（序0）→ 仅写失败清单内的文件,内容须能解析', async () => {
  const { ctx, root, cleanup } = await tmpRepo()
  try {
    const target = '定稿/正文/0001-起.md'
    await fs.mkdir(path.join(root, '定稿/正文'), { recursive: true })
    await fs.writeFile(path.join(root, target), '---\n坏: yaml: :\n---\n正文', 'utf8')
    const good = '---\n章号: 1\n标题: 起\n---\n正文'
    const r = await persistRepair(ctx, { repairs: [{ file: target, content: good }] }, { allowedFiles: [target] })
    assert.equal(r.ok, true)
    assert.deepEqual(r.written, [target])
    assert.match(await read(root, target), /章号: 1/)
  } finally { await cleanup() }
})

test('persistRepair：拒绝写不在失败清单内的文件（安全网,防 AI 任意写）', async () => {
  const { ctx, root, cleanup } = await tmpRepo()
  try {
    const r = await persistRepair(
      ctx,
      { repairs: [{ file: '定稿/正文/9999-注入.md', content: '---\n章号: 9\n---\nx' }] },
      { allowedFiles: ['定稿/正文/0001-起.md'] }
    )
    assert.equal(r.ok, false)
    await assert.rejects(() => read(root, '定稿/正文/9999-注入.md'))
  } finally { await cleanup() }
})

test('persistRepair：修复内容仍解析失败 → ok=false 不写', async () => {
  const { ctx, root, cleanup } = await tmpRepo()
  try {
    const target = '定稿/正文/0001-起.md'
    const r = await persistRepair(
      ctx,
      { repairs: [{ file: target, content: '---\n仍坏: : :\n---\nx' }] },
      { allowedFiles: [target] }
    )
    assert.equal(r.ok, false)
  } finally { await cleanup() }
})