package keeper_test import ( "math/big" "testing" "time" "github.com/stretchr/testify/suite" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/ethermint/app" "github.com/cosmos/ethermint/x/evm/keeper" ethcmn "github.com/ethereum/go-ethereum/common" ethtypes "github.com/ethereum/go-ethereum/core/types" abci "github.com/tendermint/tendermint/abci/types" ) const addrHex = "0x756F45E3FA69347A9A973A725E3C98bC4db0b4c1" var ( address = ethcmn.HexToAddress(addrHex) hash = ethcmn.FromHex("0x0d87a3a5f73140f46aac1bf419263e4e94e87c292f25007700ab7f2060e2af68") ) type KeeperTestSuite struct { suite.Suite ctx sdk.Context querier sdk.Querier app *app.EthermintApp } func (suite *KeeperTestSuite) SetupTest() { checkTx := false suite.app = app.Setup(checkTx) suite.ctx = suite.app.BaseApp.NewContext(checkTx, abci.Header{Height: 1, ChainID: "3", Time: time.Now().UTC()}) suite.querier = keeper.NewQuerier(suite.app.EvmKeeper) } func TestKeeperTestSuite(t *testing.T) { suite.Run(t, new(KeeperTestSuite)) } func (suite *KeeperTestSuite) TestTransactionLogs() { log := ðtypes.Log{ Address: address, Data: []byte("log"), BlockNumber: 10, } expLogs := []*ethtypes.Log{log} err := suite.app.EvmKeeper.SetTransactionLogs(suite.ctx, expLogs, hash) suite.Require().NoError(err) suite.app.EvmKeeper.AddLog(suite.ctx, expLogs[0]) logs, err := suite.app.EvmKeeper.GetTransactionLogs(suite.ctx, hash) suite.Require().NoError(err) suite.Require().Equal(expLogs, logs) logs = suite.app.EvmKeeper.AllLogs(suite.ctx) suite.Require().Equal(expLogs, logs) } func (suite *KeeperTestSuite) TestDBStorage() { // Perform state transitions suite.app.EvmKeeper.CreateAccount(suite.ctx, address) suite.app.EvmKeeper.SetBalance(suite.ctx, address, big.NewInt(5)) suite.app.EvmKeeper.SetNonce(suite.ctx, address, 4) suite.app.EvmKeeper.SetState(suite.ctx, address, ethcmn.HexToHash("0x2"), ethcmn.HexToHash("0x3")) suite.app.EvmKeeper.SetCode(suite.ctx, address, []byte{0x1}) // Test block hash mapping functionality suite.app.EvmKeeper.SetBlockHashMapping(suite.ctx, hash, 7) height, err := suite.app.EvmKeeper.GetBlockHashMapping(suite.ctx, hash) suite.Require().NoError(err) suite.Require().Equal(int64(7), height) suite.app.EvmKeeper.SetBlockHashMapping(suite.ctx, []byte{0x43, 0x32}, 8) // Test block height mapping functionality testBloom := ethtypes.BytesToBloom([]byte{0x1, 0x3}) suite.app.EvmKeeper.SetBlockBloomMapping(suite.ctx, testBloom, 4) // Get those state transitions suite.Require().Equal(suite.app.EvmKeeper.GetBalance(suite.ctx, address).Cmp(big.NewInt(5)), 0) suite.Require().Equal(suite.app.EvmKeeper.GetNonce(suite.ctx, address), uint64(4)) suite.Require().Equal(suite.app.EvmKeeper.GetState(suite.ctx, address, ethcmn.HexToHash("0x2")), ethcmn.HexToHash("0x3")) suite.Require().Equal(suite.app.EvmKeeper.GetCode(suite.ctx, address), []byte{0x1}) height, err = suite.app.EvmKeeper.GetBlockHashMapping(suite.ctx, hash) suite.Require().NoError(err) suite.Require().Equal(height, int64(7)) height, err = suite.app.EvmKeeper.GetBlockHashMapping(suite.ctx, []byte{0x43, 0x32}) suite.Require().NoError(err) suite.Require().Equal(height, int64(8)) bloom, err := suite.app.EvmKeeper.GetBlockBloomMapping(suite.ctx, 4) suite.Require().NoError(err) suite.Require().Equal(bloom, testBloom) // commit stateDB _, err = suite.app.EvmKeeper.Commit(suite.ctx, false) suite.Require().NoError(err, "failed to commit StateDB") // simulate BaseApp EndBlocker commitment suite.app.Commit() }