Rogeeandrogee
798ea43c2f
* HH-442: isolate runtime processes and harden shutdown * HH-442: harden worker shutdown races * HH-442: gate dependency shutdown on active handlers --------- Co-authored-by: Rogee <rogee@ipao.vip>
83 lines
2.8 KiB
Go
83 lines
2.8 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestMigrationCommandsRejectBlankPath(t *testing.T) {
|
|
tests := map[string]func() error{
|
|
"run": func() error { return RunMigrations("invalid-url", " \t") },
|
|
"steps": func() error { return MigrateSteps("invalid-url", " \t", 1) },
|
|
"rollback": func() error { return RollbackMigrations("invalid-url", " \t") },
|
|
"force": func() error { return ForceVersion("invalid-url", " \t", 1) },
|
|
"version": func() error {
|
|
_, _, err := CurrentVersion("invalid-url", " \t")
|
|
return err
|
|
},
|
|
}
|
|
|
|
for name, run := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
require.EqualError(t, run(), "migration path must not be empty")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestWithSanitizedPQEnvironment(t *testing.T) {
|
|
t.Setenv("PGSERVICE", "local-service")
|
|
t.Setenv("PGSERVICEFILE", "/tmp/pg_service.conf")
|
|
t.Setenv("PGREALM", "local-realm")
|
|
|
|
err := withSanitizedPQEnvironment(func() error {
|
|
for _, key := range []string{"PGSERVICE", "PGSERVICEFILE", "PGREALM"} {
|
|
_, exists := os.LookupEnv(key)
|
|
require.False(t, exists, "%s should be unset during migration", key)
|
|
}
|
|
return nil
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, "local-service", os.Getenv("PGSERVICE"))
|
|
require.Equal(t, "/tmp/pg_service.conf", os.Getenv("PGSERVICEFILE"))
|
|
require.Equal(t, "local-realm", os.Getenv("PGREALM"))
|
|
}
|
|
|
|
func TestSanitizePostgresEnvironment(t *testing.T) {
|
|
t.Setenv("PGSERVICE", "local-service")
|
|
t.Setenv("PGSERVICEFILE", "/tmp/pg_service.conf")
|
|
t.Setenv("PGREALM", "local-realm")
|
|
|
|
SanitizePostgresEnvironment()
|
|
|
|
for _, key := range unsupportedPQEnvironmentKeys {
|
|
_, exists := os.LookupEnv(key)
|
|
require.False(t, exists, "%s should be removed at application startup", key)
|
|
}
|
|
}
|
|
|
|
func TestLatestVersionAndCheckVersion(t *testing.T) {
|
|
dir := t.TempDir()
|
|
require.NoError(t, os.WriteFile(filepath.Join(dir, "000001_first.up.sql"), []byte("SELECT 1;"), 0o600))
|
|
require.NoError(t, os.WriteFile(filepath.Join(dir, "000012_latest.up.sql"), []byte("SELECT 1;"), 0o600))
|
|
require.NoError(t, os.WriteFile(filepath.Join(dir, "README"), nil, 0o600))
|
|
|
|
latest, err := LatestVersion(dir)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint(12), latest)
|
|
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.Exec("CREATE TABLE schema_migrations (version INTEGER NOT NULL, dirty BOOLEAN NOT NULL)").Error)
|
|
require.NoError(t, db.Exec("INSERT INTO schema_migrations(version, dirty) VALUES (12, false)").Error)
|
|
require.NoError(t, CheckVersion(context.Background(), db, latest))
|
|
require.Error(t, CheckVersion(context.Background(), db, 13))
|
|
require.NoError(t, db.Exec("UPDATE schema_migrations SET dirty = true").Error)
|
|
require.ErrorContains(t, CheckVersion(context.Background(), db, latest), "dirty")
|
|
}
|