105 lines
3.0 KiB
Go
105 lines
3.0 KiB
Go
package database
|
|
|
|
import (
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"path/filepath"
|
|
"testing"
|
|
"yingce/backend/internal/model"
|
|
)
|
|
|
|
func TestSkillCurationUpgradeAndRollback(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open(filepath.Join(t.TempDir(), "upgrade.db")), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
sqlDB, _ := db.DB()
|
|
defer sqlDB.Close()
|
|
if err := db.AutoMigrate(&schemaMigration{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, m := range schemaMigrations {
|
|
if m.version > 43 {
|
|
break
|
|
}
|
|
if err := db.Transaction(func(tx *gorm.DB) error {
|
|
if err := m.apply(tx); err != nil {
|
|
return err
|
|
}
|
|
return tx.Create(&schemaMigration{Version: m.version, Name: m.name, Checksum: m.checksum}).Error
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
if db.Migrator().HasTable(&model.SkillCurationSetting{}) {
|
|
t.Fatal("fixture is not an original schema")
|
|
}
|
|
if err := db.Create(&model.Skill{ID: "existing", Name: "Existing", Tag: "drama", Status: 1}).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.Exec(`CREATE TRIGGER reject_curation_migration BEFORE INSERT ON schema_migrations WHEN NEW.version = 46 BEGIN SELECT RAISE(ABORT, 'migration interrupted'); END`).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := MigrateSchema(db); err == nil {
|
|
t.Fatal("forced migration failure accepted")
|
|
}
|
|
if db.Migrator().HasTable(&model.SkillCurationSetting{}) {
|
|
t.Fatal("failed migration left partial tables")
|
|
}
|
|
if err := db.Exec("DROP TRIGGER reject_curation_migration").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := MigrateSchema(db); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := MigrateSchema(db); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var skill model.Skill
|
|
if err := db.First(&skill, "id = ?", "existing").Error; err != nil || skill.Tag != "drama" {
|
|
t.Fatalf("existing skill changed: %#v %v", skill, err)
|
|
}
|
|
var setting model.SkillCurationSetting
|
|
if err := db.First(&setting, 1).Error; err != nil || setting.Enabled {
|
|
t.Fatalf("upgrade enabled curation: %#v %v", setting, err)
|
|
}
|
|
}
|
|
|
|
func TestSkillCurationMigration(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open(filepath.Join(t.TempDir(), "migration.db")), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
sqlDB, _ := db.DB()
|
|
defer sqlDB.Close()
|
|
for i := 0; i < 2; i++ {
|
|
if err := db.Transaction(migrateSkillCuration); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
var state model.SkillCurationSetting
|
|
if err := db.First(&state, 1).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if state.Enabled || state.Revision != 0 {
|
|
t.Fatalf("migration must default off: %#v", state)
|
|
}
|
|
db.Model(&state).Updates(map[string]any{"enabled": true, "revision": 3})
|
|
if err := db.Transaction(migrateSkillCuration); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
db.First(&state, 1)
|
|
if !state.Enabled || state.Revision != 3 {
|
|
t.Fatal("migration overwrote settings")
|
|
}
|
|
found := false
|
|
for _, m := range schemaMigrations {
|
|
if m.version == 46 {
|
|
found = m.name == "skill_curation" && m.checksum == "sha256:skill-curation-v46-20261005"
|
|
}
|
|
}
|
|
if !found {
|
|
t.Fatal("migration 46 is not registered")
|
|
}
|
|
}
|