105 lines
3.8 KiB
Go
105 lines
3.8 KiB
Go
package ante_test
|
|
|
|
import (
|
|
cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types"
|
|
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/ante"
|
|
"github.com/cosmos/cosmos-sdk/x/bank/testutil"
|
|
)
|
|
|
|
func (suite *AnteTestSuite) TestEnsureMempoolFees() {
|
|
suite.SetupTest(true) // setup
|
|
suite.txBuilder = suite.clientCtx.TxConfig.NewTxBuilder()
|
|
|
|
mfd := ante.NewMempoolFeeDecorator()
|
|
antehandler := sdk.ChainAnteDecorators(mfd)
|
|
|
|
// keys and addresses
|
|
priv1, _, addr1 := testdata.KeyTestPubAddr()
|
|
|
|
// msg and signatures
|
|
msg := testdata.NewTestMsg(addr1)
|
|
feeAmount := testdata.NewTestFeeAmount()
|
|
gasLimit := testdata.NewTestGasLimit()
|
|
suite.Require().NoError(suite.txBuilder.SetMsgs(msg))
|
|
suite.txBuilder.SetFeeAmount(feeAmount)
|
|
suite.txBuilder.SetGasLimit(gasLimit)
|
|
|
|
privs, accNums, accSeqs := []cryptotypes.PrivKey{priv1}, []uint64{0}, []uint64{0}
|
|
tx, err := suite.CreateTestTx(privs, accNums, accSeqs, suite.ctx.ChainID())
|
|
suite.Require().NoError(err)
|
|
|
|
// Set high gas price so standard test fee fails
|
|
atomPrice := sdk.NewDecCoinFromDec("atom", sdk.NewDec(200).Quo(sdk.NewDec(100000)))
|
|
highGasPrice := []sdk.DecCoin{atomPrice}
|
|
suite.ctx = suite.ctx.WithMinGasPrices(highGasPrice)
|
|
|
|
// Set IsCheckTx to true
|
|
suite.ctx = suite.ctx.WithIsCheckTx(true)
|
|
|
|
// antehandler errors with insufficient fees
|
|
_, err = antehandler(suite.ctx, tx, false)
|
|
suite.Require().NotNil(err, "Decorator should have errored on too low fee for local gasPrice")
|
|
|
|
// Set IsCheckTx to false
|
|
suite.ctx = suite.ctx.WithIsCheckTx(false)
|
|
|
|
// antehandler should not error since we do not check minGasPrice in DeliverTx
|
|
_, err = antehandler(suite.ctx, tx, false)
|
|
suite.Require().Nil(err, "MempoolFeeDecorator returned error in DeliverTx")
|
|
|
|
// Set IsCheckTx back to true for testing sufficient mempool fee
|
|
suite.ctx = suite.ctx.WithIsCheckTx(true)
|
|
|
|
atomPrice = sdk.NewDecCoinFromDec("atom", sdk.NewDec(0).Quo(sdk.NewDec(100000)))
|
|
lowGasPrice := []sdk.DecCoin{atomPrice}
|
|
suite.ctx = suite.ctx.WithMinGasPrices(lowGasPrice)
|
|
|
|
_, err = antehandler(suite.ctx, tx, false)
|
|
suite.Require().Nil(err, "Decorator should not have errored on fee higher than local gasPrice")
|
|
}
|
|
|
|
func (suite *AnteTestSuite) TestDeductFees() {
|
|
suite.SetupTest(false) // setup
|
|
suite.txBuilder = suite.clientCtx.TxConfig.NewTxBuilder()
|
|
|
|
// keys and addresses
|
|
priv1, _, addr1 := testdata.KeyTestPubAddr()
|
|
|
|
// msg and signatures
|
|
msg := testdata.NewTestMsg(addr1)
|
|
feeAmount := testdata.NewTestFeeAmount()
|
|
gasLimit := testdata.NewTestGasLimit()
|
|
suite.Require().NoError(suite.txBuilder.SetMsgs(msg))
|
|
suite.txBuilder.SetFeeAmount(feeAmount)
|
|
suite.txBuilder.SetGasLimit(gasLimit)
|
|
|
|
privs, accNums, accSeqs := []cryptotypes.PrivKey{priv1}, []uint64{0}, []uint64{0}
|
|
tx, err := suite.CreateTestTx(privs, accNums, accSeqs, suite.ctx.ChainID())
|
|
suite.Require().NoError(err)
|
|
|
|
// Set account with insufficient funds
|
|
acc := suite.app.AccountKeeper.NewAccountWithAddress(suite.ctx, addr1)
|
|
suite.app.AccountKeeper.SetAccount(suite.ctx, acc)
|
|
coins := sdk.NewCoins(sdk.NewCoin("atom", sdk.NewInt(10)))
|
|
err = testutil.FundAccount(suite.app.BankKeeper, suite.ctx, addr1, coins)
|
|
suite.Require().NoError(err)
|
|
|
|
dfd := ante.NewDeductFeeDecorator(suite.app.AccountKeeper, suite.app.BankKeeper, nil)
|
|
antehandler := sdk.ChainAnteDecorators(dfd)
|
|
|
|
_, err = antehandler(suite.ctx, tx, false)
|
|
|
|
suite.Require().NotNil(err, "Tx did not error when fee payer had insufficient funds")
|
|
|
|
// Set account with sufficient funds
|
|
suite.app.AccountKeeper.SetAccount(suite.ctx, acc)
|
|
err = testutil.FundAccount(suite.app.BankKeeper, suite.ctx, addr1, sdk.NewCoins(sdk.NewCoin("atom", sdk.NewInt(200))))
|
|
suite.Require().NoError(err)
|
|
|
|
_, err = antehandler(suite.ctx, tx, false)
|
|
|
|
suite.Require().Nil(err, "Tx errored after account has been set with sufficient funds")
|
|
}
|