jsc-anima/hooks/routing.test.ts

import { expect, mock, test } from 'claude-code/testing'

type Ctl = { complete: (e: any) => unknown; usage: unknown; seen?: any; submitted: string[] }

const setup = (on: any, kv?: Map<string, unknown>): Ctl => {
  const ctl: Ctl = { complete: () => ({ isAnswered: false, reason: 'aborted' }), usage: { rateLimits: [], context: {} }, submitted: [] }
  if (kv) {
    // 自備 store 以便檢查寫入內容。
    on('store.get', (_$: any, e: any) => ({ value: kv.get(e.key) }) as never)
    on('store.set', (_$: any, e: any) => (kv.set(e.key, e.value), { value: undefined }) as never)
  } else mock.store(on)
  mock.clock(on, { now: Date.parse('2026-10-15T08:00:00Z') })
  on('ui.log', () => ({ value: undefined }) as never)
  on('command.register', () => ({ value: { name: 'x' } }) as never)
  on('session.start', () => ({ cwd: '/tmp' }) as never)
  on('command.run', () => ({ text: '' }))
  on('prompt.compose', () => ({ sections: [] }))
  on('prompt.submit', (_$: any, e: any) => (ctl.submitted.push(e.text), { text: e.text }) as never)
  on('model.complete', (_$: any, e: any) => ({ value: ctl.complete(e) }) as never)
  on('session.usage', () => ({ value: ctl.usage }) as never)
  on('turn.step', async function* (_$: any, e: any) {
    ctl.seen = e
    return { turnId: e.turnId, index: 0, answer: '', toolUses: [], stopReason: null, usage: null } as never
  })
  return ctl
}

const start = async ($: any, slug = 'sao-asuna') => {
  await $.session.start({ source: 'startup', cwd: '/tmp' } as never)
  await $.command.run({ command: slug, args: '' } as never)
}

const affinity = async ($: any): Promise<number> => {
  const r: any = await $.command.run({ command: 'anima-affinity', args: '' } as never)
  return Number(/好感度 (\d+)/.exec(r.text)?.[1])
}

const submit = ($: any, text: string) => $.prompt.submit({ text } as never)
const step = async ($: any, model: string) => {
  const s = $.turn.step({ turnId: 't', index: 0, model, messageCount: 1 } as never)
  for await (const _ of s) void _
}
const usage = { input_tokens: 1, output_tokens: 1 }
const route = (model: string, tone = 'neutral') => ({ isAnswered: true, usage, text: `{"difficulty":9,"model":"${model}","effort":"high","tone":"${tone}","ack":"好，我來看看。"}` })
const row = (text: string) => ({ message: { type: 'assistant', role: 'assistant', content: [{ type: 'text', text }] }, door: 'response', origin: { kind: 'model', model: 'm' }, uuid: 'u' }) as never

test('輸入由 Haiku 一次分類，並依分類覆寫 turn.step 的模型與思考程度', async ($, on) => {
  const ctl = setup(on)
  const asked: string[] = []
  ctl.complete = e => (asked.push(e.model), route('opus'))
  await start($)
  await submit($, '請重構整個模組')
  await step($, 'sonnet')
  expect(asked).toEqual(['haiku'])
  expect(ctl.seen.model).toBe('opus')
  expect(ctl.seen.effort).toBe('high')
})

test('額度低於門檻自動降一級', async ($, on) => {
  const ctl = setup(on)
  ctl.complete = () => route('opus')
  ctl.usage = { rateLimits: [{ kind: 'five_hour', percentUsed: 95 }], context: {} }
  await start($)
  await submit($, 'x')
  await step($, 'opus')
  expect(ctl.seen.model).toBe('sonnet')
})

test('手動鎖定模型與思考程度', { options: { lockModel: 'haiku', lockEffort: 'low' } }, async ($, on) => {
  const ctl = setup(on)
  ctl.complete = () => route('opus')
  await start($)
  await submit($, 'x')
  await step($, 'opus')
  expect([ctl.seen.model, ctl.seen.effort]).toEqual(['haiku', 'low'])
})

test('回覆改寫：保留程式碼、路徑、URL、數字；前綴只加在開頭一次', async ($, on) => {
  const ctl = setup(on)
  ctl.complete = e => ({ isAnswered: true, usage, text: String(e.prompt).replace('完成了', '做好囉') })
  await start($)
  const src = '完成了，改了 `npm test` 與 /src/a/b.ts，見 https://x.dev/a，共 42 處。'
  const r1: any = await $.session.append(row(src))
  expect(r1.message.content[0].text).toBe('亞絲娜 > 做好囉，改了 `npm test` 與 /src/a/b.ts，見 https://x.dev/a，共 42 處。')
  const r2: any = await $.session.append(row('第二段 7'))
  expect(r2.message.content[0].text.startsWith('亞絲娜 >')).toBe(false)
})

