Geth 1.13 (Deneb/Cancun) update #5
@ -18,24 +18,13 @@ package state
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
pgipfsethdb "github.com/cerc-io/ipfs-ethdb/v5/postgres/v0"
|
||||
"github.com/cerc-io/plugeth-statediff/indexer/database/sql/postgres"
|
||||
"github.com/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/core/types"
|
||||
"github.com/ethereum/go-ethereum/crypto"
|
||||
"github.com/ethereum/go-ethereum/ethdb"
|
||||
"github.com/holiman/uint256"
|
||||
|
||||
"github.com/cerc-io/ipld-eth-statedb/internal"
|
||||
)
|
||||
|
||||
var (
|
||||
testCtx = context.Background()
|
||||
testConfig, _ = postgres.TestConfig.WithEnv()
|
||||
teardownStatements = []string{`TRUNCATE ipld.blocks`}
|
||||
)
|
||||
|
||||
type stateEnv struct {
|
||||
@ -44,7 +33,8 @@ type stateEnv struct {
|
||||
}
|
||||
|
||||
func newStateEnv(t *testing.T) *stateEnv {
|
||||
db := newPgIpfsEthdb(t)
|
||||
db, cleanup := newPgIpfsEthdb(t)
|
||||
t.Cleanup(cleanup)
|
||||
sdb, err := New(types.EmptyRootHash, NewDatabase(db), nil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
@ -52,15 +42,6 @@ func newStateEnv(t *testing.T) *stateEnv {
|
||||
return &stateEnv{db: db, state: sdb}
|
||||
}
|
||||
|
||||
func newPgIpfsEthdb(t *testing.T) ethdb.Database {
|
||||
pool, err := postgres.ConnectSQLX(testCtx, testConfig)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
db := pgipfsethdb.NewDatabase(pool, internal.MakeCacheConfig(t))
|
||||
return db
|
||||
}
|
||||
|
||||
func TestNull(t *testing.T) {
|
||||
s := newStateEnv(t)
|
||||
address := common.HexToAddress("0x823140710bf13990e4500136726d8b55")
|
||||
@ -120,7 +101,9 @@ func TestSnapshotEmpty(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestSnapshot2(t *testing.T) {
|
||||
state, _ := New(types.EmptyRootHash, NewDatabase(newPgIpfsEthdb(t)), nil)
|
||||
db, cleanup := newPgIpfsEthdb(t)
|
||||
t.Cleanup(cleanup)
|
||||
state, _ := New(types.EmptyRootHash, NewDatabase(db), nil)
|
||||
|
||||
stateobjaddr0 := common.BytesToAddress([]byte("so0"))
|
||||
stateobjaddr1 := common.BytesToAddress([]byte("so1"))
|
||||
|
167
trie_by_cid/state/testutil.go
Normal file
167
trie_by_cid/state/testutil.go
Normal file
@ -0,0 +1,167 @@
|
||||
package state
|
||||
|
||||
import (
|
||||
"context"
|
||||
"math/big"
|
||||
"math/rand"
|
||||
"testing"
|
||||
|
||||
pgipfsethdb "github.com/cerc-io/ipfs-ethdb/v5/postgres/v0"
|
||||
"github.com/cerc-io/plugeth-statediff/indexer/database/sql/postgres"
|
||||
helpers "github.com/cerc-io/plugeth-statediff/test_helpers"
|
||||
"github.com/cerc-io/plugeth-statediff/test_helpers/chaingen"
|
||||
"github.com/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/core"
|
||||
"github.com/ethereum/go-ethereum/core/rawdb"
|
||||
"github.com/ethereum/go-ethereum/crypto"
|
||||
"github.com/ethereum/go-ethereum/ethdb"
|
||||
"github.com/ethereum/go-ethereum/params"
|
||||
|
||||
"github.com/cerc-io/ipld-eth-statedb/internal"
|
||||
"github.com/cerc-io/ipld-eth-statedb/internal/testdata"
|
||||
)
|
||||
|
||||
const (
|
||||
chainLength = 5
|
||||
startBlock = 1
|
||||
)
|
||||
|
||||
var (
|
||||
testCtx = context.Background()
|
||||
DBConfig, _ = postgres.TestConfig.WithEnv()
|
||||
)
|
||||
|
||||
var (
|
||||
bank, acct1, acct2 common.Address
|
||||
caddr1, caddr2 common.Address
|
||||
|
||||
contractSpec = chaingen.MustParseContract(testdata.TestContractABI, testdata.TestContractCode)
|
||||
BankKey, _ = crypto.HexToECDSA("0000000000000000000000000000000000000000000000000000000000000001")
|
||||
Account1Key, _ = crypto.HexToECDSA("0000000000000000000000000000000000000000000000000000000000000002")
|
||||
Account2Key, _ = crypto.HexToECDSA("0000000000000000000000000000000000000000000000000000000000000003")
|
||||
BankFunds = big.NewInt(params.Ether * 2)
|
||||
)
|
||||
|
||||
func newPgIpfsEthdb(t *testing.T) (ethdb.Database, func()) {
|
||||
pool, err := postgres.ConnectSQLX(testCtx, DBConfig)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
db := pgipfsethdb.NewDatabase(pool, internal.MakeCacheConfig(t))
|
||||
cleanup := func() {
|
||||
err := helpers.ClearDB(pool)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
return db, cleanup
|
||||
}
|
||||
|
||||
// generates and indexes a chain with arbitrary data, returning the final state root
|
||||
func generateAndIndexChain(t *testing.T) common.Hash {
|
||||
testDB := rawdb.NewMemoryDatabase()
|
||||
chainConfig := &*params.TestChainConfig
|
||||
mockTD := big.NewInt(1337)
|
||||
|
||||
// Make the test blockchain and state
|
||||
gen := makeGenContext(chainConfig, testDB)
|
||||
blocks, receipts, chain := gen.MakeChain(chainLength)
|
||||
defer chain.Stop()
|
||||
|
||||
indexer, err := helpers.NewIndexer(context.Background(), chainConfig, gen.Genesis.Hash(), DBConfig)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if err := helpers.IndexChain(indexer, helpers.IndexChainParams{
|
||||
StateCache: chain.StateCache(),
|
||||
Blocks: blocks,
|
||||
Receipts: receipts,
|
||||
TotalDifficulty: mockTD,
|
||||
}); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return blocks[len(blocks)-1].Root()
|
||||
}
|
||||
|
||||
// A GenContext which exactly replicates the chain generator used in existing tests
|
||||
func makeGenContext(chainConfig *params.ChainConfig, db ethdb.Database) *chaingen.GenContext {
|
||||
gen := chaingen.NewGenContext(chainConfig, db)
|
||||
bank = gen.AddOwnedAccount(BankKey)
|
||||
acct1 = gen.AddOwnedAccount(Account1Key)
|
||||
acct2 = gen.AddOwnedAccount(Account2Key)
|
||||
gen.AddContract("Test", contractSpec)
|
||||
|
||||
rng := rand.New(rand.NewSource(0))
|
||||
gen.AddFunction(func(i int, block *core.BlockGen) {
|
||||
if err := genChain(gen, i, block, rng); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
})
|
||||
gen.Genesis = helpers.GenesisBlockForTesting(
|
||||
db, bank, BankFunds, big.NewInt(params.InitialBaseFee), params.MaxGasLimit,
|
||||
)
|
||||
return gen
|
||||
}
|
||||
|
||||
func genChain(gen *chaingen.GenContext, i int, block *core.BlockGen, rng *rand.Rand) error {
|
||||
var err error
|
||||
switch i {
|
||||
case 0: // In block 1, the test bank sends account #1 and #2 some ether.
|
||||
amt := 1e15 + rng.Int63n(1e15)
|
||||
tx1, err := gen.CreateSendTx(bank, acct1, big.NewInt(amt))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
block.AddTx(tx1)
|
||||
tx2, err := gen.CreateSendTx(bank, acct2, big.NewInt(amt))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
block.AddTx(tx2)
|
||||
|
||||
case 1: // Block 2: deploy contracts
|
||||
caddr1, err = gen.DeployContract(acct1, "Test")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
caddr2, err = gen.DeployContract(acct2, "Test")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
case 2: // Block 3: call contract functions from bank
|
||||
block.SetCoinbase(acct1)
|
||||
|
||||
value := rng.Int63n(1000)
|
||||
tx1, err := gen.CreateCallTx(bank, caddr1, "Put", big.NewInt(value))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
block.AddTx(tx1)
|
||||
|
||||
value = rng.Int63n(1000)
|
||||
tx2, err := gen.CreateCallTx(bank, caddr2, "Put", big.NewInt(value))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
block.AddTx(tx2)
|
||||
|
||||
case 3: // Block 4: call contract functions from accounts 1 & 2
|
||||
block.SetCoinbase(acct2)
|
||||
|
||||
value := rng.Int63n(1000)
|
||||
tx1, err := gen.CreateCallTx(acct1, caddr2, "Put", big.NewInt(value))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
block.AddTx(tx1)
|
||||
|
||||
value = rng.Int63n(1000)
|
||||
tx2, err := gen.CreateCallTx(acct2, caddr1, "Put", big.NewInt(value))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
block.AddTx(tx2)
|
||||
}
|
||||
return nil
|
||||
}
|
Loading…
Reference in New Issue
Block a user