From 09c2a03cefa687832257d3b816da50db3cfcc0f2 Mon Sep 17 00:00:00 2001
From: ai_xiaopei <xiaopei@aisim.cn>
Date: Sat, 25 Jul 2026 23:48:11 +0800
Subject: [PATCH] feat: 实现输出格式化
---
internal/search/engine_test.go | 98 +++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 98 insertions(+), 0 deletions(-)
diff --git a/internal/search/engine_test.go b/internal/search/engine_test.go
new file mode 100644
index 0000000..2cd91d3
--- /dev/null
+++ b/internal/search/engine_test.go
@@ -0,0 +1,98 @@
+package search
+
+import (
+ "path/filepath"
+ "strings"
+ "testing"
+
+ "github.com/aisim/kb-cli/internal/graph"
+ "github.com/aisim/kb-cli/internal/index"
+)
+
+func TestSearch(t *testing.T) {
+ // 创建临时数据库
+ tmpDir := t.TempDir()
+ dbPath := filepath.Join(tmpDir, "test.db")
+
+ store, err := index.Open(dbPath)
+ if err != nil {
+ t.Fatalf("Open failed: %v", err)
+ }
+ defer store.Close()
+
+ // 插入测试数据
+ nodes := []*graph.Node{
+ {
+ ID: 1,
+ Path: "FAQ/充装类/001-test.md",
+ Title: "充装问题排查",
+ Section: "FAQ",
+ Tags: []string{"充装"},
+ Content: "关于智能枪充装问题的排查方法",
+ },
+ {
+ ID: 2,
+ Path: "知识/002-config.md",
+ Title: "充装规格配置",
+ Section: "知识",
+ Tags: []string{"配置"},
+ Content: "充装规格配置说明",
+ },
+ }
+
+ for _, n := range nodes {
+ _, err := store.InsertNode(n)
+ if err != nil {
+ t.Fatalf("InsertNode failed: %v", err)
+ }
+ }
+
+ // 创建 FTS 索引
+ if err := store.CreateFTS(); err != nil {
+ t.Fatalf("CreateFTS failed: %v", err)
+ }
+ if err := store.PopulateFTS(); err != nil {
+ t.Fatalf("PopulateFTS failed: %v", err)
+ }
+
+ // 测试搜索
+ results, err := Search(store, []string{"充装"}, SearchOptions{})
+ if err != nil {
+ t.Fatalf("Search failed: %v", err)
+ }
+
+ if len(results) == 0 {
+ t.Error("Expected results, got none")
+ }
+
+ // 验证结果包含关键词
+ found := false
+ for _, r := range results {
+ if strings.Contains(r.Title, "充装") || strings.Contains(r.Path, "充装") {
+ found = true
+ break
+ }
+ }
+ if !found {
+ t.Error("Expected results to contain '充装'")
+ }
+}
+
+func TestIsGenericWord(t *testing.T) {
+ tests := []struct {
+ word string
+ expected bool
+ }{
+ {"问题", true},
+ {"故障", true},
+ {"充装", false},
+ {"智能枪", false},
+ }
+
+ for _, tt := range tests {
+ result := IsGenericWord(tt.word)
+ if result != tt.expected {
+ t.Errorf("IsGenericWord(%q) = %v, want %v", tt.word, result, tt.expected)
+ }
+ }
+}
--
Gitblit v1.9.1