package graph
|
|
import (
|
"testing"
|
|
"github.com/aisim/kb-cli/internal/vault"
|
)
|
|
func TestBuildGraphProvenance(t *testing.T) {
|
files := []*vault.FileMeta{
|
{Path: "FAQ/001-补气失败.md", Title: "补气失败", Wikilinks: []string{"红绿闪", "不存在的链接"}},
|
{Path: "FAQ/002-红绿闪.md", Title: "红绿闪"},
|
{Path: "知识/003-称重原理详解.md", Title: "称重原理详解"}, // "称重原理" 是 "称重原理详解" 的子串 → fuzzy
|
}
|
files[0].Wikilinks = append(files[0].Wikilinks, "称重原理")
|
|
g, unresolved := BuildGraph(files)
|
|
// exact: 标题精确匹配
|
var exact, fuzzy int
|
for _, e := range g.Edges {
|
if e.Relation != "wikilink" {
|
continue
|
}
|
switch e.Provenance {
|
case "exact":
|
exact++
|
case "fuzzy":
|
fuzzy++
|
}
|
}
|
if exact != 1 || fuzzy != 1 {
|
t.Errorf("provenance 分布: exact=%d fuzzy=%d (want 1/1)", exact, fuzzy)
|
}
|
// 悬空链接
|
if len(unresolved) != 1 || unresolved[0].LinkText != "不存在的链接" {
|
t.Errorf("unresolved: %+v", unresolved)
|
}
|
if unresolved[0].NameTail != "不存在的链接" {
|
t.Errorf("name_tail: %s", unresolved[0].NameTail)
|
}
|
}
|
|
func TestBuildGraph(t *testing.T) {
|
files := []*vault.FileMeta{
|
{
|
Path: "FAQ/充装类/001-test.md",
|
Title: "测试FAQ",
|
Section: "FAQ",
|
Tags: []string{"充装", "智能枪"},
|
Entities: []string{"YSP-35.5"},
|
Wikilinks: []string{},
|
Content: "测试内容",
|
},
|
{
|
Path: "知识/002-配置.md",
|
Title: "充装规格配置",
|
Section: "知识",
|
Tags: []string{"充装"},
|
Entities: []string{},
|
Wikilinks: []string{},
|
Content: "配置内容",
|
},
|
}
|
|
g, _ := BuildGraph(files)
|
|
if len(g.Nodes) != 2 {
|
t.Errorf("node count = %d, want 2", len(g.Nodes))
|
}
|
|
// 应该有 3 条边：2条 tag + 1条 entity
|
tagEdges := 0
|
entityEdges := 0
|
for _, e := range g.Edges {
|
if e.Relation == "tag" {
|
tagEdges++
|
}
|
if e.Relation == "entity" {
|
entityEdges++
|
}
|
}
|
if tagEdges != 3 { // 充装 + 智能枪 + 充装(第二个文件)
|
t.Errorf("tag edges = %d, want 3", tagEdges)
|
}
|
if entityEdges != 1 {
|
t.Errorf("entity edges = %d, want 1", entityEdges)
|
}
|
}
|
