101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
package chaingen
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
"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/ethdb"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
|
|
"github.com/cerc-io/ipld-eth-db-validator/v5/internal/testdata"
|
|
)
|
|
|
|
var (
|
|
bank, acct1, acct2 common.Address
|
|
contractAddr common.Address
|
|
contractDataRoot string
|
|
defaultContract *chaingen.ContractSpec
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
defaultContract, err = chaingen.ParseContract(testdata.TestContractABI, testdata.TestContractCode)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// A GenContext which exactly replicates the chain generator used in existing tests
|
|
func DefaultGenContext(chainConfig *params.ChainConfig, db ethdb.Database) *chaingen.GenContext {
|
|
gen := chaingen.NewGenContext(chainConfig, db)
|
|
bank = gen.AddOwnedAccount(test_helpers.TestBankKey)
|
|
acct1 = gen.AddOwnedAccount(test_helpers.Account1Key)
|
|
acct2 = gen.AddOwnedAccount(test_helpers.Account2Key)
|
|
gen.AddContract("Test", defaultContract)
|
|
|
|
gen.AddFunction(func(i int, block *core.BlockGen) {
|
|
if err := defaultChainGen(gen, i, block); err != nil {
|
|
panic(err)
|
|
}
|
|
})
|
|
gen.Genesis = test_helpers.GenesisBlockForTesting(
|
|
db, bank, test_helpers.TestBankFunds, big.NewInt(params.InitialBaseFee), params.MaxGasLimit,
|
|
)
|
|
return gen
|
|
}
|
|
|
|
func defaultChainGen(gen *chaingen.GenContext, i int, block *core.BlockGen) error {
|
|
switch i {
|
|
case 0:
|
|
// In block 1, the test bank sends account #1 some ether.
|
|
tx, err := gen.CreateSendTx(bank, acct1, big.NewInt(10000))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
block.AddTx(tx)
|
|
case 1:
|
|
// In block 2, the test bank sends some more ether to account #1.
|
|
// acct1 passes it on to account #2.
|
|
// acct1 creates a test contract.
|
|
tx1, err := gen.CreateSendTx(bank, acct1, big.NewInt(1000))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
block.AddTx(tx1)
|
|
tx2, err := gen.CreateSendTx(acct1, acct2, big.NewInt(1000))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
block.AddTx(tx2)
|
|
contractAddr, err = gen.DeployContract(acct1, "Test")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
case 2:
|
|
block.SetCoinbase(acct2)
|
|
tx, err := gen.CreateCallTx(bank, contractAddr, "Put", big.NewInt(3))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
block.AddTx(tx)
|
|
case 3:
|
|
block.SetCoinbase(acct2)
|
|
tx, err := gen.CreateCallTx(bank, contractAddr, "Put", big.NewInt(9))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
block.AddTx(tx)
|
|
case 4:
|
|
block.SetCoinbase(acct1)
|
|
tx, err := gen.CreateCallTx(bank, contractAddr, "Put", big.NewInt(0))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
block.AddTx(tx)
|
|
}
|
|
return nil
|
|
}
|