48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
// Package testdb provides a pure-Go SQLite helper for tests.
|
||
// It uses modernc.org/sqlite (CGO-free) via gorm.io/driver/sqlite's DriverName override.
|
||
package testdb
|
||
|
||
import (
|
||
"testing"
|
||
|
||
_ "modernc.org/sqlite" // 注册纯Go SQLite驱动，驱动名 "sqlite"
|
||
gormsqlite "gorm.io/driver/sqlite"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/logger"
|
||
)
|
||
|
||
// Open 使用 modernc.org/sqlite（纯Go，无需CGO）打开内存测试数据库。
|
||
// 驱动名必须是 "sqlite"（modernc 注册），而非 gorm 默认的 "sqlite3"（mattn/CGO）。
|
||
func Open(t testing.TB) *gorm.DB {
|
||
t.Helper()
|
||
|
||
db, err := gorm.Open(gormsqlite.New(gormsqlite.Config{
|
||
DriverName: "sqlite",
|
||
DSN: "file::memory:?cache=shared&mode=memory",
|
||
}), &gorm.Config{
|
||
Logger: logger.Default.LogMode(logger.Silent),
|
||
})
|
||
if err != nil {
|
||
t.Skipf("跳过数据库测试（SQLite不可用）: %v", err)
|
||
}
|
||
|
||
return db
|
||
}
|
||
|
||
// OpenWith 使用自定义DSN
|
||
func OpenWith(t testing.TB, dsn string) *gorm.DB {
|
||
t.Helper()
|
||
|
||
db, err := gorm.Open(gormsqlite.New(gormsqlite.Config{
|
||
DriverName: "sqlite",
|
||
DSN: dsn,
|
||
}), &gorm.Config{
|
||
Logger: logger.Default.LogMode(logger.Silent),
|
||
})
|
||
if err != nil {
|
||
t.Skipf("跳过数据库测试（SQLite不可用）: %v", err)
|
||
}
|
||
|
||
return db
|
||
}
|