100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
package store
|
|
|
|
import (
|
|
"database/sql"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"myaibot/internal/config"
|
|
)
|
|
|
|
func openTestDB(t *testing.T) *sql.DB {
|
|
t.Helper()
|
|
cfg := &config.DatabaseConfig{
|
|
Driver: "sqlite3",
|
|
File: filepath.Join(t.TempDir(), "memory.db"),
|
|
}
|
|
db, err := Open(cfg)
|
|
if err != nil {
|
|
t.Fatalf("Open 出错: %v", err)
|
|
}
|
|
t.Cleanup(func() { Close(db) })
|
|
if err := Migrate(db, "sqlite3"); err != nil {
|
|
t.Fatalf("Migrate 出错: %v", err)
|
|
}
|
|
return db
|
|
}
|
|
|
|
func TestSessionRoundtrip(t *testing.T) {
|
|
db := openTestDB(t)
|
|
first := &Session{
|
|
Provider: "deepseek",
|
|
Model: "deepseek-v4-flash",
|
|
Messages: []Message{{Role: "user", Content: "你好"}, {Role: "assistant", Content: "你好！"}},
|
|
}
|
|
id1, err := SaveSession(db, first)
|
|
if err != nil {
|
|
t.Fatalf("SaveSession 出错: %v", err)
|
|
}
|
|
second := &Session{
|
|
Provider: "deepseek",
|
|
Model: "deepseek-v4-flash",
|
|
Messages: []Message{{Role: "user", Content: "现在几点"}},
|
|
}
|
|
id2, err := SaveSession(db, second)
|
|
if err != nil {
|
|
t.Fatalf("SaveSession 出错: %v", err)
|
|
}
|
|
if id2 <= id1 {
|
|
t.Errorf("id2 (%d) 应大于 id1 (%d)", id2, id1)
|
|
}
|
|
|
|
latest, err := LoadLatestSession(db)
|
|
if err != nil {
|
|
t.Fatalf("LoadLatestSession 出错: %v", err)
|
|
}
|
|
if latest == nil || latest.ID != id2 {
|
|
t.Errorf("最新会话应为 #%d, got %+v", id2, latest)
|
|
}
|
|
if len(latest.Messages) != 1 || latest.Messages[0].Content != "现在几点" {
|
|
t.Errorf("消息还原异常: %+v", latest.Messages)
|
|
}
|
|
|
|
byID, err := LoadSession(db, id1)
|
|
if err != nil {
|
|
t.Fatalf("LoadSession 出错: %v", err)
|
|
}
|
|
if byID == nil || len(byID.Messages) != 2 {
|
|
t.Errorf("按 id 加载异常: %+v", byID)
|
|
}
|
|
|
|
list, err := ListSessions(db)
|
|
if err != nil {
|
|
t.Fatalf("ListSessions 出错: %v", err)
|
|
}
|
|
if len(list) != 2 {
|
|
t.Errorf("列表数量 = %d, want 2", len(list))
|
|
}
|
|
if list[0].ID != id2 || list[0].MessageCount != 1 {
|
|
t.Errorf("列表首条应为最新会话: %+v", list[0])
|
|
}
|
|
}
|
|
|
|
func TestLoadSessionMissing(t *testing.T) {
|
|
db := openTestDB(t)
|
|
sess, err := LoadSession(db, 999)
|
|
if err != nil {
|
|
t.Fatalf("LoadSession 出错: %v", err)
|
|
}
|
|
if sess != nil {
|
|
t.Errorf("不存在的会话应返回 nil, got %+v", sess)
|
|
}
|
|
latest, err := LoadLatestSession(db)
|
|
if err != nil {
|
|
t.Fatalf("LoadLatestSession 出错: %v", err)
|
|
}
|
|
if latest != nil {
|
|
t.Errorf("空库最新会话应为 nil, got %+v", latest)
|
|
}
|
|
}
|