62 lines
2.1 KiB
JavaScript
62 lines
2.1 KiB
JavaScript
import assert from 'node:assert/strict'
|
||
import test from 'node:test'
|
||
|
||
import {
|
||
extractTrustedHtmlBlocks,
|
||
normalizeConversationText,
|
||
restoreTrustedHtmlBlocks
|
||
} from '../src/utils/conversationTrustedHtml.js'
|
||
|
||
test('conversation trusted html helper preserves valid document cards', () => {
|
||
const trustedBlock = [
|
||
'<!-- ai-trusted-html:start -->',
|
||
'<section class="ai-document-card-list" aria-label="单据结果">',
|
||
'<article class="ai-document-card" aria-label="单据详情">',
|
||
'<strong>差旅申请</strong>',
|
||
'<a class="ai-html-action-link" data-ai-action="open-document-detail" href="#ai-open-document-detail:CL-1">查看</a>',
|
||
'</article>',
|
||
'</section>',
|
||
'<!-- ai-trusted-html:end -->'
|
||
].join('')
|
||
|
||
const extracted = extractTrustedHtmlBlocks(`结果如下：\n\n${trustedBlock}`)
|
||
assert.equal(extracted.trustedHtmlBlocks.length, 1)
|
||
assert.match(extracted.content, /AI_TRUSTED_HTML_BLOCK_0/)
|
||
|
||
const restored = restoreTrustedHtmlBlocks(
|
||
'<p>AI_TRUSTED_HTML_BLOCK_0</p>\n',
|
||
extracted.trustedHtmlBlocks
|
||
)
|
||
assert.match(restored, /class="ai-document-card-list"/)
|
||
assert.match(restored, /href="#ai-open-document-detail:CL-1"/)
|
||
assert.doesNotMatch(restored, /AI_TRUSTED_HTML_BLOCK_0/)
|
||
})
|
||
|
||
test('conversation trusted html helper rejects unsafe trusted blocks', () => {
|
||
const extracted = extractTrustedHtmlBlocks([
|
||
'<!-- ai-trusted-html:start -->',
|
||
'<section class="ai-document-card-list">',
|
||
'<a href="javascript:alert(1)" onclick="alert(1)">危险</a>',
|
||
'</section>',
|
||
'<!-- ai-trusted-html:end -->'
|
||
].join(''))
|
||
|
||
assert.equal(extracted.trustedHtmlBlocks.length, 0)
|
||
assert.equal(extracted.content.trim(), '')
|
||
})
|
||
|
||
test('conversation trusted html helper normalizes business copy outside fences', () => {
|
||
const normalized = normalizeConversationText([
|
||
'基础信息识别结果：请核对',
|
||
'时间：2026-02-20',
|
||
'',
|
||
'```',
|
||
'金额：不要改代码块',
|
||
'```'
|
||
].join('\n'), { trim: true })
|
||
|
||
assert.match(normalized, /### 基础信息识别结果/)
|
||
assert.match(normalized, /- \*\*时间\*\*：2026-02-20/)
|
||
assert.match(normalized, /```\n金额：不要改代码块\n```/)
|
||
})
|