103 lines
3.6 KiB
Go
103 lines
3.6 KiB
Go
package ante_test
|
|
|
|
import (
|
|
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
|
|
|
"github.com/tendermint/tendermint/crypto"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/ante"
|
|
)
|
|
|
|
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 := []crypto.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(true) // 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 := []crypto.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)
|
|
suite.app.BankKeeper.SetBalances(suite.ctx, addr1, sdk.NewCoins(sdk.NewCoin("atom", sdk.NewInt(10))))
|
|
|
|
dfd := ante.NewDeductFeeDecorator(suite.app.AccountKeeper, suite.app.BankKeeper)
|
|
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)
|
|
suite.app.BankKeeper.SetBalances(suite.ctx, addr1, sdk.NewCoins(sdk.NewCoin("atom", sdk.NewInt(200))))
|
|
|
|
_, err = antehandler(suite.ctx, tx, false)
|
|
|
|
suite.Require().Nil(err, "Tx errored after account has been set with sufficient funds")
|
|
}
|