yonlu
1179f56c36
- Add backupDatabase() for timestamped .db file backup before migration - Add migration6MultiLibrary() with all 14 steps: FK OFF, create libraries table, insert default library from TOML config, add library_id to audio_files, rebuild playlist_tracks with SET NULL FK and 6 phantom columns, backfill phantom metadata, recreate track_metadata VIEW with library_id, FK ON, clean TOML config - Add readLibraryDirFromTOML() and removeLibraryDirFromTOML() helpers - Update runMigrations signature to accept dbPath for backup - Add sentinel library row in NewTestDB for FK constraint satisfaction
84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
"io/fs"
|
|
"log/slog"
|
|
"path"
|
|
"testing"
|
|
|
|
_ "modernc.org/sqlite" // Register sqlite driver.
|
|
|
|
"yellowjacket/backend/database/sql/sqlcgen"
|
|
)
|
|
|
|
// NewTestDB returns an in-memory SQLite database that mirrors the
|
|
// production setup (PRAGMAs + all migrations). The database is
|
|
// automatically closed when the test completes via t.Cleanup.
|
|
func NewTestDB(t *testing.T) *DB {
|
|
t.Helper()
|
|
|
|
db, err := sql.Open(
|
|
"sqlite",
|
|
":memory:?_busy_timeout=5000&_journal_mode=WAL",
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("could not open test database: %v", err)
|
|
}
|
|
|
|
db.SetMaxOpenConns(1)
|
|
|
|
ctx := t.Context()
|
|
|
|
if err := applyPRAGMAs(ctx, db); err != nil {
|
|
t.Fatalf("could not apply PRAGMAs: %v", err)
|
|
}
|
|
|
|
dirEntries, err := schemas.ReadDir("sql/schemas")
|
|
if err != nil {
|
|
t.Fatalf("could not read schemas directory: %v", err)
|
|
}
|
|
|
|
for _, dirEntry := range dirEntries {
|
|
if !dirEntry.IsDir() {
|
|
filePath := path.Join("sql/schemas", dirEntry.Name())
|
|
|
|
sqlContent, err := fs.ReadFile(schemas, filePath)
|
|
if err != nil {
|
|
t.Fatalf("could not read file %s: %v", filePath, err)
|
|
}
|
|
|
|
if _, err = db.ExecContext(ctx, string(sqlContent)); err != nil {
|
|
t.Fatalf(
|
|
"error executing sql from file %s: %v",
|
|
filePath, err,
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
if err := runMigrations(ctx, db, slog.Default(), ":memory:"); err != nil {
|
|
t.Fatalf("could not run migrations: %v", err)
|
|
}
|
|
|
|
// Insert a sentinel library row at id=0 so audio_files inserts
|
|
// using the DEFAULT library_id=0 satisfy the FK constraint.
|
|
if _, err := db.ExecContext(
|
|
ctx,
|
|
"INSERT INTO libraries (id, name, path) VALUES (0, 'Test', '/test')",
|
|
); err != nil {
|
|
t.Fatalf("could not insert test library: %v", err)
|
|
}
|
|
|
|
queries := sqlcgen.New(db)
|
|
|
|
t.Cleanup(func() { _ = db.Close() })
|
|
|
|
return &DB{
|
|
db: db,
|
|
Ctx: ctx,
|
|
Queries: queries,
|
|
logger: slog.Default(),
|
|
}
|
|
}
|