test('改寫失敗或遺失佔位符時顯示原文', async ($, on) => {
  const ctl = setup(on)
  await start($)
  const r1: any = await $.session.append(row('原文 `x` 3'))
  expect(r1.message.content[0].text).toBe('亞絲娜 > 原文 `x` 3')
  ctl.complete = () => ({ isAnswered: true, usage, text: '丟了佔位符' })
  const r2: any = await $.session.append(row('原文 `y` 5'))
  expect(r2.message.content[0].text).toBe('原文 `y` 5')
})

test('改寫模型：長回覆用 Sonnet，一般回覆用 Haiku', async ($, on) => {
  const ctl = setup(on)
  const asked: string[] = []
  ctl.complete = e => (asked.push(e.model), { isAnswered: true, usage, text: String(e.prompt) })
  await start($)
  await $.session.append(row('短回覆'))
  await $.session.append(row('長'.repeat(1300)))
  expect(asked).toEqual(['haiku', 'sonnet'])
})

test('longReply 可調整長回覆門檻', { options: { longReply: 10 } }, async ($, on) => {
  const ctl = setup(on)
  const asked: string[] = []
  ctl.complete = e => (asked.push(e.model), { isAnswered: true, usage, text: String(e.prompt) })
  await start($)
  await $.session.append(row('十一個字的回覆內容喔喔'))
  await $.session.append(row('短'))
  expect(asked).toEqual(['sonnet', 'haiku'])
})

test('改寫逾時、額度不足或拋錯時直接顯示原文', async ($, on) => {
  const ctl = setup(on)
  await start($)
  ctl.complete = () => ({ isAnswered: false, reason: 'timeout' })
  const r1: any = await $.session.append(row('逾時 1'))
  expect(r1.message.content[0].text).toBe('亞絲娜 > 逾時 1')
  ctl.complete = () => ({ isAnswered: false, reason: 'rate_limited' })
  const r2: any = await $.session.append(row('額度 2'))
  expect(r2.message.content[0].text).toBe('額度 2')
  ctl.complete = () => { throw new Error('quota') }
  const r3: any = await $.session.append(row('拋錯 3'))
  expect(r3.message.content[0].text).toBe('拋錯 3')
})

test('改寫逾時設定傳給模型呼叫', { options: { rewriteTimeoutSec: 5 } }, async ($, on) => {
  const ctl = setup(on)
  let timeout = 0
  ctl.complete = e => (timeout = e.timeoutMs, { isAnswered: false, reason: 'timeout' })
  await start($)
  await $.session.append(row('x'))
  expect(timeout).toBe(5000)
})

const rawLog = (kv: Map<string, unknown>): string[] => ((kv.get('rawlog') as { raw: string }[] | undefined) ?? []).map(x => x.raw)

test('原文 log 預設寫入，改寫失敗時不寫', async ($, on) => {
  const kv = new Map<string, unknown>()
  const ctl = setup(on, kv)
  await start($)
  ctl.complete = e => ({ isAnswered: true, usage, text: String(e.prompt) })
  await $.session.append(row('原文甲'))
  ctl.complete = () => ({ isAnswered: false, reason: 'timeout' })
  await $.session.append(row('原文乙'))
  expect(rawLog(kv)).toEqual(['原文甲'])
})

test('在 /config 關閉原文 log 時不寫入', { options: { rawLog: false } }, async ($, on) => {
  const kv = new Map<string, unknown>()
  const ctl = setup(on, kv)
  ctl.complete = e => ({ isAnswered: true, usage, text: String(e.prompt) })
  await start($)
  const r: any = await $.session.append(row('原文丙'))
  expect(r.message.content[0].text).toBe('亞絲娜 > 原文丙')
  expect(rawLog(kv)).toEqual([])
})

test('前綴只加在整則回覆開頭一次，多段文字區塊與多次寫入都不重複', async ($, on) => {
  const ctl = setup(on)
  ctl.complete = e => ({ isAnswered: true, usage, text: String(e.prompt) })
  await start($)
  const multi = { message: { type: 'assistant', role: 'assistant', content: [{ type: 'text', text: '第一段' }, { type: 'tool_use', id: 't', name: 'x', input: {} }, { type: 'text', text: '第二段' }] }, door: 'response', origin: { kind: 'model', model: 'm' }, uuid: 'u' } as never
  const r: any = await $.session.append(multi)
  expect(r.message.content[0].text).toBe('亞絲娜 > 第一段')
  expect(r.message.content[2].text).toBe('第二段')
  const r2: any = await $.session.append(row('同一回覆後續'))
  expect(r2.message.content[0].text).toBe('同一回覆後續')
})

