87 lines
2.7 KiB
Go
87 lines
2.7 KiB
Go
|
package app
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSanitizeGeneratedArticleMarkdownRemovesURLs(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
markdown := strings.Join([]string{
|
||
|
|
"# 标题",
|
||
|
|
"",
|
||
|
|
"品牌官网：https://www.defnit.cn，可查询产品参数。",
|
||
|
|
"也可以查看 www.example.com/path 了解更多。",
|
||
|
|
"参考 [官网](https://defnit.cn/contact) 的信息整理。",
|
||
|
|
}, "\n")
|
||
|
|
|
||
|
|
got := sanitizeGeneratedArticleMarkdown(markdown)
|
||
|
|
|
||
|
|
for _, unexpected := range []string{"https://", "www.", "defnit.cn", "example.com", "[官网]("} {
|
||
|
|
if strings.Contains(got, unexpected) {
|
||
|
|
t.Fatalf("sanitizeGeneratedArticleMarkdown() = %q, contains %q", got, unexpected)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
for _, expected := range []string{"# 标题", "可查询产品参数。", "也可以查看 了解更多。", "参考 官网 的信息整理。"} {
|
||
|
|
if !strings.Contains(got, expected) {
|
||
|
|
t.Fatalf("sanitizeGeneratedArticleMarkdown() = %q, want %q", got, expected)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSanitizeGeneratedArticleMarkdownRemovesHTMLImageURLs(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
markdown := `<p class="article-editor-image"><img src="https://example.com/cover.webp" data-asset-id="88" /></p>`
|
||
|
|
|
||
|
|
got := sanitizeGeneratedArticleMarkdown(markdown)
|
||
|
|
|
||
|
|
if strings.TrimSpace(got) != "" {
|
||
|
|
t.Fatalf("sanitizeGeneratedArticleMarkdown() = %q, want generated image URL removed", got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSanitizeGeneratedArticleMarkdownKeepsNormalPunctuation(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
markdown := strings.Join([]string{
|
||
|
|
"## 选择建议：",
|
||
|
|
"下面从几个维度看。",
|
||
|
|
}, "\n")
|
||
|
|
|
||
|
|
got := sanitizeGeneratedArticleMarkdown(markdown)
|
||
|
|
|
||
|
|
if got != markdown {
|
||
|
|
t.Fatalf("sanitizeGeneratedArticleMarkdown() = %q, want %q", got, markdown)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNormalizeOutlineNodesRemovesURLsAndWebsiteListLabel(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
nodes := []OutlineNode{
|
||
|
|
{
|
||
|
|
Outline: "网站列表",
|
||
|
|
Children: []OutlineNode{
|
||
|
|
{Outline: "丹福尼门锁官方网站：www.defnit.cn，可查询产品参数"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
got := normalizeOutlineNodes(nodes, 2)
|
||
|
|
if len(got) != 1 || got[0].Outline != "品牌信息整理" {
|
||
|
|
t.Fatalf("normalizeOutlineNodes() = %#v, want top outline sanitized", got)
|
||
|
|
}
|
||
|
|
if len(got[0].Children) != 1 {
|
||
|
|
t.Fatalf("normalizeOutlineNodes() children = %#v, want one child", got[0].Children)
|
||
|
|
}
|
||
|
|
child := got[0].Children[0].Outline
|
||
|
|
if strings.Contains(child, "www.") || strings.Contains(child, "defnit.cn") || strings.Contains(child, "官方网站") {
|
||
|
|
t.Fatalf("normalizeOutlineNodes() child = %q, still contains website text", child)
|
||
|
|
}
|
||
|
|
if !strings.Contains(child, "丹福尼门锁") || !strings.Contains(child, "可查询产品参数") {
|
||
|
|
t.Fatalf("normalizeOutlineNodes() child = %q, lost useful context", child)
|
||
|
|
}
|
||
|
|
}
|