mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-10 13:01:21 +00:00
91 lines
3.0 KiB
Go
91 lines
3.0 KiB
Go
package stream_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/ledgerwatch/erigon-lib/kv"
|
|
"github.com/ledgerwatch/erigon-lib/kv/memdb"
|
|
"github.com/ledgerwatch/erigon-lib/kv/stream"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMerge(t *testing.T) {
|
|
db := memdb.NewTestDB(t)
|
|
ctx := context.Background()
|
|
t.Run("simple", func(t *testing.T) {
|
|
require := require.New(t)
|
|
tx, _ := db.BeginRw(ctx)
|
|
defer tx.Rollback()
|
|
_ = tx.Put(kv.AccountsHistory, []byte{1}, []byte{1})
|
|
_ = tx.Put(kv.AccountsHistory, []byte{3}, []byte{1})
|
|
_ = tx.Put(kv.AccountsHistory, []byte{4}, []byte{1})
|
|
_ = tx.Put(kv.PlainState, []byte{2}, []byte{9})
|
|
_ = tx.Put(kv.PlainState, []byte{3}, []byte{9})
|
|
it, _ := tx.Range(kv.AccountsHistory, nil, nil)
|
|
it2, _ := tx.Range(kv.PlainState, nil, nil)
|
|
k, err := stream.MergePairs(it, it2).Keys()
|
|
require.NoError(err)
|
|
require.Equal([][]byte{{1}, {2}, {3}, {4}}, k)
|
|
})
|
|
t.Run("simple values", func(t *testing.T) {
|
|
require := require.New(t)
|
|
tx, _ := db.BeginRw(ctx)
|
|
defer tx.Rollback()
|
|
_ = tx.Put(kv.AccountsHistory, []byte{1}, []byte{1})
|
|
_ = tx.Put(kv.AccountsHistory, []byte{3}, []byte{1})
|
|
_ = tx.Put(kv.AccountsHistory, []byte{4}, []byte{1})
|
|
_ = tx.Put(kv.PlainState, []byte{2}, []byte{9})
|
|
_ = tx.Put(kv.PlainState, []byte{3}, []byte{9})
|
|
it, _ := tx.Range(kv.AccountsHistory, nil, nil)
|
|
it2, _ := tx.Range(kv.PlainState, nil, nil)
|
|
v, err := stream.MergePairs(it, it2).Values()
|
|
require.NoError(err)
|
|
require.Equal([][]byte{{1}, {9}, {1}, {1}}, v)
|
|
})
|
|
t.Run("empty 1st", func(t *testing.T) {
|
|
require := require.New(t)
|
|
tx, _ := db.BeginRw(ctx)
|
|
defer tx.Rollback()
|
|
_ = tx.Put(kv.PlainState, []byte{2}, []byte{9})
|
|
_ = tx.Put(kv.PlainState, []byte{3}, []byte{9})
|
|
it, _ := tx.Range(kv.AccountsHistory, nil, nil)
|
|
it2, _ := tx.Range(kv.PlainState, nil, nil)
|
|
k, err := stream.MergePairs(it, it2).Keys()
|
|
require.NoError(err)
|
|
require.Equal([][]byte{{2}, {3}}, k)
|
|
})
|
|
t.Run("empty 2nd", func(t *testing.T) {
|
|
require := require.New(t)
|
|
tx, _ := db.BeginRw(ctx)
|
|
defer tx.Rollback()
|
|
_ = tx.Put(kv.AccountsHistory, []byte{1}, []byte{1})
|
|
_ = tx.Put(kv.AccountsHistory, []byte{3}, []byte{1})
|
|
_ = tx.Put(kv.AccountsHistory, []byte{4}, []byte{1})
|
|
it, _ := tx.Range(kv.AccountsHistory, nil, nil)
|
|
it2, _ := tx.Range(kv.PlainState, nil, nil)
|
|
k, err := stream.MergePairs(it, it2).Keys()
|
|
require.NoError(err)
|
|
require.Equal([][]byte{{1}, {3}, {4}}, k)
|
|
})
|
|
t.Run("empty both", func(t *testing.T) {
|
|
require := require.New(t)
|
|
tx, _ := db.BeginRw(ctx)
|
|
defer tx.Rollback()
|
|
it, _ := tx.Range(kv.AccountsHistory, nil, nil)
|
|
it2, _ := tx.Range(kv.PlainState, nil, nil)
|
|
m := stream.MergePairs(it, it2)
|
|
require.False(m.HasNext())
|
|
})
|
|
t.Run("error handling", func(t *testing.T) {
|
|
require := require.New(t)
|
|
tx, _ := db.BeginRw(ctx)
|
|
defer tx.Rollback()
|
|
it := stream.PairsWithError(10)
|
|
it2 := stream.PairsWithError(12)
|
|
k, err := stream.MergePairs(it, it2).Keys()
|
|
require.Equal("expected error at iteration: 10", err.Error())
|
|
require.Equal(10, len(k))
|
|
})
|
|
}
|