| 165 | } |
| 166 | |
| 167 | func setupMigrate(t *testing.T, db *sql.DB, name, path string) (source.Driver, *migrate.Migrate) { |
| 168 | t.Helper() |
| 169 | |
| 170 | ctx := context.Background() |
| 171 | |
| 172 | conn, err := db.Conn(ctx) |
| 173 | require.NoError(t, err) |
| 174 | |
| 175 | dbDriver, err := migratepostgres.WithConnection(ctx, conn, &migratepostgres.Config{ |
| 176 | MigrationsTable: "test_migrate_" + name, |
| 177 | }) |
| 178 | require.NoError(t, err) |
| 179 | |
| 180 | dirFS := os.DirFS(path) |
| 181 | d, err := iofs.New(dirFS, ".") |
| 182 | require.NoError(t, err) |
| 183 | t.Cleanup(func() { |
| 184 | d.Close() |
| 185 | }) |
| 186 | |
| 187 | m, err := migrate.NewWithInstance(name, d, "", dbDriver) |
| 188 | require.NoError(t, err) |
| 189 | t.Cleanup(func() { |
| 190 | m.Close() |
| 191 | }) |
| 192 | |
| 193 | return d, m |
| 194 | } |
| 195 | |
| 196 | type tableStats struct { |
| 197 | mu sync.Mutex |