25 lines
560 B
Go
25 lines
560 B
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"go.uber.org/zap/zaptest"
|
||
|
|
"gorm.io/driver/sqlite"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// 使用内存 sqlite 验证 AutoMigrate 关键路径，无需真实 Postgres
|
||
|
|
func TestAutoMigrate_WithSQLite(t *testing.T) {
|
||
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("open sqlite err: %v", err)
|
||
|
|
}
|
||
|
|
dbInstance = db
|
||
|
|
defer func() { dbInstance = nil }()
|
||
|
|
|
||
|
|
logger := zaptest.NewLogger(t)
|
||
|
|
if err := AutoMigrate(logger); err != nil {
|
||
|
|
t.Fatalf("AutoMigrate sqlite err: %v", err)
|
||
|
|
}
|
||
|
|
}
|