sqldb: add helpers to create test DBs migrated up to a select version

This commit is contained in:
Andras Banki-Horvath committed 2024-07-09 08:39:57 +02:00
1 parent 5292c76e10
commit ed36598504
4 files changed
+88 -13

No files matched your search

+17 -1
View File
@@ -9,5 +9,21 @@ import (
// NewTestDB is a helper function that creates a Postgres database for testing.
func NewTestDB(t *testing.T) *PostgresStore {
return NewTestPostgresDB(t)
pgFixture := NewTestPgFixture(t, DefaultPostgresFixtureLifetime)
t.Cleanup(func() {
pgFixture.TearDown(t)
})
return NewTestPostgresDB(t, pgFixture)
}
// NewTestDBWithVersion is a helper function that creates a Postgres database
// for testing and migrates it to the given version.
func NewTestDBWithVersion(t *testing.T, version uint) *PostgresStore {
pgFixture := NewTestPgFixture(t, DefaultPostgresFixtureLifetime)
t.Cleanup(func() {
pgFixture.TearDown(t)
})
return NewTestPostgresDBWithVersion(t, pgFixture, version)
}