mirror/erigon-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/erigon-pulse.git synced 2025-01-15 23:38:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

tests: one-liner to create tx (#644)

Browse Source
This commit is contained in:
Alex Sharov 2022-09-20 13:18:30 +07:00 committed by GitHub
parent f05cd214bd
commit cd20a6862f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
kv/memdb/memory_database.go
View File

@ -20,10 +20,9 @@ import (
"context" "context"
"testing" "testing"
"github.com/ledgerwatch/log/v3"
"github.com/ledgerwatch/erigon-lib/kv" "github.com/ledgerwatch/erigon-lib/kv"
"github.com/ledgerwatch/erigon-lib/kv/mdbx" "github.com/ledgerwatch/erigon-lib/kv/mdbx"
"github.com/ledgerwatch/log/v3"
) )
func New() kv.RwDB { func New() kv.RwDB {
@ -47,6 +46,26 @@ func NewTestDB(tb testing.TB) kv.RwDB {
return db return db
} }
func BeginRw(tb testing.TB, db kv.RwDB) kv.RwTx {
tb.Helper()
tx, err := db.BeginRw(context.Background())
if err != nil {
tb.Fatal(err)
}
tb.Cleanup(tx.Rollback)
return tx
}
func BeginRo(tb testing.TB, db kv.RoDB) kv.Tx {
tb.Helper()
tx, err := db.BeginRo(context.Background())
if err != nil {
tb.Fatal(err)
}
tb.Cleanup(tx.Rollback)
return tx
}
func NewTestPoolDB(tb testing.TB) kv.RwDB { func NewTestPoolDB(tb testing.TB) kv.RwDB {
tb.Helper() tb.Helper()
db := NewPoolDB() db := NewPoolDB()