mirror/erigon-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/erigon-pulse.git synced 2025-01-15 15:28: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"
"testing"
"github.com/ledgerwatch/log/v3"
"github.com/ledgerwatch/erigon-lib/kv"
"github.com/ledgerwatch/erigon-lib/kv/mdbx"
"github.com/ledgerwatch/log/v3"
)
func New() kv.RwDB {
@ -47,6 +46,26 @@ func NewTestDB(tb testing.TB) kv.RwDB {
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 {
tb.Helper()
db := NewPoolDB()