diff --git a/sqldb/postgres_fixture.go b/sqldb/postgres_fixture.go index 91b95d66e..6cae3e075 100644 --- a/sqldb/postgres_fixture.go +++ b/sqldb/postgres_fixture.go @@ -155,6 +155,10 @@ func NewTestPostgresDB(t testing.TB, fixture *TestPgFixture) *PostgresStore { context.Background(), GetMigrations()), ) + t.Cleanup(func() { + require.NoError(t, store.DB.Close()) + }) + return store } @@ -182,5 +186,9 @@ func NewTestPostgresDBWithVersion(t *testing.T, fixture *TestPgFixture, err = store.ExecuteMigrations(TargetVersion(version)) require.NoError(t, err) + t.Cleanup(func() { + require.NoError(t, store.DB.Close()) + }) + return store }