package codes_test import ( "context" "database/sql" "fmt" "testing" "time" "github.com/wangjia/pangolin/server/internal/config" "github.com/wangjia/pangolin/server/internal/store" ) // openMigratedSQLite opens an in-memory sqlite DB, runs pangolin's full // golang-migrate chain (000001-000020) followed by the shared codes // library's own migrations. Shared by backfill_test.go and // service_sqlite_test.go. func openMigratedSQLite(t *testing.T) *sql.DB { t.Helper() db, err := store.Open(&config.Config{Driver: "sqlite", DSN: ":memory:"}) if err != nil { t.Fatalf("open: %v", err) } t.Cleanup(func() { _ = db.Close() }) if err := store.MigrateUp(db, "sqlite"); err != nil { t.Fatalf("migrate: %v", err) } if err := store.ApplyCodesLibMigrations(context.Background(), db, "sqlite"); err != nil { t.Fatalf("lib migrate: %v", err) } return db } // seedUser inserts a minimal `users` row so redeem tests can satisfy the // subscriptions/audit_log foreign keys. func seedUser(t *testing.T, db *sql.DB, id int64) { t.Helper() if _, err := db.Exec( `INSERT INTO users (id, uuid, email, pw_hash, dp_uuid, status, created_at) VALUES (?, ?, ?, 'x', ?, 'active', ?)`, id, fmt.Sprintf("u-%d", id), fmt.Sprintf("u%d@example.com", id), fmt.Sprintf("dp-%d", id), time.Now().UTC()); err != nil { t.Fatalf("seed user: %v", err) } }