1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
| package graph
|
| import (
| "strings"
| "testing"
|
| "github.com/aisim/kb-cli/internal/vault"
| )
|
| 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)
|
| // 2 个文件节点 + 2 个 tag 虚拟节点（充装、智能枪）+ 1 个 entity 虚拟节点
| if len(g.Nodes) != 5 {
| t.Errorf("node count = %d, want 5", len(g.Nodes))
| }
|
| // 验证虚拟节点 path 命名空间与 NodeType
| tagCount, entityCount, fileCount := 0, 0, 0
| for _, n := range g.Nodes {
| switch n.NodeType {
| case "tag":
| tagCount++
| if !strings.HasPrefix(n.Path, "tag:") {
| t.Errorf("tag node path = %q, want prefix tag:", n.Path)
| }
| case "entity":
| entityCount++
| if !strings.HasPrefix(n.Path, "entity:") {
| t.Errorf("entity node path = %q, want prefix entity:", n.Path)
| }
| case "file":
| fileCount++
| default:
| t.Errorf("unexpected node_type %q for %s", n.NodeType, n.Path)
| }
| }
| if tagCount != 2 || entityCount != 1 || fileCount != 2 {
| t.Errorf("file/tag/entity = %d/%d/%d, want 2/2/1", fileCount, tagCount, entityCount)
| }
|
| // 应该有 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)
| }
| }
|
|
