erigon-pulse/migrations/dupsort_state_test.go

118 lines
3.5 KiB
Go
Raw Normal View History

2020-08-12 02:57:55 +00:00
package migrations
import (
"context"
"fmt"
"testing"
"github.com/ledgerwatch/turbo-geth/common"
"github.com/ledgerwatch/turbo-geth/common/dbutils"
"github.com/ledgerwatch/turbo-geth/ethdb"
"github.com/stretchr/testify/require"
)
func TestDupSortHashState(t *testing.T) {
2020-08-12 02:57:55 +00:00
require, db := require.New(t), ethdb.NewMemDatabase()
err := db.KV().Update(context.Background(), func(tx ethdb.Tx) error {
return tx.(ethdb.BucketMigrator).CreateBucket(dbutils.HashedStorageBucket)
2020-08-12 02:57:55 +00:00
})
require.NoError(err)
accKey := string(common.FromHex(fmt.Sprintf("%064x", 0)))
inc := string(common.FromHex("0000000000000001"))
storageKey := accKey + inc + accKey
err = db.Put(dbutils.HashedStorageBucket, []byte(storageKey), []byte{2})
2020-08-12 02:57:55 +00:00
require.NoError(err)
migrator := NewMigrator()
migrator.Migrations = []Migration{dupSortHashState}
2020-08-12 02:57:55 +00:00
err = migrator.Apply(db, "")
require.NoError(err)
// test high-level data access didn't change
i := 0
err = db.Walk(dbutils.HashedStorageBucket, nil, 0, func(k, v []byte) (bool, error) {
2020-08-12 02:57:55 +00:00
i++
return true, nil
})
require.NoError(err)
require.Equal(1, i)
2020-08-12 02:57:55 +00:00
v, err := db.Get(dbutils.HashedStorageBucket, []byte(storageKey))
2020-08-12 02:57:55 +00:00
require.NoError(err)
require.Equal([]byte{2}, v)
tx, err := db.Begin(context.Background(), ethdb.RW)
2020-08-12 02:57:55 +00:00
require.NoError(err)
2020-09-08 19:39:43 +00:00
defer tx.Rollback()
c := tx.(ethdb.HasTx).Tx().CursorDupSort(dbutils.HashedStorageBucket)
2020-09-08 19:39:43 +00:00
// test low-level data layout
2020-08-12 02:57:55 +00:00
require.NoError(err)
keyLen := common.HashLength + common.IncarnationLength
2020-09-08 19:39:43 +00:00
k, v, err := c.SeekBothRange([]byte(storageKey)[:keyLen], []byte(storageKey)[keyLen:])
2020-08-12 02:57:55 +00:00
require.NoError(err)
require.Equal([]byte(storageKey)[:keyLen], k)
require.Equal([]byte(storageKey)[keyLen:], v[:common.HashLength])
require.Equal([]byte{2}, v[common.HashLength:])
}
func TestDupSortPlainState(t *testing.T) {
require, db := require.New(t), ethdb.NewMemDatabase()
err := db.KV().Update(context.Background(), func(tx ethdb.Tx) error {
return tx.(ethdb.BucketMigrator).CreateBucket(dbutils.PlainStateBucketOld1)
})
require.NoError(err)
accKey := string(common.FromHex(fmt.Sprintf("%040x", 0)))
inc := string(common.FromHex("0000000000000001"))
storageKey := accKey + inc + string(common.FromHex(fmt.Sprintf("%064x", 0)))
err = db.Put(dbutils.PlainStateBucketOld1, []byte(accKey), []byte{1})
require.NoError(err)
err = db.Put(dbutils.PlainStateBucketOld1, []byte(storageKey), []byte{2})
require.NoError(err)
migrator := NewMigrator()
migrator.Migrations = []Migration{dupSortPlainState}
err = migrator.Apply(db, "")
require.NoError(err)
// test high-level data access didn't change
i := 0
err = db.Walk(dbutils.PlainStateBucket, nil, 0, func(k, v []byte) (bool, error) {
i++
return true, nil
})
require.NoError(err)
require.Equal(2, i)
v, err := db.Get(dbutils.PlainStateBucket, []byte(accKey))
require.NoError(err)
require.Equal([]byte{1}, v)
v, err = db.Get(dbutils.PlainStateBucket, []byte(storageKey))
require.NoError(err)
require.Equal([]byte{2}, v)
tx, err := db.Begin(context.Background(), ethdb.RW)
require.NoError(err)
2020-09-08 19:39:43 +00:00
defer tx.Rollback()
2020-09-08 19:39:43 +00:00
c := tx.(ethdb.HasTx).Tx().CursorDupSort(dbutils.PlainStateBucket)
ChangeSets dupsort (#1342) * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * change_set_dup * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * working version * aa * aa * aa * aa * aa * aa * aa * aa * aa * aa * aa * aa * aa * aa * aa * squash * squash * fix * fix * fix * fix * fix * fix * fix * fix * fix * fix * fix * history_early_stop * history_early_stop * vmConfig with ReadOnly false * auto_increment * auto_increment * rebase master Co-authored-by: Alexey Akhunov <akhounov@gmail.com>
2020-11-16 12:08:28 +00:00
_, v, err = c.SeekExact([]byte(accKey))
require.NoError(err)
require.Equal([]byte{1}, v)
keyLen := common.AddressLength + common.IncarnationLength
2020-09-08 19:39:43 +00:00
k, v, err := c.SeekBothRange([]byte(storageKey)[:keyLen], []byte(storageKey)[keyLen:])
require.NoError(err)
require.Equal([]byte(storageKey)[:keyLen], k)
require.Equal([]byte(storageKey)[keyLen:], v[:common.HashLength])
require.Equal([]byte{2}, v[common.HashLength:])
}