MailUI4Agents/plugins/pi-mail-bridge/test/user-question.test.mjs

/**
 * `ask_user_question` ↔ AgentMail 询问邮件 的映射约定。
 *
 * 三桥共用同一份（deploy/check-shared-libs.sh 校验逐字节相同）。
 *
 * 这里钉住的是**会让模型永久挂死或拿到错答案**的边界：
 *   - 摊平多问题时选项不能丢、不能重复
 *   - 回写 answers 时每个问题必须拿到属于自己的选择（不能张冠李戴）
 *   - 无选项题（自由文本）必须能通过 custom 把话带回去
 *   - 空回答不能被当成「答了」
 */

import { test } from 'node:test';
import assert from 'node:assert/strict';

import {
  flattenQuestions,
  answersFromDecision,
  isBlankAnswer,
  hasOptions,
  optionLabels,
  questionTitle,
} from '../lib/user-question.js';

test('单问题单选项：忠实映射问题与选项', () => {
  const flat = flattenQuestions([{
    id: 'q1',
    question: '用哪种方案？',
    header: '选择',
    options: [{ label: '方案 A' }, { label: '方案 B' }],
  }]);
  assert.equal(flat.question, '选择: 用哪种方案？');
  assert.deepEqual(flat.options, ['方案 A', '方案 B']);
  assert.equal(flat.multiSelect, false);
});

test('单问题多选：multiSelect 透传', () => {
  const flat = flattenQuestions([{
    id: 'q1',
    question: '要哪些？',
    options: [{ label: 'a' }, { label: 'b' }],
    multiSelect: true,
  }]);
  assert.equal(flat.multiSelect, true);
});

test('多问题：选项取并集且去重保序，multiSelect 置真', () => {
  const flat = flattenQuestions([
    { id: 'q1', question: '前端？', options: [{ label: 'React' }, { label: 'Vue' }] },
    { id: 'q2', question: '后端？', options: [{ label: 'Vue' }, { label: 'Go' }] },
  ]);
  assert.deepEqual(flat.options, ['React', 'Vue', 'Go'], '重复 label 只出现一次');
  assert.equal(flat.multiSelect, true, '多问题必须允许多选，否则无法同时回答两题');
  assert.match(flat.context, /前端？/);
  assert.match(flat.context, /后端？/);
});

test('无选项题：options 为空，正文提示直接填写', () => {
  const flat = flattenQuestions([{ id: 'q1', question: '你的名字？' }]);
  assert.deepEqual(flat.options, []);
  assert.match(flat.context, /直接填写|没有预设选项/);
});

test('空问题列表：抛错而不是造一封没有内容的信', () => {
  assert.throws(() => flattenQuestions([]), /至少需要一个/);
  assert.throws(() => flattenQuestions(undefined), /至少需要一个/);
});

test('单问题回写：选项进 selected，备注进 custom', () => {
  const qs = [{ id: 'q1', question: '选哪个', options: [{ label: 'A' }, { label: 'B' }] }];
  const ans = answersFromDecision(qs, 'A', '再确认下');
  assert.deepEqual(ans.answers, [{ id: 'q1', selected: ['A'], custom: '再确认下' }]);
});

test('单问题多选回写：多行决策拆成多个 selected', () => {
  const qs = [{ id: 'q1', question: '选哪些', options: [{ label: 'A' }, { label: 'B' }], multiSelect: true }];
  const ans = answersFromDecision(qs, 'A\nB', '');
  assert.deepEqual(ans.answers[0].selected, ['A', 'B']);
  assert.equal(ans.answers[0].custom, undefined, '空备注不该变成空 custom');
});

test('无选项题回写：答案进 custom，selected 为空', () => {
  const qs = [{ id: 'q1', question: '名字？' }];
  const ans = answersFromDecision(qs, '', '张三');
  assert.deepEqual(ans.answers, [{ id: 'q1', selected: [], custom: '张三' }]);
});

test('无选项题只有决策文本时：文本进 custom（否则答案永远为空）', () => {
  const qs = [{ id: 'q1', question: '名字？' }];
  const ans = answersFromDecision(qs, '李四', undefined);
  assert.deepEqual(ans.answers, [{ id: 'q1', selected: [], custom: '李四' }]);
});

test('多问题回写：选择按 label 归属分配到各自的问题（不张冠李戴）', () => {
  const qs = [
    { id: 'q1', question: '前端？', options: [{ label: 'React' }, { label: 'Vue' }] },
    { id: 'q2', question: '后端？', options: [{ label: 'Go' }, { label: 'Rust' }] },
  ];
  const ans = answersFromDecision(qs, 'Vue\nGo', '都行');
  assert.deepEqual(ans.answers[0].selected, ['Vue'], 'q1 只拿前端的选择');
  assert.deepEqual(ans.answers[1].selected, ['Go'], 'q2 只拿后端的选择');
  assert.equal(ans.answers[0].custom, '都行', '备注归第一个问题');
  assert.equal(ans.answers[1].custom, undefined, '备注不重复分发');
});

test('多问题里认不出的 label：不匹配任何问题，不猜测放行', () => {
  const qs = [
    { id: 'q1', question: 'a', options: [{ label: 'X' }] },
    { id: 'q2', question: 'b', options: [{ label: 'Y' }] },
  ];
  const ans = answersFromDecision(qs, 'Z', '');
  assert.deepEqual(ans.answers[0].selected, [], '认不出的 label 不得被塞进任意问题');
  assert.deepEqual(ans.answers[1].selected, []);
});

test('answers 的 id 与问题一一对应', () => {
  const qs = [{ id: 'alpha', question: 'a', options: [{ label: 'X' }] }, { id: 'beta', question: 'b' }];
  const ans = answersFromDecision(qs, 'X', 'note');
  assert.deepEqual(ans.answers.map((a) => a.id), ['alpha', 'beta']);
});

test('空回答判定：有选项的问题什么都没选 = 空', () => {
  const qs = [{ id: 'q1', question: 'a', options: [{ label: 'X' }] }];
  assert.equal(isBlankAnswer(qs, '', ''), true);
  assert.equal(isBlankAnswer(qs, 'X', ''), false);
  assert.equal(isBlankAnswer(qs, '', '自由文本'), false, '自由文本也算答了');
});

test('hasOptions / optionLabels / questionTitle 的边界', () => {
  assert.equal(hasOptions({}), false);
  assert.equal(hasOptions({ options: [] }), false);
  assert.equal(hasOptions({ options: [{ label: 'a' }] }), true);
  assert.deepEqual(optionLabels({ options: ['a', { label: 'b' }, { description: 'x' }] }), ['a', 'b']);
  assert.equal(questionTitle({ question: '只问一句' }), '只问一句');
  assert.equal(questionTitle({}), '(未提供问题)');
});