forked from cerc-io/plugeth
internal/blocktest: add package for shared test code (#27270)
This commit is contained in:
parent
e1fd3d67e5
commit
aecf3f9579
@ -18,42 +18,18 @@ package rawdb
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"hash"
|
|
||||||
"math/big"
|
"math/big"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/ethereum/go-ethereum/common"
|
"github.com/ethereum/go-ethereum/common"
|
||||||
"github.com/ethereum/go-ethereum/core/types"
|
"github.com/ethereum/go-ethereum/core/types"
|
||||||
"github.com/ethereum/go-ethereum/ethdb"
|
"github.com/ethereum/go-ethereum/ethdb"
|
||||||
|
"github.com/ethereum/go-ethereum/internal/blocktest"
|
||||||
"github.com/ethereum/go-ethereum/params"
|
"github.com/ethereum/go-ethereum/params"
|
||||||
"github.com/ethereum/go-ethereum/rlp"
|
"github.com/ethereum/go-ethereum/rlp"
|
||||||
"golang.org/x/crypto/sha3"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// testHasher is the helper tool for transaction/receipt list hashing.
|
var newHasher = blocktest.NewHasher
|
||||||
// The original hasher is trie, in order to get rid of import cycle,
|
|
||||||
// use the testing hasher instead.
|
|
||||||
type testHasher struct {
|
|
||||||
hasher hash.Hash
|
|
||||||
}
|
|
||||||
|
|
||||||
func newHasher() *testHasher {
|
|
||||||
return &testHasher{hasher: sha3.NewLegacyKeccak256()}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Reset() {
|
|
||||||
h.hasher.Reset()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Update(key, val []byte) error {
|
|
||||||
h.hasher.Write(key)
|
|
||||||
h.hasher.Write(val)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Hash() common.Hash {
|
|
||||||
return common.BytesToHash(h.hasher.Sum(nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
// Tests that positional lookup metadata can be stored and retrieved.
|
// Tests that positional lookup metadata can be stored and retrieved.
|
||||||
func TestLookupStorage(t *testing.T) {
|
func TestLookupStorage(t *testing.T) {
|
||||||
|
@ -18,7 +18,6 @@ package types
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"hash"
|
|
||||||
"math/big"
|
"math/big"
|
||||||
"reflect"
|
"reflect"
|
||||||
"testing"
|
"testing"
|
||||||
@ -26,9 +25,9 @@ import (
|
|||||||
"github.com/ethereum/go-ethereum/common"
|
"github.com/ethereum/go-ethereum/common"
|
||||||
"github.com/ethereum/go-ethereum/common/math"
|
"github.com/ethereum/go-ethereum/common/math"
|
||||||
"github.com/ethereum/go-ethereum/crypto"
|
"github.com/ethereum/go-ethereum/crypto"
|
||||||
|
"github.com/ethereum/go-ethereum/internal/blocktest"
|
||||||
"github.com/ethereum/go-ethereum/params"
|
"github.com/ethereum/go-ethereum/params"
|
||||||
"github.com/ethereum/go-ethereum/rlp"
|
"github.com/ethereum/go-ethereum/rlp"
|
||||||
"golang.org/x/crypto/sha3"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// from bcValidBlockTest.json, "SimpleTx"
|
// from bcValidBlockTest.json, "SimpleTx"
|
||||||
@ -217,31 +216,6 @@ func BenchmarkEncodeBlock(b *testing.B) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// testHasher is the helper tool for transaction/receipt list hashing.
|
|
||||||
// The original hasher is trie, in order to get rid of import cycle,
|
|
||||||
// use the testing hasher instead.
|
|
||||||
type testHasher struct {
|
|
||||||
hasher hash.Hash
|
|
||||||
}
|
|
||||||
|
|
||||||
func newHasher() *testHasher {
|
|
||||||
return &testHasher{hasher: sha3.NewLegacyKeccak256()}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Reset() {
|
|
||||||
h.hasher.Reset()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Update(key, val []byte) error {
|
|
||||||
h.hasher.Write(key)
|
|
||||||
h.hasher.Write(val)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Hash() common.Hash {
|
|
||||||
return common.BytesToHash(h.hasher.Sum(nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
func makeBenchBlock() *Block {
|
func makeBenchBlock() *Block {
|
||||||
var (
|
var (
|
||||||
key, _ = crypto.GenerateKey()
|
key, _ = crypto.GenerateKey()
|
||||||
@ -280,7 +254,7 @@ func makeBenchBlock() *Block {
|
|||||||
Extra: []byte("benchmark uncle"),
|
Extra: []byte("benchmark uncle"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return NewBlock(header, txs, uncles, receipts, newHasher())
|
return NewBlock(header, txs, uncles, receipts, blocktest.NewHasher())
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestRlpDecodeParentHash(t *testing.T) {
|
func TestRlpDecodeParentHash(t *testing.T) {
|
||||||
|
59
internal/blocktest/test_hash.go
Normal file
59
internal/blocktest/test_hash.go
Normal file
@ -0,0 +1,59 @@
|
|||||||
|
// Copyright 2023 The go-ethereum Authors
|
||||||
|
// This file is part of the go-ethereum library.
|
||||||
|
//
|
||||||
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||||
|
// it under the terms of the GNU Lesser General Public License as published by
|
||||||
|
// the Free Software Foundation, either version 3 of the License, or
|
||||||
|
// (at your option) any later version.
|
||||||
|
//
|
||||||
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
||||||
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
// GNU Lesser General Public License for more details.
|
||||||
|
//
|
||||||
|
// You should have received a copy of the GNU Lesser General Public License
|
||||||
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
// Package utesting provides a standalone replacement for package testing.
|
||||||
|
//
|
||||||
|
// This package exists because package testing cannot easily be embedded into a
|
||||||
|
// standalone go program. It provides an API that mirrors the standard library
|
||||||
|
// testing API.
|
||||||
|
|
||||||
|
package blocktest
|
||||||
|
|
||||||
|
import (
|
||||||
|
"hash"
|
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum/common"
|
||||||
|
"golang.org/x/crypto/sha3"
|
||||||
|
)
|
||||||
|
|
||||||
|
// testHasher is the helper tool for transaction/receipt list hashing.
|
||||||
|
// The original hasher is trie, in order to get rid of import cycle,
|
||||||
|
// use the testing hasher instead.
|
||||||
|
type testHasher struct {
|
||||||
|
hasher hash.Hash
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewHasher returns a new testHasher instance.
|
||||||
|
func NewHasher() *testHasher {
|
||||||
|
return &testHasher{hasher: sha3.NewLegacyKeccak256()}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reset resets the hash state.
|
||||||
|
func (h *testHasher) Reset() {
|
||||||
|
h.hasher.Reset()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update updates the hash state with the given key and value.
|
||||||
|
func (h *testHasher) Update(key, val []byte) error {
|
||||||
|
h.hasher.Write(key)
|
||||||
|
h.hasher.Write(val)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Hash returns the hash value.
|
||||||
|
func (h *testHasher) Hash() common.Hash {
|
||||||
|
return common.BytesToHash(h.hasher.Sum(nil))
|
||||||
|
}
|
@ -21,7 +21,6 @@ import (
|
|||||||
"crypto/ecdsa"
|
"crypto/ecdsa"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"hash"
|
|
||||||
"math/big"
|
"math/big"
|
||||||
"reflect"
|
"reflect"
|
||||||
"testing"
|
"testing"
|
||||||
@ -42,10 +41,10 @@ import (
|
|||||||
"github.com/ethereum/go-ethereum/crypto"
|
"github.com/ethereum/go-ethereum/crypto"
|
||||||
"github.com/ethereum/go-ethereum/ethdb"
|
"github.com/ethereum/go-ethereum/ethdb"
|
||||||
"github.com/ethereum/go-ethereum/event"
|
"github.com/ethereum/go-ethereum/event"
|
||||||
|
"github.com/ethereum/go-ethereum/internal/blocktest"
|
||||||
"github.com/ethereum/go-ethereum/params"
|
"github.com/ethereum/go-ethereum/params"
|
||||||
"github.com/ethereum/go-ethereum/rpc"
|
"github.com/ethereum/go-ethereum/rpc"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
"golang.org/x/crypto/sha3"
|
|
||||||
"golang.org/x/exp/slices"
|
"golang.org/x/exp/slices"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -668,31 +667,6 @@ func hex2Bytes(str string) *hexutil.Bytes {
|
|||||||
return &rpcBytes
|
return &rpcBytes
|
||||||
}
|
}
|
||||||
|
|
||||||
// testHasher is the helper tool for transaction/receipt list hashing.
|
|
||||||
// The original hasher is trie, in order to get rid of import cycle,
|
|
||||||
// use the testing hasher instead.
|
|
||||||
type testHasher struct {
|
|
||||||
hasher hash.Hash
|
|
||||||
}
|
|
||||||
|
|
||||||
func newHasher() *testHasher {
|
|
||||||
return &testHasher{hasher: sha3.NewLegacyKeccak256()}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Reset() {
|
|
||||||
h.hasher.Reset()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Update(key, val []byte) error {
|
|
||||||
h.hasher.Write(key)
|
|
||||||
h.hasher.Write(val)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *testHasher) Hash() common.Hash {
|
|
||||||
return common.BytesToHash(h.hasher.Sum(nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRPCMarshalBlock(t *testing.T) {
|
func TestRPCMarshalBlock(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
var (
|
var (
|
||||||
@ -723,7 +697,7 @@ func TestRPCMarshalBlock(t *testing.T) {
|
|||||||
}
|
}
|
||||||
txs = append(txs, tx)
|
txs = append(txs, tx)
|
||||||
}
|
}
|
||||||
block := types.NewBlock(&types.Header{Number: big.NewInt(100)}, txs, nil, nil, newHasher())
|
block := types.NewBlock(&types.Header{Number: big.NewInt(100)}, txs, nil, nil, blocktest.NewHasher())
|
||||||
|
|
||||||
var testSuite = []struct {
|
var testSuite = []struct {
|
||||||
inclTx bool
|
inclTx bool
|
||||||
@ -795,7 +769,7 @@ func TestRPCGetBlockOrHeader(t *testing.T) {
|
|||||||
Address: common.Address{0x12, 0x34},
|
Address: common.Address{0x12, 0x34},
|
||||||
Amount: 10,
|
Amount: 10,
|
||||||
}
|
}
|
||||||
pending = types.NewBlockWithWithdrawals(&types.Header{Number: big.NewInt(11), Time: 42}, []*types.Transaction{tx}, nil, nil, []*types.Withdrawal{withdrawal}, newHasher())
|
pending = types.NewBlockWithWithdrawals(&types.Header{Number: big.NewInt(11), Time: 42}, []*types.Transaction{tx}, nil, nil, []*types.Withdrawal{withdrawal}, blocktest.NewHasher())
|
||||||
)
|
)
|
||||||
backend := newTestBackend(t, genBlocks, genesis, func(i int, b *core.BlockGen) {
|
backend := newTestBackend(t, genBlocks, genesis, func(i int, b *core.BlockGen) {
|
||||||
// Transfer from account[0] to account[1]
|
// Transfer from account[0] to account[1]
|
||||||
|
Loading…
Reference in New Issue
Block a user