thzxx
b66945c8a7
CI / verify (push) Successful in 1m2s
修复： - main.ts 退出释放模型显存改用 getSetting(serverUrl)，不再硬编码 127.0.0.1:11434（避免非默认地址时释放请求打到错误端口） - 备份导出/导入并入 localStorage 持久化状态（会话摘要、度量历史、轨迹降级缓存、主题），版本升级到 v2，实现完整备份 - 工具数量改为 getEnabledToolDefinitions().length 动态计算，删除写死"32 个"的硬编码 - 记忆日志区分操作来源：memory:write 透传 reason，标注"新增记忆/替换/删除/清空/TTL 衰减清理/访问统计写回（无新条目）"，避免"写了但看不到新记忆"的困惑 可维护性： - 上下文压力逻辑收敛到统一 calculateContextStats，删除 getContextPressureLevel / getTrendAwareCompressThreshold 的重复实现 - 消除 validateToolArgs 同名碰撞（agent-engine 本地版改名 validateToolArgsQuick） - 子代理工具集改用 getEnabledToolDefinitions() 基线，跟随全局启用开关与 Plan 模式 - 抽取 html-utils.ts 纯函数模块（实体解码/HTML→文本/HTML→Markdown/拦截页检测/相关性评分），tool-handlers-system 净减约 190 行重复代码 - 统一静态导入（savePlanTracker/setPlanModeActive/collectDiagnostics/addWrittenFile） - console.* 使用处补充豁免说明（启动/退出/刷盘阶段无渲染进程可推送日志） - run_command 工具描述改为反映可配置执行模式 测试： - 新增 7 个测试文件 + 扩展 2 个，共 273 个测试（原 34 → 273） - 覆盖 agent-engine / agent-safety / context-manager / tool-registry / result-formatter / tool-parsing / memory-service / crypto / build-context / html-utils / utils / tool-handlers-fs - 全部通过 npm run typecheck && npm test && npm run build
140 lines
4.3 KiB
TypeScript
140 lines
4.3 KiB
TypeScript
import { describe, it, expect } from 'vitest';
|
|
import {
|
|
decodeHTMLEntities,
|
|
htmlToText,
|
|
htmlToMarkdown,
|
|
isBlockedPage,
|
|
computeRelevance,
|
|
} from '../src/main/html-utils.js';
|
|
|
|
describe('decodeHTMLEntities', () => {
|
|
it('解码常见命名实体', () => {
|
|
expect(decodeHTMLEntities('<div>&"x"')).toBe('<div>&"x"');
|
|
expect(decodeHTMLEntities(' ')).toBe(' ');
|
|
});
|
|
|
|
it('解码十进制数字实体', () => {
|
|
expect(decodeHTMLEntities('AB')).toBe('AB');
|
|
});
|
|
|
|
it('解码十六进制数字实体', () => {
|
|
expect(decodeHTMLEntities('AB')).toBe('AB');
|
|
});
|
|
|
|
it('无实体时原样返回', () => {
|
|
expect(decodeHTMLEntities('plain text')).toBe('plain text');
|
|
});
|
|
|
|
it('多实体混合解码', () => {
|
|
expect(decodeHTMLEntities('© 2026 — €10')).toBe('\u00A9 2026 \u2014 \u20AC10');
|
|
});
|
|
});
|
|
|
|
describe('htmlToText', () => {
|
|
it('移除 script/style 噪音标签', () => {
|
|
const html = '<html><body><script>alert(1)</script><p>正文内容</p><style>body{display:none}</style></body></html>';
|
|
const text = htmlToText(html);
|
|
expect(text).toContain('正文内容');
|
|
expect(text).not.toContain('alert');
|
|
expect(text).not.toContain('display:none');
|
|
});
|
|
|
|
it('块级标签转为换行', () => {
|
|
const text = htmlToText('<div>第一段</div><div>第二段</div>');
|
|
expect(text).toContain('第一段');
|
|
expect(text).toContain('第二段');
|
|
});
|
|
|
|
it('去除剩余标签并解码实体', () => {
|
|
const text = htmlToText('<p>hello & goodbye</p>');
|
|
expect(text).toBe('hello & goodbye');
|
|
});
|
|
|
|
it('空输入返回空', () => {
|
|
expect(htmlToText('')).toBe('');
|
|
});
|
|
});
|
|
|
|
describe('htmlToMarkdown', () => {
|
|
it('标题转为 Markdown 标题', () => {
|
|
const md = htmlToMarkdown('<h1>大标题</h1><h2>副标题</h2>');
|
|
expect(md).toContain('# 大标题');
|
|
expect(md).toContain('## 副标题');
|
|
});
|
|
|
|
it('链接转为 Markdown 链接', () => {
|
|
const md = htmlToMarkdown('<a href="https://example.com">example</a>');
|
|
expect(md).toContain('[example](https://example.com)');
|
|
});
|
|
|
|
it('代码块转为围栏代码', () => {
|
|
const md = htmlToMarkdown('<pre><code>const x = 1;</code></pre>');
|
|
expect(md).toContain('```');
|
|
expect(md).toContain('const x = 1;');
|
|
});
|
|
|
|
it('列表项转为 - 列表', () => {
|
|
const md = htmlToMarkdown('<ul><li>项目A</li><li>项目B</li></ul>');
|
|
expect(md).toContain('- 项目A');
|
|
expect(md).toContain('- 项目B');
|
|
});
|
|
|
|
it('加粗/斜体标签转换', () => {
|
|
const md = htmlToMarkdown('<strong>加粗</strong><em>斜体</em>');
|
|
expect(md).toContain('**加粗**');
|
|
expect(md).toContain('*斜体*');
|
|
});
|
|
});
|
|
|
|
describe('isBlockedPage', () => {
|
|
it('短内容视为拦截页', () => {
|
|
expect(isBlockedPage('<html></html>')).toBe(true);
|
|
});
|
|
|
|
it('Cloudflare 拦截特征', () => {
|
|
const html = '<html><head><title>Just a moment...</title></head></html>'.repeat(5);
|
|
expect(isBlockedPage(html)).toBe(true);
|
|
});
|
|
|
|
it('403 拦截特征', () => {
|
|
const html = '<title>403 Forbidden</title>'.repeat(10);
|
|
expect(isBlockedPage(html)).toBe(true);
|
|
});
|
|
|
|
it('验证码特征', () => {
|
|
const html = '请启用JavaScript'.repeat(10);
|
|
expect(isBlockedPage(html)).toBe(true);
|
|
});
|
|
|
|
it('正常长页面不视为拦截', () => {
|
|
const html = '<html><body>' + '<p>正常内容</p>'.repeat(50) + '</body></html>';
|
|
expect(isBlockedPage(html)).toBe(false);
|
|
});
|
|
});
|
|
|
|
describe('computeRelevance', () => {
|
|
it('无 query 返回中性 50 分', () => {
|
|
expect(computeRelevance('', '标题', '摘要')).toBe(50);
|
|
});
|
|
|
|
it('CJK 关键词命中标题得高分', () => {
|
|
const score = computeRelevance('rust 语言', 'Rust 语言教程', '本教程介绍 rust');
|
|
expect(score).toBeGreaterThanOrEqual(25);
|
|
});
|
|
|
|
it('英文词命中标题得 15 分', () => {
|
|
const score = computeRelevance('rust backend', 'rust backend guide', 'a guide');
|
|
expect(score).toBeGreaterThanOrEqual(15);
|
|
});
|
|
|
|
it('完全无关标题得 0 分', () => {
|
|
const score = computeRelevance('rust', 'cooking recipes', 'food');
|
|
expect(score).toBe(0);
|
|
});
|
|
|
|
it('得分上限 100', () => {
|
|
const score = computeRelevance('rust language guide', 'rust language guide', 'rust language guide');
|
|
expect(score).toBeLessThanOrEqual(100);
|
|
});
|
|
});
|