OmniCard/admin/tests/ai-knowledge.test.ts

import test from 'node:test';
import assert from 'node:assert/strict';
import fs from 'node:fs';
import path from 'node:path';
import { fileURLToPath } from 'node:url';
import knowledgeI18n from '../src/pages/ai/knowledge/locale/index.ts';

const adminRoot = path.resolve(path.dirname(fileURLToPath(import.meta.url)), '..');
const knowledgePagePath = path.join(adminRoot, 'src/pages/ai/knowledge/index.tsx');
const knowledgeLocalePath = path.join(adminRoot, 'src/pages/ai/knowledge/locale/index.ts');

test('1. ai/knowledge 页面必须请求 knowledge/page 与 add，并使用 Arco Table/Form/Modal', () => {
  assert.ok(fs.existsSync(knowledgePagePath), '知识库页面必须存在');
  const content = fs.readFileSync(knowledgePagePath, 'utf-8');

  assert.ok(
    content.includes('admin/ai/knowledge/page'),
    '必须请求 POST /api/admin/ai/knowledge/page'
  );
  assert.ok(
    content.includes('admin/ai/knowledge/add'),
    '必须请求 POST /api/admin/ai/knowledge/add'
  );
  assert.ok(content.includes('Table'), '必须使用 Arco Table');
  assert.ok(content.includes('Form'), '必须使用 Arco Form');
  assert.ok(content.includes('Modal'), '必须使用 Arco Modal');
  assert.ok(
    content.includes("from '@arco-design/web-react'"),
    '必须从 @arco-design/web-react 引入组件'
  );
  assert.ok(content.includes('AiKnowledgeDoc'), '必须使用 AiKnowledgeDoc 类型');
  assert.ok(
    content.includes('code === 1000') || content.includes('code ===1000'),
    '必须按 Cool-Admin code === 1000 判断成功'
  );
  assert.ok(content.includes('Message.error'), '失败必须使用 Arco Message.error');
  assert.ok(!content.includes('Mock.js') && !content.includes("from 'mockjs'"), '禁止用 Mock.js 假数据冒充成功');
  assert.ok(!/\bany\b/.test(content), '禁止使用 any');
});

