waku-org/go-waku

package legacy_store

import (
    "database/sql"
    "testing"

    "github.com/prometheus/client_golang/prometheus"
    "github.com/stretchr/testify/require"
    "github.com/waku-org/go-waku/waku/persistence"
    "github.com/waku-org/go-waku/waku/persistence/sqlite"
    "github.com/waku-org/go-waku/waku/v2/utils"
)

func MemoryDB(t *testing.T) *persistence.DBStore {
    var db *sql.DB
    db, err := sqlite.NewDB(":memory:", utils.Logger())
    require.NoError(t, err)

    dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
    require.NoError(t, err)

    return dbStore
}