tt/ttstd-infinite-canvas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(database): 认证通知 - 补齐 v35 增量迁移

Browse Source
This commit is contained in:
ddcat666
2026-09-24 12:44:18 +08:00
parent 28e9eff01f
commit d93130f352
4 changed files with 77 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
backend/internal/database/migrations_test.go
View File

@@ -66,11 +66,67 @@ func TestMigrateSchemaRecordsAndValidatesVersion(t *testing.T) {
if !db.Migrator().HasColumn(&model.BannerAnnouncement{}, "notice_type") {
t.Fatal("schema migration v22 did not create banner announcements notice_type")
}
if !db.Migrator().HasTable(&model.AuthVerification{}) ||
!db.Migrator().HasTable(&model.NotificationQuota{}) ||
!db.Migrator().HasTable(&model.SMSChannel{}) ||
!db.Migrator().HasTable(&model.SMSRecord{}) {
t.Fatal("schema migration v35 did not create auth notification tables")
}
if !db.Migrator().HasColumn(&model.User{}, "phone") ||
!db.Migrator().HasColumn(&model.User{}, "email_verified_at") ||
!db.Migrator().HasColumn(&model.User{}, "phone_verified_at") ||
!db.Migrator().HasColumn(&model.EmailVerificationCode{}, "attempts") {
t.Fatal("schema migration v35 did not add authentication verification fields")
}
if err := MigrateSchema(db); err != nil {
t.Fatalf("migration should be idempotent: %v", err)
}
}
func TestMigrateSchemaV35UpgradesExistingDatabase(t *testing.T) {
db, err := Open(Config{Driver: "sqlite", DSN: "file:migration-auth-notifications-v35?mode=memory&cache=shared"})
if err != nil {
t.Fatal(err)
}
if err := MigrateSchema(db); err != nil {
t.Fatal(err)
}
for _, table := range []any{&model.AuthVerification{}, &model.NotificationQuota{}, &model.SMSChannel{}, &model.SMSRecord{}} {
if err := db.Migrator().DropTable(table); err != nil {
t.Fatal(err)
}
}
for _, column := range []string{"phone", "email_verified_at", "phone_verified_at"} {
if err := db.Migrator().DropColumn(&model.User{}, column); err != nil {
t.Fatal(err)
}
}
if err := db.Migrator().DropColumn(&model.EmailVerificationCode{}, "attempts"); err != nil {
t.Fatal(err)
}
if err := db.Where("version = ?", 35).Delete(&schemaMigration{}).Error; err != nil {
t.Fatal(err)
}
if err := MigrateSchema(db); err != nil {
t.Fatalf("upgrade from v34: %v", err)
}
for _, table := range []any{&model.AuthVerification{}, &model.NotificationQuota{}, &model.SMSChannel{}, &model.SMSRecord{}} {
if !db.Migrator().HasTable(table) {
t.Fatalf("migration v35 did not restore table %T", table)
}
}
for _, column := range []string{"phone", "email_verified_at", "phone_verified_at"} {
if !db.Migrator().HasColumn(&model.User{}, column) {
t.Fatalf("migration v35 did not restore users.%s", column)
}
}
if !db.Migrator().HasColumn(&model.EmailVerificationCode{}, "attempts") {
t.Fatal("migration v35 did not restore email verification attempts")
}
}
func TestMigrateSchemaV15UpgradesExistingDatabase(t *testing.T) {
db, err := Open(Config{Driver: "sqlite", DSN: "file:migration-agent-profiles-v15?mode=memory&cache=shared"})
if err != nil {