Function TestMigrator

rivermigrate/river_migrate_test.go:62–930

(t *testing.T)

}

func TestMigrator(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	migrationsBundle := buildTestMigrationsBundle(t)

	type testBundle struct {
		dbPool *pgxpool.Pool
		driver *driverWithAlternateLine
		logger *slog.Logger
		schema string
	}

	setup := func(t *testing.T) (*Migrator[pgx.Tx], *testBundle) {
		t.Helper()

		// Not all migrations can be executed together in a single transaction.
		// Examples include `CREATE INDEX CONCURRENTLY`, or adding an enum value
		// that's used by a later migration. As such, the migrator and its tests
		// must use a full database with commits between each migration.
		//
		// To make this easier to clean up afterward, we create a new, clean schema
		// for each test run and then drop it afterward.
		dbPool := riversharedtest.DBPool(ctx, t)
		schema := "river_migrate_test_" + randutil.Hex(8)
		_, err := dbPool.Exec(ctx, "CREATE SCHEMA "+schema)
		require.NoError(t, err)

		t.Cleanup(func() {
			_, err := dbPool.Exec(ctx, fmt.Sprintf("DROP SCHEMA %s CASCADE", schema))
			require.NoError(t, err)
		})

		bundle := &testBundle{
			dbPool: dbPool,
			driver: &driverWithAlternateLine{Driver: riverpgxv5.New(dbPool)},
			logger: riversharedtest.Logger(t),
			schema: schema,
		}

		migrator, err := New(bundle.driver, &Config{
			Logger: bundle.logger,
			Schema: schema,
		})
		require.NoError(t, err)
		migrator.migrations = migrationsBundle.WithTestVersionsMap

		return migrator, bundle
	}

	t.Run("NewUnknownLine", func(t *testing.T) {
		t.Parallel()

		_, bundle := setup(t)

		_, err := New(bundle.driver, &Config{Line: "unknown_line"})
		require.EqualError(t, err, "migration line does not exist: unknown_line")