package keeper_test import ( "errors" "math/big" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/ethereum/go-ethereum/common" ethtypes "github.com/ethereum/go-ethereum/core/types" "github.com/tharsis/ethermint/x/evm/keeper" "github.com/tharsis/ethermint/x/evm/types" ) // LogRecordHook records all the logs type LogRecordHook struct { Logs []*ethtypes.Log } func (dh *LogRecordHook) PostTxProcessing(ctx sdk.Context, from common.Address, to *common.Address, receipt *ethtypes.Receipt) error { dh.Logs = receipt.Logs return nil } // FailureHook always fail type FailureHook struct{} func (dh FailureHook) PostTxProcessing(ctx sdk.Context, from common.Address, to *common.Address, receipt *ethtypes.Receipt) error { return errors.New("post tx processing failed") } func (suite *KeeperTestSuite) TestEvmHooks() { testCases := []struct { msg string setupHook func() types.EvmHooks expFunc func(hook types.EvmHooks, result error) }{ { "log collect hook", func() types.EvmHooks { return &LogRecordHook{} }, func(hook types.EvmHooks, result error) { suite.Require().NoError(result) suite.Require().Equal(1, len((hook.(*LogRecordHook).Logs))) }, }, { "always fail hook", func() types.EvmHooks { return &FailureHook{} }, func(hook types.EvmHooks, result error) { suite.Require().Error(result) }, }, } for _, tc := range testCases { suite.SetupTest() hook := tc.setupHook() suite.app.EvmKeeper.SetHooks(keeper.NewMultiEvmHooks(hook)) k := suite.app.EvmKeeper txHash := common.BigToHash(big.NewInt(1)) k.SetTxHashTransient(txHash) k.AddLog(ðtypes.Log{ Topics: []common.Hash{}, Address: suite.address, }) logs := k.GetTxLogsTransient(txHash) receipt := ðtypes.Receipt{ TxHash: txHash, Logs: logs, } result := k.PostTxProcessing(common.Address{}, nil, receipt) tc.expFunc(hook, result) } }