package ante_test import ( "math/big" "testing" "time" "github.com/stretchr/testify/suite" cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types" "github.com/cosmos/cosmos-sdk/simapp/params" "github.com/cosmos/cosmos-sdk/testutil/testdata" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/x/auth/legacy/legacytx" authtypes "github.com/cosmos/cosmos-sdk/x/auth/types" "github.com/cosmos/ethermint/app" ante "github.com/cosmos/ethermint/app/ante" "github.com/cosmos/ethermint/crypto/ethsecp256k1" ethermint "github.com/cosmos/ethermint/types" evmtypes "github.com/cosmos/ethermint/x/evm/types" ethcrypto "github.com/ethereum/go-ethereum/crypto" tmproto "github.com/tendermint/tendermint/proto/tendermint/types" ) type AnteTestSuite struct { suite.Suite ctx sdk.Context app *app.EthermintApp encodingConfig params.EncodingConfig anteHandler sdk.AnteHandler chainID *big.Int } func (suite *AnteTestSuite) SetupTest() { checkTx := false suite.app = app.Setup(checkTx) suite.ctx = suite.app.BaseApp.NewContext(checkTx, tmproto.Header{Height: 2, ChainID: "ethermint-888", Time: time.Now().UTC()}) suite.app.AccountKeeper.SetParams(suite.ctx, authtypes.DefaultParams()) suite.app.EvmKeeper.SetParams(suite.ctx, evmtypes.DefaultParams()) suite.encodingConfig = app.MakeEncodingConfig() // We're using TestMsg amino encoding in some tests, so register it here. suite.encodingConfig.Amino.RegisterConcrete(&testdata.TestMsg{}, "testdata.TestMsg", nil) suite.anteHandler = ante.NewAnteHandler(suite.app.AccountKeeper, suite.app.BankKeeper, suite.app.EvmKeeper, suite.encodingConfig.TxConfig.SignModeHandler()) suite.chainID = big.NewInt(888) } func TestAnteTestSuite(t *testing.T) { suite.Run(t, new(AnteTestSuite)) } func newTestMsg(addrs ...sdk.AccAddress) *testdata.TestMsg { return testdata.NewTestMsg(addrs...) } func newTestCoins() sdk.Coins { return sdk.NewCoins(ethermint.NewPhotonCoinInt64(500000000)) } func newTestStdFee() legacytx.StdFee { return legacytx.NewStdFee(220000, sdk.NewCoins(ethermint.NewPhotonCoinInt64(150))) } // GenerateAddress generates an Ethereum address. func newTestAddrKey() (sdk.AccAddress, cryptotypes.PrivKey) { privkey, _ := ethsecp256k1.GenerateKey() addr := ethcrypto.PubkeyToAddress(privkey.ToECDSA().PublicKey) return sdk.AccAddress(addr.Bytes()), privkey } func newTestSDKTx( ctx sdk.Context, msgs []sdk.Msg, privs []cryptotypes.PrivKey, accNums []uint64, seqs []uint64, fee legacytx.StdFee, ) sdk.Tx { sigs := make([]legacytx.StdSignature, len(privs)) for i, priv := range privs { signBytes := legacytx.StdSignBytes(ctx.ChainID(), accNums[i], seqs[i], 0, fee, msgs, "") sig, err := priv.Sign(signBytes) if err != nil { panic(err) } sigs[i] = legacytx.StdSignature{ PubKey: priv.PubKey(), Signature: sig, } } return legacytx.NewStdTx(msgs, fee, sigs, "") } func (suite *AnteTestSuite) newTestEthTx(msg *evmtypes.MsgEthereumTx, priv cryptotypes.PrivKey) (sdk.Tx, error) { privkey := ðsecp256k1.PrivKey{Key: priv.Bytes()} if err := msg.Sign(suite.chainID, privkey.ToECDSA()); err != nil { return nil, err } return msg, nil }