package output
|
|
import (
|
"strings"
|
"testing"
|
|
"github.com/aisim/kb-cli/internal/search"
|
)
|
|
func TestFormatTable(t *testing.T) {
|
results := []search.SearchResult{
|
{Path: "FAQ/充装类/001.md", Title: "充装问题", Section: "FAQ", Score: 100},
|
}
|
output := FormatTable(results)
|
if !strings.Contains(output, "FAQ") {
|
t.Error("output should contain section")
|
}
|
if !strings.Contains(output, "100") {
|
t.Error("output should contain score")
|
}
|
}
|
|
func TestFormatTableEmpty(t *testing.T) {
|
output := FormatTable(nil)
|
if output != "未找到匹配结果" {
|
t.Errorf("output = %q, want %q", output, "未找到匹配结果")
|
}
|
}
|
|
func TestFormatJSON(t *testing.T) {
|
results := []search.SearchResult{
|
{Path: "test.md", Title: "Test", Score: 50},
|
}
|
output, err := FormatJSON(results)
|
if err != nil {
|
t.Fatalf("FormatJSON failed: %v", err)
|
}
|
if !strings.Contains(output, "test.md") {
|
t.Error("JSON should contain path")
|
}
|
}
|
