131 lines
3.4 KiB
Go
131 lines
3.4 KiB
Go
package docDB
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"apigo.cc/go/log"
|
|
"apigo.cc/go/tableDB"
|
|
)
|
|
|
|
func TestDocDB(t *testing.T) {
|
|
logger := log.DefaultLogger
|
|
dbFile := "test_doc.db"
|
|
os.Remove(dbFile)
|
|
defer os.Remove(dbFile)
|
|
os.RemoveAll("./test_docs")
|
|
defer os.RemoveAll("./test_docs")
|
|
|
|
unauthorizedDB := tableDB.GetDB("sqlite://"+dbFile, logger)
|
|
|
|
// 使用新的 GetDB 入口，内部完成系统表初始化与业务表初始化
|
|
docDBInst := GetDB(unauthorizedDB, "./test_docs")
|
|
app := docDBInst.Auth("user1")
|
|
|
|
var createdCalled, updatedCalled, removedCalled bool
|
|
docDBInst.OnCreatedDoc(func(doc *Document) { createdCalled = true })
|
|
docDBInst.OnUpdatedDoc(func(doc *Document) { updatedCalled = true })
|
|
docDBInst.OnRemoved(func(id string) { removedCalled = true })
|
|
|
|
// 1. 测试 SetDoc (创建)
|
|
doc1 := &Document{
|
|
Path: "/test/doc1.md",
|
|
Title: "Test Doc",
|
|
TextContent: "# Header 1\nContent 1",
|
|
Type: "markdown",
|
|
}
|
|
|
|
err := app.SetDoc(doc1)
|
|
if err != nil {
|
|
t.Fatalf("SetDoc failed: %v", err)
|
|
}
|
|
|
|
if doc1.Version != 1 {
|
|
t.Fatalf("Expected version 1, got %d", doc1.Version)
|
|
}
|
|
if !createdCalled {
|
|
t.Error("OnCreatedDoc event not triggered")
|
|
}
|
|
|
|
// 2. 测试 Get
|
|
d, err := app.Get(doc1.ID)
|
|
if err != nil || d == nil {
|
|
t.Fatalf("Get failed: %v", err)
|
|
}
|
|
if d.Title != "Test Doc" {
|
|
t.Fatalf("Expected title 'Test Doc', got %s", d.Title)
|
|
}
|
|
|
|
// 3. 测试 SetDoc (更新内容, 提升版本并存入历史)
|
|
doc1.TextContent = "# Header 1\nUpdated Content"
|
|
err = app.SetDoc(doc1)
|
|
if err != nil {
|
|
t.Fatalf("Update SetDoc failed: %v", err)
|
|
}
|
|
if doc1.Version != 2 {
|
|
t.Fatalf("Expected version 2 after content update, got %d", doc1.Version)
|
|
}
|
|
if !updatedCalled {
|
|
t.Error("OnUpdatedDoc event not triggered")
|
|
}
|
|
|
|
// 4. 测试 GetByVersion
|
|
hist, err := app.GetByVersion(doc1.ID, 1)
|
|
if err != nil || hist == nil {
|
|
t.Fatalf("GetByVersion(1) failed: %v", err)
|
|
}
|
|
if !strings.Contains(hist.TextContent, "Content 1") {
|
|
t.Fatalf("Expected 'Content 1' in history version 1, got %s", hist.TextContent)
|
|
}
|
|
|
|
curr, _ := app.GetByVersion(doc1.ID, 2)
|
|
if curr.Version != 2 || !strings.Contains(curr.TextContent, "Updated Content") {
|
|
t.Fatalf("Expected version 2 content, got version %d", curr.Version)
|
|
}
|
|
|
|
// 5. 测试 ExtractToC 代码块识别
|
|
docWithCode := &Document{
|
|
Path: "/test/code.md",
|
|
TextContent: "# H1\n```markdown\n# This is not a header\n```\n# H2",
|
|
Type: "markdown",
|
|
}
|
|
_ = app.SetDoc(docWithCode)
|
|
if len(docWithCode.ToC) != 2 {
|
|
t.Fatalf("Expected 2 headers in ToC, got %d", len(docWithCode.ToC))
|
|
}
|
|
if docWithCode.ToC[1].Title != "H2" {
|
|
t.Fatalf("Expected second header to be 'H2', got %s", docWithCode.ToC[1].Title)
|
|
}
|
|
|
|
// 6. 测试 SetMeta (更新元数据, 不提升版本)
|
|
err = app.SetMeta(doc1.ID, map[string]any{"title": "New Title"})
|
|
if err != nil {
|
|
t.Fatalf("SetMeta failed: %v", err)
|
|
}
|
|
|
|
d2, _ := app.Get(doc1.ID)
|
|
if d2.Title != "New Title" {
|
|
t.Fatalf("Expected title 'New Title', got %s", d2.Title)
|
|
}
|
|
if d2.Version != 2 {
|
|
t.Fatalf("Expected version 2 after meta update, got %d", d2.Version)
|
|
}
|
|
|
|
// 7. 测试 Remove
|
|
err = app.Remove(doc1.ID)
|
|
if err != nil {
|
|
t.Fatalf("Remove failed: %v", err)
|
|
}
|
|
if !removedCalled {
|
|
t.Error("OnRemoved event not triggered")
|
|
}
|
|
}
|
|
|
|
func BenchmarkExtractToC(b *testing.B) {
|
|
content := "# H1\n## H1.1\n### H1.1.1\n" + strings.Repeat("Some content\n", 100) + "# H2\n## H2.1\n"
|
|
for i := 0; i < b.N; i++ {
|
|
_ = ExtractToC(content)
|
|
}
|
|
}
|