test('新的使用者輸入後，下一則回覆重新加前綴', async ($, on) => {
  const ctl = setup(on)
  ctl.complete = e => ({ isAnswered: true, usage, text: String(e.prompt) })
  await start($)
  await $.session.append(row('第一則'))
  ctl.complete = () => route('haiku')
  await submit($, '再問一次')
  ctl.complete = e => ({ isAnswered: true, usage, text: String(e.prompt) })
  const r: any = await $.session.append(row('第二則'))
  expect(r.message.content[0].text).toBe('亞絲娜 > 第二則')
})

test('好感度：由 0 開始，每日首次互動加分且有每日上限', async ($, on) => {
  const ctl = setup(on)
  ctl.complete = () => route('haiku', 'praise')
  await start($)
  expect(await affinity($)).toBe(0)
  await submit($, '謝謝你，做得很好')
  const n = await affinity($)
  expect(n > 0 && n <= 5).toBe(true)
})

test('關閉好感度時不記錄', { options: { affinity: false } }, async ($, on) => {
  const ctl = setup(on)
  ctl.complete = () => route('haiku', 'praise')
  await start($)
  await submit($, '謝謝')
  expect(await affinity($)).toBe(0)
})

const personaText = async ($: any): Promise<string> => {
  const res: any = await $.prompt.compose({ model: 'm', promptModel: 'm', surfaces: [], tools: [], outputStyle: null, traits: [] } as never)
  return res.sections.find((s: any) => s.id === 'jsc-anima:persona')?.text ?? ''
}

test('好感度被責罵扣分但不跨階段掉落；可手動設定與重置', async ($, on) => {
  const ctl = setup(on)
  ctl.complete = () => route('haiku', 'scold')
  await start($)
  await $.command.run({ command: 'anima-affinity', args: '20' } as never)
  await submit($, '你好笨')
  expect(await affinity($) >= 20).toBe(true)
  await $.command.run({ command: 'anima-affinity', args: 'reset' } as never)
  expect(await affinity($)).toBe(0)
})

test('好感度階段反映在語氣，且不使用情緒勒索', async ($, on) => {
  setup(on)
  await start($)
  const t0 = await personaText($)
  expect(t0.includes('生疏')).toBe(true)
  expect(t0.includes('不使用愧疚感或情緒勒索')).toBe(true)
  await $.command.run({ command: 'anima-affinity', args: '85' } as never)
  expect((await personaText($)).includes('特別的存在')).toBe(true)
})

test('未成年角色維持友情，即使好感度最高且開啟戀愛式依賴', { options: { romance: true } }, async ($, on) => {
  setup(on)
  await start($, 'sao-suguha')
  await $.command.run({ command: 'anima-affinity', args: '100' } as never)
  expect((await personaText($)).includes('關係定位：友情')).toBe(true)
})

const workRoute = (model: string, work: boolean) => ({ isAnswered: true, usage, text: `{"difficulty":8,"model":"${model}","effort":"high","tone":"neutral","ack":"好，我來處理。","work":${work}}` })

test('需要動手的要求交給背景工作線，對話線改用 Haiku', async ($, on) => {
  const ctl = setup(on)
  const spawned: any[] = []
  on('agent.spawn', (_$: any, e: any) => (spawned.push(e), { model: e.model, agentId: 'a1' }) as never)
  ctl.complete = () => workRoute('opus', true)
  await start($)
  await submit($, '幫我重構 src 目錄')
  expect(spawned.length).toBe(1)
  expect(spawned[0].model).toBe('opus')
  await step($, 'main')
  expect(ctl.seen.model).toBe('haiku')
  const sys = await $.prompt.compose({ model: 'm', promptModel: 'm', surfaces: [], tools: [], outputStyle: null, traits: [] } as never)
  expect(sys.sections.find((s: any) => s.id === 'jsc-anima:persona')?.text).toContain('背景工作線')
})

test('work 為 false 或關閉背景工作線時不派工', async ($, on) => {
  const ctl = setup(on)
  let n = 0
  on('agent.spawn', () => (n++, { model: 'm', agentId: 'a' }) as never)
  ctl.complete = () => workRoute('sonnet', false)
  await start($)
  await submit($, '你好')
  expect(n).toBe(0)
})