taskAnchorReminder.test.ts

import { describe, expect, test } from 'bun:test'
import type { Task } from '../../../utils/tasks.js'
import {
  buildTaskAnchorReminder,
  buildUnfinishedTaskNotice,
  isAnchorlessNudge,
} from '../taskAnchorReminder.js'

function task(partial: Partial<Task> & Pick<Task, 'id' | 'status'>): Task {
  return {
    subject: `Task ${partial.id}`,
    description: '',
    blocks: [],
    blockedBy: [],
    ...partial,
  }
}

describe('isAnchorlessNudge', () => {
  test('matches short Chinese keep-going nudges', () => {
    expect(isAnchorlessNudge('继续执行后续的工作')).toBe(true)
    expect(isAnchorlessNudge('继续完成当前任务')).toBe(true)
    expect(isAnchorlessNudge('自动执行后续的所有流程')).toBe(true)
    expect(isAnchorlessNudge('接着做')).toBe(true)
  })

  test('matches short English keep-going nudges', () => {
    expect(isAnchorlessNudge('continue')).toBe(true)
    expect(isAnchorlessNudge('keep going')).toBe(true)
    expect(isAnchorlessNudge('go on please')).toBe(true)
  })

  test('does not match empty or whitespace', () => {
    expect(isAnchorlessNudge('')).toBe(false)
    expect(isAnchorlessNudge('   ')).toBe(false)
  })

  test('does not match long messages that merely contain a nudge word', () => {
    expect(
      isAnchorlessNudge(
        '继续用状态机方案重构 REPL 的输入处理模块，并补齐对应的单元测试',
      ),
    ).toBe(false)
  })

  test('does not match substantive requests without nudge words', () => {
    expect(isAnchorlessNudge('给我看下 query.ts 的结构')).toBe(false)
    expect(isAnchorlessNudge('fix the login bug')).toBe(false)
  })
})

describe('buildTaskAnchorReminder', () => {
  test('returns null when there are no unfinished tasks', () => {
    expect(buildTaskAnchorReminder([])).toBeNull()
    expect(
      buildTaskAnchorReminder([
        task({ id: '1', status: 'completed' }),
        task({ id: '2', status: 'completed' }),
      ]),
    ).toBeNull()
  })

  test('lists only unfinished tasks, in_progress before pending', () => {
    const reminder = buildTaskAnchorReminder([
      task({ id: '1', status: 'completed', subject: 'done' }),
      task({ id: '2', status: 'pending', subject: 'second' }),
      task({ id: '3', status: 'in_progress', subject: 'third' }),
    ])
    expect(reminder).not.toBeNull()
    const body = reminder!
    expect(body).toContain('2 unfinished task(s)')
    // in_progress (#3) must appear before pending (#2)
    expect(body.indexOf('#3 [in_progress]')).toBeLessThan(
      body.indexOf('#2 [pending]'),
    )
    expect(body).not.toContain('#1')
    expect(body).toContain('<system-reminder>')
    expect(body).toContain('</system-reminder>')
  })

  test('orders same-status tasks by numeric id', () => {
    const reminder = buildTaskAnchorReminder([
      task({ id: '10', status: 'pending', subject: 'ten' }),
      task({ id: '2', status: 'pending', subject: 'two' }),
    ])!
    expect(reminder.indexOf('#2 ')).toBeLessThan(reminder.indexOf('#10 '))
  })

  test('caps the list and reports overflow', () => {
    const many: Task[] = Array.from({ length: 25 }, (_, i) =>
      task({ id: String(i + 1), status: 'pending' }),
    )
    const reminder = buildTaskAnchorReminder(many)!
    expect(reminder).toContain('25 unfinished task(s)')
    expect(reminder).toContain('…and 5 more')
  })
})

describe('buildUnfinishedTaskNotice', () => {
  test('returns null when no task is in_progress', () => {
    expect(buildUnfinishedTaskNotice([])).toBeNull()
    expect(
      buildUnfinishedTaskNotice([
        task({ id: '1', status: 'pending' }),
        task({ id: '2', status: 'completed' }),
      ]),
    ).toBeNull()
  })

  test('surfaces in_progress task names and pending count', () => {
    const notice = buildUnfinishedTaskNotice([
      task({ id: '1', status: 'in_progress', subject: '写模块' }),
      task({ id: '2', status: 'pending' }),
      task({ id: '3', status: 'pending' }),
      task({ id: '4', status: 'completed' }),
    ])!
    expect(notice).toContain('#1 写模块')
    expect(notice).toContain('2 个待办')
  })

  test('collapses overflow when more than three in_progress tasks', () => {
    const notice = buildUnfinishedTaskNotice(
      Array.from({ length: 5 }, (_, i) =>
        task({ id: String(i + 1), status: 'in_progress' }),
      ),
    )!
    expect(notice).toContain('等 5 个')
  })
})