test('2. ai/knowledge 必须支持筛选、摘要列、编辑删除与空状态主 CTA', () => {
  const content = fs.readFileSync(knowledgePagePath, 'utf-8');

  assert.ok(content.includes("from '@/types/omni'"), '必须复用 admin/src/types/omni.ts');
  assert.ok(content.includes('lightColors'), '颜色必须引用 lightColors token');
  assert.ok(!/#([0-9A-Fa-f]{3}|[0-9A-Fa-f]{6})\b/.test(content), '禁止硬编码颜色');

  assert.ok(content.includes('title'), '必须展示 title');
  assert.ok(content.includes('cardId'), '必须展示并筛选 cardId');
  assert.ok(content.includes('locale'), '必须展示并筛选 locale');
  assert.ok(content.includes('keyWord'), '筛选必须支持 keyWord');
  assert.ok(content.includes('content'), '必须展示 content 摘要');

  assert.ok(
    content.includes('admin/ai/knowledge/update'),
    '编辑必须请求 POST /api/admin/ai/knowledge/update'
  );
  assert.ok(
    content.includes('admin/ai/knowledge/delete'),
    '删除必须请求 POST /api/admin/ai/knowledge/delete'
  );
  assert.ok(content.includes('Popconfirm'), '删除必须使用 Arco Popconfirm');
  assert.ok(content.includes('Upload'), '可选文本导入必须使用 Arco Upload');
  assert.ok(
    content.includes('readAsText') || content.includes('FileReader'),
    'Upload 只允许把文本读进 content'
  );
  assert.ok(
    !content.includes('oss') && !content.includes('OSS'),
    '禁止手写 OSS 上传'
  );

  assert.ok(content.includes('type="primary"'), '必须有一个主 CTA');
  assert.ok(
    content.includes('aiKnowledge.action.add') || content.includes('aiKnowledge.empty'),
    '空状态必须复用新增 CTA 文案'
  );
  assert.ok(
    content.includes('noDataElement') || content.includes('aiKnowledge.empty'),
    '必须提供空状态'
  );
  assert.ok(!content.includes('<Pagination'), '禁止手写分页器，必须使用 Table 自带分页');
  assert.ok(!content.includes('Mock.js') && !content.includes("from 'mockjs'"), '禁止用 Mock.js');
  assert.ok(!/\bany\b/.test(content), '禁止使用 any');
});

test('3. ai/knowledge locale 必须包含 menu.ai.knowledge 与中英文案', () => {
  assert.ok(fs.existsSync(knowledgeLocalePath), '知识库语言包必须存在');
  const content = fs.readFileSync(knowledgeLocalePath, 'utf-8');

  assert.ok(content.includes("'menu.ai.knowledge'"), '页面语言包必须包含 menu.ai.knowledge');
  assert.ok(content.includes("'en-US'"), '必须包含英文案');
  assert.ok(content.includes("'zh-CN'"), '必须包含中文案');
  assert.ok(content.includes('aiKnowledge.request.fail'), '必须包含失败提示文案');
  assert.ok(content.includes('aiKnowledge.action.add'), '必须包含新增主 CTA 文案');
  assert.ok(content.includes('aiKnowledge.empty'), '必须包含空状态文案');
  assert.ok(content.includes('aiKnowledge.filter.cardId'), '必须包含 cardId 筛选文案');
  assert.ok(content.includes('aiKnowledge.filter.locale'), '必须包含 locale 筛选文案');
  assert.ok(content.includes('aiKnowledge.filter.keyWord'), '必须包含 keyWord 筛选文案');
  assert.ok(!/\bany\b/.test(content), 'locale 禁止使用 any');

  assert.equal(knowledgeI18n['en-US']['menu.ai.knowledge'], 'Knowledge');
  assert.equal(knowledgeI18n['zh-CN']['menu.ai.knowledge'], '知识库');
  assert.ok(
    /keyword overlap/i.test(knowledgeI18n['en-US']['aiKnowledge.sandbox.desc']),
    '英文沙盒说明须含 keyword overlap'
  );
  assert.ok(
    knowledgeI18n['zh-CN']['aiKnowledge.sandbox.desc'].includes('关键词重叠召回'),
    '中文沙盒说明须含关键词重叠召回'
  );
  assert.ok(
    /تداخل الكلمات المفتاحية/.test(
      knowledgeI18n['ar-SA']['aiKnowledge.sandbox.desc']
    ),
    '阿语沙盒说明须含 تداخل الكلمات المفتاحية'
  );
});

test('4. 沙盒文案必须写关键词重叠召回，禁止伪装向量库/命中率', () => {
  const page = fs.readFileSync(knowledgePagePath, 'utf-8');
  const i18n = fs.readFileSync(knowledgeLocalePath, 'utf-8');

  assert.ok(i18n.includes("'ar-SA'"), '必须包含阿语文案');
  assert.ok(
    i18n.includes('aiKnowledge.sandbox.desc'),
    '必须包含沙盒说明文案'
  );
  assert.ok(
    i18n.includes('aiKnowledge.sandbox.recallCount'),
    '必须包含召回条数文案'
  );

  // 中英文必须明确「关键词重叠」召回，禁止向量库表述
  assert.match(
    i18n,
    /关键词重叠召回/,
    '中文沙盒说明必须写关键词重叠召回'
  );
  assert.match(
    i18n,
    /keyword overlap/i,
    '英文沙盒说明必须写 keyword overlap'
  );
  assert.match(
    i18n,
    /تداخل الكلمات|تداخل المفردات|تداخل الكلمات المفتاحية/,
    '阿语沙盒说明须用自然的关键词重叠产品文案'
  );

  assert.ok(!/vector|embedding|向量库|向量召回|向量命中|向量相似度/i.test(i18n), 'locale 禁止向量库/向量召回表述');
  assert.ok(!/simulated Top-K|模拟召回|向量命中/i.test(i18n), 'locale 禁止伪装模拟向量 Top-K');
  assert.ok(!/vector|embedding|向量库|向量召回|向量命中/i.test(page), '页面禁止向量库/向量召回表述');

  // 禁止伪造命中率百分比 Progress；改为展示召回条数
  assert.ok(!page.includes('<Progress'), '禁止用 Progress 展示伪造命中率百分比');
  assert.ok(!page.includes('calculatePercent'), '禁止把 score 换算成假命中率百分比');
  assert.ok(
    page.includes('aiKnowledge.sandbox.recallCount') &&
      page.includes('recallResults.length'),
    '有召回结果时必须展示召回条数'
  );
  assert.ok(
    page.includes('sandbox-test') && page.includes('chunkSize') && page.includes('overlap'),
    '不得改动沙盒请求字段（chunkSize/overlap 等）'
  );
});

test('5. 知识库须提供 URL 录入；无真实抓取接口时按钮禁用且禁止假切片', () => {
  const page = fs.readFileSync(knowledgePagePath, 'utf-8');
  const i18n = fs.readFileSync(knowledgeLocalePath, 'utf-8');

  assert.ok(
    i18n.includes("'aiKnowledge.form.url'") || i18n.includes('aiKnowledge.form.url'),
    'locale 必须包含 URL 录入文案'
  );
  assert.ok(
    i18n.includes("'aiKnowledge.crawl.unavailable'") ||
      i18n.includes('aiKnowledge.crawl.unavailable'),
    'locale 必须说明暂不可抓取'
  );
  assert.ok(
    knowledgeI18n['zh-CN']['aiKnowledge.crawl.unavailable'].includes('暂不可'),
    '中文须说明暂不可抓取'
  );
  assert.ok(
    /unavailable|not available|not supported/i.test(
      knowledgeI18n['en-US']['aiKnowledge.crawl.unavailable']
    ),
    '英文须说明抓取暂不可用'
  );
  assert.ok(
    page.includes("t['aiKnowledge.form.url']") || page.includes('aiKnowledge.form.url'),
    '页面必须出现 URL 录入'
  );
  assert.ok(
    page.includes("t['aiKnowledge.crawl.unavailable']") ||
      page.includes('aiKnowledge.crawl.unavailable'),
    '页面必须展示暂不可抓取说明'
  );

  const hasCrawlApi =
    /admin\/ai\/knowledge\/(crawl|fetch|scrape|ingest-url|fetch-url|import-url)/i.test(
      page
    ) || /CRAWL_API|FETCH_URL_API|SCRAPE_API|INGEST_URL_API/i.test(page);

  if (hasCrawlApi) {
    assert.ok(
      page.includes('code === 1000') || page.includes('code ===1000'),
      '有抓取接口时必须按 code === 1000 判断成功'
    );
    assert.ok(
      !/Message\.success[\s\S]{0,80}crawl|Message\.success[\s\S]{0,80}fetch/i.test(
        page
      ) || page.includes('code === 1000'),
      '失败时禁止 Message.success'
    );
  } else {
    assert.ok(
      /crawl[\s\S]{0,200}disabled|disabled[\s\S]{0,200}crawl|aiKnowledge\.crawl[\s\S]{0,120}disabled/i.test(
        page
      ) ||
        (page.includes('disabled') &&
          (page.includes('aiKnowledge.crawl') || page.includes('crawl'))),
      '无真实抓取接口时抓取按钮必须禁用'
    );
    assert.ok(
      !/fakeChunk|mockChunk|假切片|假命中|simulatedChunk|dummyChunk/i.test(page),
      '禁止生成假切片或假命中'
    );
    assert.ok(
      !page.includes('Math.random') && !page.includes('mockjs'),
      '禁止用随机数或 Mock 伪造抓取切片'
    );
    assert.ok(
      !/handleCrawl|fetchWebsite|scrapeUrl|ingestUrl/i.test(page),
      '无真实接口时禁止手写抓取处理函数写假正文'
    );
    assert.ok(
      !/setFieldValue\(\s*['"]content['"][\s\S]{0,80}url/i.test(page) &&
        !/setSandboxContent\([\s\S]{0,80}url/i.test(page),
      '禁止把 URL 录入伪装成抓取正文或沙盒切片'
    );
  }
});