highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-11-11 06:37:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Andras Banki-Horvath
2024-06-25 15:37:51 +02:00
parent 5292c76e10
commit ed36598504
4 changed files with 88 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
sqldb/postgres_test.go
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)
}