mirror of
https://github.com/lightninglabs/lightning-terminal.git
synced 2026-08-13 12:33:36 +02:00
37 lines
928 B
Go
37 lines
928 B
Go
//go:build test_db_postgres || test_db_sqlite
|
|
|
|
package session
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/lightninglabs/lightning-terminal/accounts"
|
|
"github.com/lightninglabs/lightning-terminal/db/sqlc"
|
|
"github.com/lightningnetwork/lnd/clock"
|
|
"github.com/lightningnetwork/lnd/sqldb/v2"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func NewTestDBWithAccounts(t *testing.T, clock clock.Clock,
|
|
acctStore accounts.Store) Store {
|
|
|
|
accounts, ok := acctStore.(*accounts.SQLStore)
|
|
require.True(t, ok)
|
|
|
|
return createStore(t, accounts.BaseDB, clock)
|
|
}
|
|
|
|
// createStore is a helper function that creates a new SQLStore and ensure that
|
|
// it is closed when during the test cleanup.
|
|
func createStore(t *testing.T, sqlDB *sqldb.BaseDB,
|
|
clock clock.Clock) *SQLStore {
|
|
|
|
queries := sqlc.NewForType(sqlDB, sqlDB.BackendType)
|
|
|
|
store := NewSQLStore(sqlDB, queries, clock)
|
|
t.Cleanup(func() {
|
|
require.NoError(t, store.Close())
|
|
})
|
|
|
|
return store
|
|
}
|