Function buildTestMigrationsBundle

rivermigrate/river_migrate_test.go:982–1014

(t *testing.T)

}

func buildTestMigrationsBundle(t *testing.T) *testMigrationsBundle {
	t.Helper()

	// `migration/` subdir is added by migrationsFromFS
	migrationFS := os.DirFS("../riverdriver/riverpgxv5")

	migrations, err := migrationsFromFS(migrationFS, riverdriver.MigrationLineMain)
	require.NoError(t, err)

	// We base our test migrations on the actual line of migrations, so get
	// their maximum version number which we'll use to define test version
	// numbers so that the tests don't break anytime we add a new one.
	migrationsMaxVersion := migrations[len(migrations)-1].Version

	testVersions := []Migration{
		{
			Version: migrationsMaxVersion + 1,
			SQLUp:   "CREATE TABLE /* TEMPLATE: schema */test_table(id bigserial PRIMARY KEY);",
			SQLDown: "DROP TABLE /* TEMPLATE: schema */test_table;",
		},
		{
			Version: migrationsMaxVersion + 2,
			SQLUp:   "ALTER TABLE /* TEMPLATE: schema */test_table ADD COLUMN name varchar(200); CREATE INDEX idx_test_table_name ON /* TEMPLATE: schema */test_table(name);",
			SQLDown: "DROP INDEX /* TEMPLATE: schema */idx_test_table_name; ALTER TABLE /* TEMPLATE: schema */test_table DROP COLUMN name;",
		},
	}

	return &testMigrationsBundle{
		MaxVersion:                 migrationsMaxVersion,
		WithTestVersionsMap:        validateAndInit(append(migrations, testVersions...)),
		WithTestVersionsMaxVersion: migrationsMaxVersion + len(testVersions),
	}
}

// A command returning an error aborts the transaction. This is a shortcut to
// execute a command in a subtransaction so that we can verify an error, but