lightning-terminal/session/test_sql.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
}