72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
|
package migrate_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"code.icb4dc0.de/prskr/searcherside/core/ports"
|
||
|
|
"code.icb4dc0.de/prskr/searcherside/infrastructure/db/migrate"
|
||
|
|
_ "code.icb4dc0.de/prskr/searcherside/internal/db"
|
||
|
|
"code.icb4dc0.de/prskr/searcherside/internal/ent"
|
||
|
|
"code.icb4dc0.de/prskr/searcherside/scripts/migrations"
|
||
|
|
|
||
|
|
"entgo.io/ent/dialect"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAtlasMigrator_Migrate_SQLite(t *testing.T) {
|
||
|
|
dbPath := filepath.Join(t.TempDir(), "gophershare.db")
|
||
|
|
|
||
|
|
req := ports.MigrationRequest{
|
||
|
|
Driver: ports.DriverSQLite,
|
||
|
|
URL: fmt.Sprintf("file://%s?_pragma=foreign_keys(1)", dbPath),
|
||
|
|
}
|
||
|
|
|
||
|
|
client, err := ent.Open(dialect.SQLite, req.URL)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to open sqlite3 client: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
migrator := migrate.AtlasMigrator{
|
||
|
|
MigrationsFS: migrations.Fs,
|
||
|
|
RevisionRW: migrate.NewDBRevisionRW(client),
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := migrator.Migrate(context.Background(), req); err != nil {
|
||
|
|
t.Fatalf("failed to migrate: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAtlasMigrator_Migrate_Postgres(t *testing.T) {
|
||
|
|
connString, closer, err := dbtest.TestPostgresDatabase(context.Background())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to setup Postgres test DB: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
t.Cleanup(func() {
|
||
|
|
if err := closer.Close(context.Background()); err != nil {
|
||
|
|
t.Errorf("failed to close Postgres test DB: %v", err)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
req := ports.MigrationRequest{
|
||
|
|
Driver: ports.DriverPostgres,
|
||
|
|
URL: connString,
|
||
|
|
}
|
||
|
|
|
||
|
|
client, err := ent.Open(dialect.Postgres, req.URL)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to open sqlite3 client: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
migrator := migrate.AtlasMigrator{
|
||
|
|
MigrationsFS: migrations.Fs,
|
||
|
|
RevisionRW: migrate.NewDBRevisionRW(client),
|
||
|
|
}
|
||
|
|
|
||
|
|
if err = migrator.Migrate(context.Background(), req); err != nil {
|
||
|
|
t.Fatalf("failed to migrate: %v", err)
|
||
|
|
}
|
||
|
|
}
|