更新若干功能

database_test.go

@@ -34,6 +34,62 @@ func TestStoreProjectAndSnapshot(t *testing.T) {
 	}
 }
 
+func TestAISummaryRoundTrip(t *testing.T) {
+	s, e := OpenStore(filepath.Join(t.TempDir(), "test.db"))
+	if e != nil {
+		t.Fatal(e)
+	}
+	defer s.db.Close()
+	p, e := s.SaveProject(0, ProjectInput{Name: "demo", Path: t.TempDir()})
+	if e != nil {
+		t.Fatal(e)
+	}
+	if e = s.SaveAISummary(p.ID, "project", "spark", "第一版介绍"); e != nil {
+		t.Fatal(e)
+	}
+	// 覆盖更新（同项目同模块只保留最新一份）
+	if e = s.SaveAISummary(p.ID, "project", "deepseek", "第二版介绍"); e != nil {
+		t.Fatal(e)
+	}
+	if e = s.SaveAISummary(p.ID, "git", "spark", "Git 介绍"); e != nil {
+		t.Fatal(e)
+	}
+	got, e := s.GetAISummaries(p.ID)
+	if e != nil || len(got) != 2 {
+		t.Fatalf("expected 2 summaries, got %d (%v)", len(got), e)
+	}
+	for _, x := range got {
+		switch x.Kind {
+		case "project":
+			if x.Content != "第二版介绍" || x.Provider != "deepseek" {
+				t.Fatalf("project summary mismatch: %#v", x)
+			}
+		case "git":
+			if x.Content != "Git 介绍" || x.Provider != "spark" {
+				t.Fatalf("git summary mismatch: %#v", x)
+			}
+		}
+	}
+}
+
+func TestAISummaryCleanup(t *testing.T) {
+	if got := stripFence("```markdown\n介绍正文\n- 列表\n```"); got != "介绍正文\n- 列表" {
+		t.Fatalf("stripFence: %q", got)
+	}
+	if got := stripFence("普通文本"); got != "普通文本" {
+		t.Fatalf("stripFence should keep plain text: %q", got)
+	}
+	if got := stripFence("```\n# 标题\n正文没闭合"); got != "# 标题\n正文没闭合" {
+		t.Fatalf("stripFence should drop unclosed opening fence: %q", got)
+	}
+	if got := dedentCommon("    第一行\n    第二行"); got != "第一行\n第二行" {
+		t.Fatalf("dedentCommon: %q", got)
+	}
+	if got := dedentCommon("第一行\n  嵌套保留"); got != "第一行\n  嵌套保留" {
+		t.Fatalf("dedentCommon should keep relative indent: %q", got)
+	}
+}
+
 func TestNormalizePath(t *testing.T) {
 	if _, e := normalizePath(filepath.Join(t.TempDir(), "missing")); e == nil {
 		t.Fatal("missing directory accepted")
@@ -199,6 +255,43 @@ func TestLoadingStylePersistsAndValidates(t *testing.T) {
 	}
 }
 
+func TestMigrateSkippedWhenVersionCurrent(t *testing.T) {
+	db := filepath.Join(t.TempDir(), "versioned.db")
+	s, e := OpenStore(db)
+	if e != nil {
+		t.Fatal(e)
+	}
+	// 删除一张迁移会创建的表：若重开时迁移被跳过，该表应保持缺失。
+	if _, e = s.db.Exec(`DROP TABLE git_hotspots`); e != nil {
+		t.Fatal(e)
+	}
+	s.db.Close()
+	s, e = OpenStore(db)
+	if e != nil {
+		t.Fatal(e)
+	}
+	defer s.db.Close()
+	var n int
+	_ = s.db.QueryRow(`SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='git_hotspots'`).Scan(&n)
+	if n != 0 {
+		t.Fatal("migration ran again despite current schema version")
+	}
+	// 回退版本号后重开，迁移应重新执行并补回该表。
+	if _, e = s.db.Exec(`DELETE FROM schema_migrations`); e != nil {
+		t.Fatal(e)
+	}
+	s.db.Close()
+	s, e = OpenStore(db)
+	if e != nil {
+		t.Fatal(e)
+	}
+	defer s.db.Close()
+	_ = s.db.QueryRow(`SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='git_hotspots'`).Scan(&n)
+	if n != 1 {
+		t.Fatal("migration did not run after version reset")
+	}
+}
+
 func TestLegacyLoadingStyleDefaultMigratesOnce(t *testing.T) {
 	db := filepath.Join(t.TempDir(), "settings.db")
 	s, e := OpenStore(db)
@@ -208,6 +301,10 @@ func TestLegacyLoadingStyleDefaultMigratesOnce(t *testing.T) {
 	if _, e = s.db.Exec(`UPDATE settings SET value='bar' WHERE key='loadingStyle'`); e != nil {
 		t.Fatal(e)
 	}
+	// 模拟旧版本数据库：回退已记录的结构版本，重开时才会执行一次迁移转换。
+	if _, e = s.db.Exec(`DELETE FROM schema_migrations`); e != nil {
+		t.Fatal(e)
+	}
 	s.db.Close()
 	s, e = OpenStore(db)
 	if e != nil {
@@ -237,3 +334,52 @@ func TestLegacyLoadingStyleDefaultMigratesOnce(t *testing.T) {
 		t.Fatalf("explicit loadingStyle=%q", got.LoadingStyle)
 	}
 }
+
+func TestMultiNoteCRUD(t *testing.T) {
+	s, e := OpenStore(filepath.Join(t.TempDir(), "notes.db"))
+	if e != nil {
+		t.Fatal(e)
+	}
+	defer s.db.Close()
+	a, e := s.SaveNoteByID(0, "第一条")
+	if e != nil || a.ID == 0 || a.UUID == "" {
+		t.Fatalf("create note: %#v %v", a, e)
+	}
+	b, e := s.SaveNoteByID(0, "第二条")
+	if e != nil || b.ID == a.ID {
+		t.Fatalf("second note: %#v %v", b, e)
+	}
+	notes, e := s.ListNotes(0)
+	if e != nil || len(notes) != 2 {
+		t.Fatalf("list notes: %d %v", len(notes), e)
+	}
+	if notes[0].ID != b.ID {
+		t.Fatalf("latest first, got %#v", notes[0])
+	}
+	// GetNote 返回最近更新的一条；SaveNote 兼容入口更新它
+	if n, _ := s.GetNote(); n.ID != b.ID {
+		t.Fatalf("GetNote should return latest, got %d", n.ID)
+	}
+	if a2, e := s.SaveNoteByID(a.ID, "第一条改"); e != nil || a2.Content != "第一条改" {
+		t.Fatalf("update note: %#v %v", a2, e)
+	}
+	// 更新过的 a 变为最近一条
+	if n, _ := s.GetNote(); n.ID != a.ID {
+		t.Fatalf("GetNote should follow update, got %d", n.ID)
+	}
+	if _, e := s.SaveNoteByID(99999, "missing"); e == nil || e.Error() != "NOTE_NOT_FOUND" {
+		t.Fatalf("expect NOTE_NOT_FOUND, got %v", e)
+	}
+	if e := s.DeleteNote(b.ID); e != nil {
+		t.Fatal(e)
+	}
+	if notes, _ = s.ListNotes(0); len(notes) != 1 || notes[0].ID != a.ID {
+		t.Fatalf("after delete: %#v", notes)
+	}
+	// 软删行仍在表中且 dirty（供同步推送）
+	var deleted, dirty int
+	_ = s.db.QueryRow(`SELECT deleted,dirty FROM notes WHERE id=?`, b.ID).Scan(&deleted, &dirty)
+	if deleted != 1 || dirty != 1 {
+		t.Fatalf("soft delete flags: deleted=%d dirty=%d", deleted, dirty)
+	}
+}