package keeper_test import ( "fmt" "math/big" codectypes "github.com/cosmos/cosmos-sdk/codec/types" cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types" sdk "github.com/cosmos/cosmos-sdk/types" authsigning "github.com/cosmos/cosmos-sdk/x/auth/signing" authtx "github.com/cosmos/cosmos-sdk/x/auth/tx" authtypes "github.com/cosmos/cosmos-sdk/x/auth/types" "github.com/ethereum/go-ethereum/common" ethtypes "github.com/ethereum/go-ethereum/core/types" "github.com/ethereum/go-ethereum/core/vm" "github.com/ethereum/go-ethereum/crypto" "github.com/tharsis/ethermint/tests" "github.com/tharsis/ethermint/x/evm/statedb" "github.com/tharsis/ethermint/x/evm/types" ) func (suite *KeeperTestSuite) TestCreateAccount() { testCases := []struct { name string addr common.Address malleate func(vm.StateDB, common.Address) callback func(vm.StateDB, common.Address) }{ { "reset account (keep balance)", suite.address, func(vmdb vm.StateDB, addr common.Address) { vmdb.AddBalance(addr, big.NewInt(100)) suite.Require().NotZero(vmdb.GetBalance(addr).Int64()) }, func(vmdb vm.StateDB, addr common.Address) { suite.Require().Equal(vmdb.GetBalance(addr).Int64(), int64(100)) }, }, { "create account", tests.GenerateAddress(), func(vmdb vm.StateDB, addr common.Address) { suite.Require().False(vmdb.Exist(addr)) }, func(vmdb vm.StateDB, addr common.Address) { suite.Require().True(vmdb.Exist(addr)) }, }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() tc.malleate(vmdb, tc.addr) vmdb.CreateAccount(tc.addr) tc.callback(vmdb, tc.addr) }) } } func (suite *KeeperTestSuite) TestAddBalance() { testCases := []struct { name string amount *big.Int isNoOp bool }{ { "positive amount", big.NewInt(100), false, }, { "zero amount", big.NewInt(0), true, }, { "negative amount", big.NewInt(-1), false, // seems to be consistent with go-ethereum's implementation }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() prev := vmdb.GetBalance(suite.address) vmdb.AddBalance(suite.address, tc.amount) post := vmdb.GetBalance(suite.address) if tc.isNoOp { suite.Require().Equal(prev.Int64(), post.Int64()) } else { suite.Require().Equal(new(big.Int).Add(prev, tc.amount).Int64(), post.Int64()) } }) } } func (suite *KeeperTestSuite) TestSubBalance() { testCases := []struct { name string amount *big.Int malleate func(vm.StateDB) isNoOp bool }{ { "positive amount, below zero", big.NewInt(100), func(vm.StateDB) {}, false, }, { "positive amount, above zero", big.NewInt(50), func(vmdb vm.StateDB) { vmdb.AddBalance(suite.address, big.NewInt(100)) }, false, }, { "zero amount", big.NewInt(0), func(vm.StateDB) {}, true, }, { "negative amount", big.NewInt(-1), func(vm.StateDB) {}, false, }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() tc.malleate(vmdb) prev := vmdb.GetBalance(suite.address) vmdb.SubBalance(suite.address, tc.amount) post := vmdb.GetBalance(suite.address) if tc.isNoOp { suite.Require().Equal(prev.Int64(), post.Int64()) } else { suite.Require().Equal(new(big.Int).Sub(prev, tc.amount).Int64(), post.Int64()) } }) } } func (suite *KeeperTestSuite) TestGetNonce() { testCases := []struct { name string address common.Address expectedNonce uint64 malleate func(vm.StateDB) }{ { "account not found", tests.GenerateAddress(), 0, func(vm.StateDB) {}, }, { "existing account", suite.address, 1, func(vmdb vm.StateDB) { vmdb.SetNonce(suite.address, 1) }, }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() tc.malleate(vmdb) nonce := vmdb.GetNonce(tc.address) suite.Require().Equal(tc.expectedNonce, nonce) }) } } func (suite *KeeperTestSuite) TestSetNonce() { testCases := []struct { name string address common.Address nonce uint64 malleate func() }{ { "new account", tests.GenerateAddress(), 10, func() {}, }, { "existing account", suite.address, 99, func() {}, }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() vmdb.SetNonce(tc.address, tc.nonce) nonce := vmdb.GetNonce(tc.address) suite.Require().Equal(tc.nonce, nonce) }) } } func (suite *KeeperTestSuite) TestGetCodeHash() { addr := tests.GenerateAddress() baseAcc := &authtypes.BaseAccount{Address: sdk.AccAddress(addr.Bytes()).String()} suite.app.AccountKeeper.SetAccount(suite.ctx, baseAcc) testCases := []struct { name string address common.Address expHash common.Hash malleate func(vm.StateDB) }{ { "account not found", tests.GenerateAddress(), common.Hash{}, func(vm.StateDB) {}, }, { "account not EthAccount type, EmptyCodeHash", addr, common.BytesToHash(types.EmptyCodeHash), func(vm.StateDB) {}, }, { "existing account", suite.address, crypto.Keccak256Hash([]byte("codeHash")), func(vmdb vm.StateDB) { vmdb.SetCode(suite.address, []byte("codeHash")) }, }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() tc.malleate(vmdb) hash := vmdb.GetCodeHash(tc.address) suite.Require().Equal(tc.expHash, hash) }) } } func (suite *KeeperTestSuite) TestSetCode() { addr := tests.GenerateAddress() baseAcc := &authtypes.BaseAccount{Address: sdk.AccAddress(addr.Bytes()).String()} suite.app.AccountKeeper.SetAccount(suite.ctx, baseAcc) testCases := []struct { name string address common.Address code []byte isNoOp bool }{ { "account not found", tests.GenerateAddress(), []byte("code"), false, }, { "account not EthAccount type", addr, nil, true, }, { "existing account", suite.address, []byte("code"), false, }, { "existing account, code deleted from store", suite.address, nil, false, }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() prev := vmdb.GetCode(tc.address) vmdb.SetCode(tc.address, tc.code) post := vmdb.GetCode(tc.address) if tc.isNoOp { suite.Require().Equal(prev, post) } else { suite.Require().Equal(tc.code, post) } suite.Require().Equal(len(post), vmdb.GetCodeSize(tc.address)) }) } } func (suite *KeeperTestSuite) TestRefund() { testCases := []struct { name string malleate func(vm.StateDB) expRefund uint64 expPanic bool }{ { "success - add and subtract refund", func(vmdb vm.StateDB) { vmdb.AddRefund(11) }, 1, false, }, { "fail - subtract amount > current refund", func(vm.StateDB) { }, 0, true, }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() tc.malleate(vmdb) if tc.expPanic { suite.Require().Panics(func() { vmdb.SubRefund(10) }) } else { vmdb.SubRefund(10) suite.Require().Equal(tc.expRefund, vmdb.GetRefund()) } }) } } func (suite *KeeperTestSuite) TestState() { testCases := []struct { name string key, value common.Hash }{ { "set state - delete from store", common.BytesToHash([]byte("key")), common.Hash{}, }, { "set state - update value", common.BytesToHash([]byte("key")), common.BytesToHash([]byte("value")), }, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() vmdb.SetState(suite.address, tc.key, tc.value) value := vmdb.GetState(suite.address, tc.key) suite.Require().Equal(tc.value, value) }) } } func (suite *KeeperTestSuite) TestCommittedState() { suite.SetupTest() key := common.BytesToHash([]byte("key")) value1 := common.BytesToHash([]byte("value1")) value2 := common.BytesToHash([]byte("value2")) vmdb := suite.StateDB() vmdb.SetState(suite.address, key, value1) vmdb.Commit() vmdb = suite.StateDB() vmdb.SetState(suite.address, key, value2) tmp := vmdb.GetState(suite.address, key) suite.Require().Equal(value2, tmp) tmp = vmdb.GetCommittedState(suite.address, key) suite.Require().Equal(value1, tmp) vmdb.Commit() vmdb = suite.StateDB() tmp = vmdb.GetCommittedState(suite.address, key) suite.Require().Equal(value2, tmp) } func (suite *KeeperTestSuite) TestSuicide() { code := []byte("code") db := suite.StateDB() // Add code to account db.SetCode(suite.address, code) suite.Require().Equal(code, db.GetCode(suite.address)) // Add state to account for i := 0; i < 5; i++ { db.SetState(suite.address, common.BytesToHash([]byte(fmt.Sprintf("key%d", i))), common.BytesToHash([]byte(fmt.Sprintf("value%d", i)))) } suite.Require().NoError(db.Commit()) db = suite.StateDB() // Call Suicide suite.Require().Equal(true, db.Suicide(suite.address)) // Check suicided is marked suite.Require().Equal(true, db.HasSuicided(suite.address)) suite.Require().NoError(db.Commit()) db = suite.StateDB() // Check code is deleted suite.Require().Nil(db.GetCode(suite.address)) // Check state is deleted var storage types.Storage suite.app.EvmKeeper.ForEachStorage(suite.ctx, suite.address, func(key, value common.Hash) bool { storage = append(storage, types.NewState(key, value)) return true }) suite.Require().Equal(0, len(storage)) // Check account is deleted suite.Require().Equal(common.Hash{}, db.GetCodeHash(suite.address)) } func (suite *KeeperTestSuite) TestExist() { testCases := []struct { name string address common.Address malleate func(vm.StateDB) exists bool }{ {"success, account exists", suite.address, func(vm.StateDB) {}, true}, {"success, has suicided", suite.address, func(vmdb vm.StateDB) { vmdb.Suicide(suite.address) }, true}, {"success, account doesn't exist", tests.GenerateAddress(), func(vm.StateDB) {}, false}, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() tc.malleate(vmdb) suite.Require().Equal(tc.exists, vmdb.Exist(tc.address)) }) } } func (suite *KeeperTestSuite) TestEmpty() { suite.SetupTest() testCases := []struct { name string address common.Address malleate func(vm.StateDB) empty bool }{ {"empty, account exists", suite.address, func(vm.StateDB) {}, true}, { "not empty, positive balance", suite.address, func(vmdb vm.StateDB) { vmdb.AddBalance(suite.address, big.NewInt(100)) }, false, }, {"empty, account doesn't exist", tests.GenerateAddress(), func(vm.StateDB) {}, true}, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() tc.malleate(vmdb) suite.Require().Equal(tc.empty, vmdb.Empty(tc.address)) }) } } func (suite *KeeperTestSuite) TestSnapshot() { key := common.BytesToHash([]byte("key")) value1 := common.BytesToHash([]byte("value1")) value2 := common.BytesToHash([]byte("value2")) testCases := []struct { name string malleate func(vm.StateDB) }{ {"simple revert", func(vmdb vm.StateDB) { revision := vmdb.Snapshot() suite.Require().Zero(revision) vmdb.SetState(suite.address, key, value1) suite.Require().Equal(value1, vmdb.GetState(suite.address, key)) vmdb.RevertToSnapshot(revision) // reverted suite.Require().Equal(common.Hash{}, vmdb.GetState(suite.address, key)) }}, {"nested snapshot/revert", func(vmdb vm.StateDB) { revision1 := vmdb.Snapshot() suite.Require().Zero(revision1) vmdb.SetState(suite.address, key, value1) revision2 := vmdb.Snapshot() vmdb.SetState(suite.address, key, value2) suite.Require().Equal(value2, vmdb.GetState(suite.address, key)) vmdb.RevertToSnapshot(revision2) suite.Require().Equal(value1, vmdb.GetState(suite.address, key)) vmdb.RevertToSnapshot(revision1) suite.Require().Equal(common.Hash{}, vmdb.GetState(suite.address, key)) }}, {"jump revert", func(vmdb vm.StateDB) { revision1 := vmdb.Snapshot() vmdb.SetState(suite.address, key, value1) vmdb.Snapshot() vmdb.SetState(suite.address, key, value2) vmdb.RevertToSnapshot(revision1) suite.Require().Equal(common.Hash{}, vmdb.GetState(suite.address, key)) }}, } for _, tc := range testCases { suite.Run(tc.name, func() { suite.SetupTest() vmdb := suite.StateDB() tc.malleate(vmdb) }) } } func (suite *KeeperTestSuite) CreateTestTx(msg *types.MsgEthereumTx, priv cryptotypes.PrivKey) authsigning.Tx { option, err := codectypes.NewAnyWithValue(&types.ExtensionOptionsEthereumTx{}) suite.Require().NoError(err) txBuilder := suite.clientCtx.TxConfig.NewTxBuilder() builder, ok := txBuilder.(authtx.ExtensionOptionsTxBuilder) suite.Require().True(ok) builder.SetExtensionOptions(option) err = msg.Sign(suite.ethSigner, tests.NewSigner(priv)) suite.Require().NoError(err) err = txBuilder.SetMsgs(msg) suite.Require().NoError(err) return txBuilder.GetTx() } func (suite *KeeperTestSuite) TestAddLog() { addr, privKey := tests.NewAddrKey() msg := types.NewTx(big.NewInt(1), 0, &suite.address, big.NewInt(1), 100000, big.NewInt(1), nil, nil, []byte("test"), nil) msg.From = addr.Hex() tx := suite.CreateTestTx(msg, privKey) msg, _ = tx.GetMsgs()[0].(*types.MsgEthereumTx) txHash := msg.AsTransaction().Hash() msg2 := types.NewTx(big.NewInt(1), 1, &suite.address, big.NewInt(1), 100000, big.NewInt(1), nil, nil, []byte("test"), nil) msg2.From = addr.Hex() tx2 := suite.CreateTestTx(msg2, privKey) msg2, _ = tx2.GetMsgs()[0].(*types.MsgEthereumTx) msg3 := types.NewTx(big.NewInt(1), 0, &suite.address, big.NewInt(1), 100000, nil, big.NewInt(1), big.NewInt(1), []byte("test"), nil) msg3.From = addr.Hex() tx3 := suite.CreateTestTx(msg3, privKey) msg3, _ = tx3.GetMsgs()[0].(*types.MsgEthereumTx) txHash3 := msg3.AsTransaction().Hash() msg4 := types.NewTx(big.NewInt(1), 1, &suite.address, big.NewInt(1), 100000, nil, big.NewInt(1), big.NewInt(1), []byte("test"), nil) msg4.From = addr.Hex() tx4 := suite.CreateTestTx(msg4, privKey) msg4, _ = tx4.GetMsgs()[0].(*types.MsgEthereumTx) testCases := []struct { name string hash common.Hash log, expLog *ethtypes.Log // pre and post populating log fields malleate func(vm.StateDB) }{ { "tx hash from message", txHash, ðtypes.Log{ Address: addr, Topics: make([]common.Hash, 0), }, ðtypes.Log{ Address: addr, TxHash: txHash, Topics: make([]common.Hash, 0), }, func(vm.StateDB) {}, }, { "dynamicfee tx hash from message", txHash3, ðtypes.Log{ Address: addr, Topics: make([]common.Hash, 0), }, ðtypes.Log{ Address: addr, TxHash: txHash3, Topics: make([]common.Hash, 0), }, func(vm.StateDB) {}, }, } for _, tc := range testCases { suite.Run(tc.name, func() { suite.SetupTest() vmdb := statedb.New(suite.ctx, suite.app.EvmKeeper, statedb.NewTxConfig( common.BytesToHash(suite.ctx.HeaderHash().Bytes()), tc.hash, 0, 0, )) tc.malleate(vmdb) vmdb.AddLog(tc.log) logs := vmdb.Logs() suite.Require().Equal(1, len(logs)) suite.Require().Equal(tc.expLog, logs[0]) }) } } func (suite *KeeperTestSuite) TestPrepareAccessList() { dest := tests.GenerateAddress() precompiles := []common.Address{tests.GenerateAddress(), tests.GenerateAddress()} accesses := ethtypes.AccessList{ {Address: tests.GenerateAddress(), StorageKeys: []common.Hash{common.BytesToHash([]byte("key"))}}, {Address: tests.GenerateAddress(), StorageKeys: []common.Hash{common.BytesToHash([]byte("key1"))}}, } vmdb := suite.StateDB() vmdb.PrepareAccessList(suite.address, &dest, precompiles, accesses) suite.Require().True(vmdb.AddressInAccessList(suite.address)) suite.Require().True(vmdb.AddressInAccessList(dest)) for _, precompile := range precompiles { suite.Require().True(vmdb.AddressInAccessList(precompile)) } for _, access := range accesses { for _, key := range access.StorageKeys { addrOK, slotOK := vmdb.SlotInAccessList(access.Address, key) suite.Require().True(addrOK, access.Address.Hex()) suite.Require().True(slotOK, key.Hex()) } } } func (suite *KeeperTestSuite) TestAddAddressToAccessList() { testCases := []struct { name string addr common.Address }{ {"new address", suite.address}, {"existing address", suite.address}, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() vmdb.AddAddressToAccessList(tc.addr) addrOk := vmdb.AddressInAccessList(tc.addr) suite.Require().True(addrOk, tc.addr.Hex()) }) } } func (suite *KeeperTestSuite) AddSlotToAccessList() { testCases := []struct { name string addr common.Address slot common.Hash }{ {"new address and slot (1)", tests.GenerateAddress(), common.BytesToHash([]byte("hash"))}, {"new address and slot (2)", suite.address, common.Hash{}}, {"existing address and slot", suite.address, common.Hash{}}, {"existing address, new slot", suite.address, common.BytesToHash([]byte("hash"))}, } for _, tc := range testCases { suite.Run(tc.name, func() { vmdb := suite.StateDB() vmdb.AddSlotToAccessList(tc.addr, tc.slot) addrOk, slotOk := vmdb.SlotInAccessList(tc.addr, tc.slot) suite.Require().True(addrOk, tc.addr.Hex()) suite.Require().True(slotOk, tc.slot.Hex()) }) } } // FIXME skip for now func (suite *KeeperTestSuite) _TestForEachStorage() { var storage types.Storage testCase := []struct { name string malleate func(vm.StateDB) callback func(key, value common.Hash) (stop bool) expValues []common.Hash }{ { "aggregate state", func(vmdb vm.StateDB) { for i := 0; i < 5; i++ { vmdb.SetState(suite.address, common.BytesToHash([]byte(fmt.Sprintf("key%d", i))), common.BytesToHash([]byte(fmt.Sprintf("value%d", i)))) } }, func(key, value common.Hash) bool { storage = append(storage, types.NewState(key, value)) return true }, []common.Hash{ common.BytesToHash([]byte("value0")), common.BytesToHash([]byte("value1")), common.BytesToHash([]byte("value2")), common.BytesToHash([]byte("value3")), common.BytesToHash([]byte("value4")), }, }, { "filter state", func(vmdb vm.StateDB) { vmdb.SetState(suite.address, common.BytesToHash([]byte("key")), common.BytesToHash([]byte("value"))) vmdb.SetState(suite.address, common.BytesToHash([]byte("filterkey")), common.BytesToHash([]byte("filtervalue"))) }, func(key, value common.Hash) bool { if value == common.BytesToHash([]byte("filtervalue")) { storage = append(storage, types.NewState(key, value)) return false } return true }, []common.Hash{ common.BytesToHash([]byte("filtervalue")), }, }, } for _, tc := range testCase { suite.Run(tc.name, func() { suite.SetupTest() // reset vmdb := suite.StateDB() tc.malleate(vmdb) err := vmdb.ForEachStorage(suite.address, tc.callback) suite.Require().NoError(err) suite.Require().Equal(len(tc.expValues), len(storage), fmt.Sprintf("Expected values:\n%v\nStorage Values\n%v", tc.expValues, storage)) vals := make([]common.Hash, len(storage)) for i := range storage { vals[i] = common.HexToHash(storage[i].Value) } // TODO: not sure why Equals fails suite.Require().ElementsMatch(tc.expValues, vals) }) storage = types.Storage{} } }