node-resource-utils 0.0.0-stage → 1.1.7

package/resources/src/lib/aiForm.test.ts

@@ -0,0 +1,191 @@
+import { describe, expect, it } from 'vitest';
+import {
+  buildAskForm,
+  buildWriteForm,
+  formValuesText,
+  mergeFormValues,
+  readFormToken,
+} from './aiForm';
+import { validateWrite } from './aiTools';
+
+/**
+ * AI 输入表单的底座测试。
+ *
+ * 这几条断言守的是"通用性"，不是某个业务：
+ *   - 字段必须从 schema 反推（换表不用改代码）
+ *   - 参数齐了不能弹表单（否则每件事都多一层）
+ *   - 令牌里的字段白名单必须生效（否则表单成了改任意字段的后门）
+ */
+
+const USER = 'u_test_001';
+
+describe('buildWriteForm', () => {
+  it('缺必填参数时，按 schema 生成字段（枚举/日期/外键都有正确控件）', async () => {
+    const form = await buildWriteForm('create_employee', {}, USER);
+    expect(form).not.toBeNull();
+    if (!form) return;
+
+    const byName = new Map(form.fields.map((f) => [f.name, f]));
+
+    // 必填项一定在
+    expect(byName.get('code')?.required).toBe(true);
+    expect(byName.get('name')?.required).toBe(true);
+
+    // 枚举 → 下拉，选项来自 schema 注释（gender: male | female）
+    const gender = byName.get('gender');
+    expect(gender?.kind).toBe('enum');
+    expect(gender?.options?.map((o) => o.value)).toEqual(['male', 'female']);
+
+    // 日期字段 → 日期控件，不是文本框
+    expect(byName.get('hireDate')?.kind).toBe('date');
+
+    // 外键字段 → 下拉（选项可能因库里没数据为空，但类型必须是 fk）
+    expect(byName.get('deptId')?.kind).toBe('fk');
+
+    // 一张表单不该无限长
+    expect(form.fields.length).toBeLessThanOrEqual(8);
+  });
+
+  it('参数齐全时返回 null —— 不缺就直出确认卡，不打扰用户', async () => {
+    const form = await buildWriteForm(
+      'create_employee',
+      { code: 'E9001', name: '测试员工', gender: 'male' },
+      USER
+    );
+    expect(form).toBeNull();
+  });
+
+  it('模型填的枚举标签能对上值（填「男」不该丢成空）', async () => {
+    const form = await buildWriteForm('create_employee', { gender: 'male' }, USER);
+    const gender = form?.fields.find((f) => f.name === 'gender');
+    expect(gender?.value).toBe('male');
+  });
+
+  it('令牌能被验签，且只属于签发它的用户', async () => {
+    const form = await buildWriteForm('create_employee', {}, USER);
+    expect(form).not.toBeNull();
+    if (!form) return;
+
+    const ok = readFormToken(form.token, USER);
+    expect(ok.ok).toBe(true);
+    if (ok.ok) {
+      expect(ok.payload.n).toBe('create_employee');
+      expect(ok.payload.m).toBe('fill');
+      expect(ok.payload.f.map((x) => x.n)).toContain('code');
+    }
+
+    const other = readFormToken(form.token, 'someone_else');
+    expect(other.ok).toBe(false);
+  });
+});
+
+describe('mergeFormValues', () => {
+  it('只接受令牌里的字段，前端多塞的键被丢掉', async () => {
+    const form = await buildWriteForm('create_employee', {}, USER);
+    expect(form).not.toBeNull();
+    if (!form) return;
+
+    const read = readFormToken(form.token, USER);
+    expect(read.ok).toBe(true);
+    if (!read.ok) return;
+
+    const merged = mergeFormValues(read.payload, {
+      code: 'E9002',
+      name: '张三',
+      // 下面这两个都不在字段白名单里，必须被丢弃
+      id: 'hacked',
+      status: 0,
+    });
+
+    expect(merged.code).toBe('E9002');
+    expect(merged.name).toBe('张三');
+    expect(merged).not.toHaveProperty('id');
+    expect(merged.status).toBeUndefined();
+  });
+});
+
+describe('buildAskForm', () => {
+  it('绑定写工具时，字段定义来自 schema（不是模型想象的控件）', async () => {
+    const form = await buildAskForm(
+      { title: '补全员工信息', tool: 'create_employee', fields: ['code', 'hireDate'] },
+      USER
+    );
+    expect(form).not.toBeNull();
+    if (!form) return;
+
+    expect(form.mode).toBe('fill');
+    expect(form.tool).toBe('create_employee');
+    expect(form.fields.map((f) => f.name)).toEqual(['code', 'hireDate']);
+    expect(form.fields.find((f) => f.name === 'hireDate')?.kind).toBe('date');
+  });
+
+  it('自由追问时退化成文本框，不猜类型', async () => {
+    const form = await buildAskForm(
+      { title: '补充查询条件', question: '想看哪个时间段？', fields: ['startDate', 'endDate'] },
+      USER
+    );
+    expect(form).not.toBeNull();
+    if (!form) return;
+
+    expect(form.mode).toBe('ask');
+    expect(form.fields.map((f) => f.kind)).toEqual(['text', 'text']);
+  });
+
+  it('fields 为空时不生成表单（避免弹一张空卡片）', async () => {
+    const form = await buildAskForm({ title: '问你个事', fields: [] }, USER);
+    expect(form).toBeNull();
+  });
+});
+
+describe('validateWrite 复用输入校验底座', () => {
+  it('AI 写入员工时，手机号格式错误会被拦住', () => {
+    const result = validateWrite('create_employee', {
+      code: 'E1',
+      name: '测试员工',
+      gender: 'male',
+      phone: '123',
+    });
+    expect(result.ok).toBe(false);
+    expect(result.message).toContain('手机号应为 11 位');
+  });
+
+  it('AI 写入员工时，身份证校验位错误会被拦住', () => {
+    const result = validateWrite('create_employee', {
+      code: 'E1',
+      name: '测试员工',
+      gender: 'male',
+      idCard: '110105194912310021',
+    });
+    expect(result.ok).toBe(false);
+    expect(result.message).toContain('校验位不正确');
+  });
+
+  it('合法参数可以通过预检', () => {
+    const result = validateWrite('create_employee', {
+      code: 'E1',
+      name: '测试员工',
+      gender: 'male',
+      idCard: '11010519491231002X',
+      phone: '13800138000',
+      email: 'test@example.com',
+    });
+    expect(result.ok).toBe(true);
+  });
+});
+
+describe('formValuesText', () => {
+  it('回灌给模型时用中文标签，模型不用猜键名', async () => {
+    const form = await buildAskForm(
+      { title: '补充查询条件', fields: ['startDate', 'endDate'] },
+      USER
+    );
+    if (!form) throw new Error('form should exist');
+    const read = readFormToken(form.token, USER);
+    if (!read.ok) throw new Error('token should verify');
+
+    const text = formValuesText(read.payload, { startDate: '2026-10-01', endDate: '2026-10-31' });
+    expect(text).toContain('2026-10-01');
+    expect(text).toContain('2026-10-31');
+    expect(text).toContain('【用户填写了表单：补充查询条件】');
+  });
+});