feat(routing): routing_profiles 表 + Profile 类型 + store(双方言)
This commit is contained in:
@@ -0,0 +1,62 @@
|
||||
package routing
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/wangjia/pangolin/server/internal/config"
|
||||
"github.com/wangjia/pangolin/server/internal/store"
|
||||
)
|
||||
|
||||
func openDB(t *testing.T) *sql.DB {
|
||||
t.Helper()
|
||||
db, err := store.Open(&config.Config{Driver: "sqlite", DSN: ":memory:"})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Cleanup(func() { _ = db.Close() })
|
||||
if err := store.MigrateUp(db, "sqlite"); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
_ = store.ApplyCodesLibMigrations(context.Background(), db, "sqlite")
|
||||
return db
|
||||
}
|
||||
|
||||
func seedU(t *testing.T, db *sql.DB, id int64, uuid string) {
|
||||
t.Helper()
|
||||
if _, err := db.Exec(`INSERT INTO users (id,uuid,email,pw_hash,dp_uuid,status,created_at)
|
||||
VALUES (?,?,?, 'x','dp-'||?, 'active', ?)`, id, uuid, uuid+"@x", uuid, time.Now().UTC()); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSQLiteRoutingStoreUpsertGet(t *testing.T) {
|
||||
db := openDB(t) // 内存库 + MigrateUp(sqlite)
|
||||
seedU(t, db, 1, "u1")
|
||||
st := NewStore(db)
|
||||
ctx := context.Background()
|
||||
// 无档案 → nil,nil
|
||||
got, err := st.Get(ctx, 1)
|
||||
if err != nil || got != nil {
|
||||
t.Fatalf("empty want nil,nil got %v,%v", got, err)
|
||||
}
|
||||
// Upsert 后可取回
|
||||
p := Default()
|
||||
p.Final = "direct"
|
||||
if err := st.Upsert(ctx, 1, p); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
got, err = st.Get(ctx, 1)
|
||||
if err != nil || got == nil || got.Final != "direct" {
|
||||
t.Fatalf("got %v,%v", got, err)
|
||||
}
|
||||
// 二次 Upsert 覆盖
|
||||
p.Final = "proxy"
|
||||
_ = st.Upsert(ctx, 1, p)
|
||||
got, _ = st.Get(ctx, 1)
|
||||
if got.Final != "proxy" {
|
||||
t.Fatalf("upsert overwrite failed: %s", got.Final)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user