yonluandClaude Sonnet 5
e190fd75b9
Ships the fresh-start schema cleanup: rebuilt explore catalog index pipeline (dump import, artifact fetch/build, incremental listen-count refresh), a new download subsystem (Lidarr/Prowlarr/qBittorrent/SABnzbd/ slskd/yt-dlp providers, staging, reconciliation, wanted list), and the supporting schema/query/store changes across backend and frontend. Also includes two smaller follow-ups: bump the central index's rebuild-after cadence from 90 to 180 days, and remove the Explore "library only" online/offline toggle entirely (frontend-only, no backend counterpart) rather than carry unused UI/state. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Y2Agd9af5hE7qzti2ackiS
88 lines
1.9 KiB
Go
88 lines
1.9 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
"log/slog"
|
|
"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)
|
|
}
|
|
|
|
// The same call production uses, so a test database and a real one
|
|
// cannot diverge.
|
|
if err := applySchema(ctx, db); err != nil {
|
|
t.Fatalf("could not apply schema: %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,
|
|
// The in-memory test DB shares one connection, so reads and
|
|
// writes use the same handle; ReadQueries aliases Queries.
|
|
Queries: queries,
|
|
ReadQueries: queries,
|
|
logger: slog.Default(),
|
|
}
|
|
}
|
|
|
|
// NewTestDBWithLibrary returns a test DB with a library row
|
|
// pre-inserted. Returns the DB and the library ID.
|
|
func NewTestDBWithLibrary(
|
|
t *testing.T,
|
|
name, libPath string,
|
|
) (*DB, int64) {
|
|
t.Helper()
|
|
|
|
db := NewTestDB(t)
|
|
|
|
lib, err := db.Queries.CreateLibrary(
|
|
db.Ctx,
|
|
sqlcgen.CreateLibraryParams{
|
|
Name: name,
|
|
Path: libPath,
|
|
},
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("could not create test library: %v", err)
|
|
}
|
|
|
|
return db, lib.ID
|
|
}
|