39 lines
1.1 KiB
Go
39 lines
1.1 KiB
Go
package database
|
|
|
|
import (
|
|
"testing"
|
|
"yingce/backend/internal/model"
|
|
)
|
|
|
|
func TestTaskMediaRecoveryMigrationPreservesHistoricalTasks(t *testing.T) {
|
|
db, err := Open(Config{Driver: "sqlite", DSN: ":memory:"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
sqlDB, _ := db.DB()
|
|
t.Cleanup(func() { _ = sqlDB.Close() })
|
|
if err := db.Exec(`CREATE TABLE tasks (id TEXT PRIMARY KEY, status TEXT, error TEXT)`).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.Exec(`INSERT INTO tasks VALUES ('old-task', 'failed', 'download failed')`).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for i := 0; i < 2; i++ {
|
|
if err := schemaMigrations[33].apply(db); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
var task model.Task
|
|
if err := db.First(&task, "id = ?", "old-task").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if task.Status != model.TaskStatusFailed || task.Error != "download failed" || task.MediaRecoveryJSON != "" || task.MediaStage != "" {
|
|
t.Fatalf("migration rewrote historical task: %+v", task)
|
|
}
|
|
for _, field := range []string{"media_stage", "media_recovery_json"} {
|
|
if !db.Migrator().HasColumn(&model.Task{}, field) {
|
|
t.Fatalf("missing %s", field)
|
|
}
|
|
}
|
|
}
|