96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
|
package ante_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math/big"
|
||
|
|
|
||
|
|
sdkmath "cosmossdk.io/math"
|
||
|
|
"github.com/cerc-io/laconicd/app/ante"
|
||
|
|
"github.com/cerc-io/laconicd/tests"
|
||
|
|
evmtypes "github.com/cerc-io/laconicd/x/evm/types"
|
||
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||
|
|
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
|
||
|
|
ethtypes "github.com/ethereum/go-ethereum/core/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (suite AnteTestSuite) TestGasWantedDecorator() {
|
||
|
|
suite.enableFeemarket = true
|
||
|
|
suite.SetupTest()
|
||
|
|
dec := ante.NewGasWantedDecorator(suite.app.EvmKeeper, suite.app.FeeMarketKeeper)
|
||
|
|
from, fromPrivKey := tests.NewAddrKey()
|
||
|
|
to := tests.GenerateAddress()
|
||
|
|
|
||
|
|
testCases := []struct {
|
||
|
|
name string
|
||
|
|
expectedGasWanted uint64
|
||
|
|
malleate func() sdk.Tx
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
"Cosmos Tx",
|
||
|
|
TestGasLimit,
|
||
|
|
func() sdk.Tx {
|
||
|
|
denom := evmtypes.DefaultEVMDenom
|
||
|
|
testMsg := banktypes.MsgSend{
|
||
|
|
FromAddress: "evmos1x8fhpj9nmhqk8z9kpgjt95ck2xwyue0ptzkucp",
|
||
|
|
ToAddress: "evmos1dx67l23hz9l0k9hcher8xz04uj7wf3yu26l2yn",
|
||
|
|
Amount: sdk.Coins{sdk.Coin{Amount: sdkmath.NewInt(10), Denom: denom}},
|
||
|
|
}
|
||
|
|
txBuilder := suite.CreateTestCosmosTxBuilder(sdkmath.NewInt(10), "stake", &testMsg)
|
||
|
|
return txBuilder.GetTx()
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"Ethereum Legacy Tx",
|
||
|
|
TestGasLimit,
|
||
|
|
func() sdk.Tx {
|
||
|
|
msg := suite.BuildTestEthTx(from, to, nil, make([]byte, 0), big.NewInt(0), nil, nil, nil)
|
||
|
|
return suite.CreateTestTx(msg, fromPrivKey, 1, false)
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"Ethereum Access List Tx",
|
||
|
|
TestGasLimit,
|
||
|
|
func() sdk.Tx {
|
||
|
|
emptyAccessList := ethtypes.AccessList{}
|
||
|
|
msg := suite.BuildTestEthTx(from, to, nil, make([]byte, 0), big.NewInt(0), nil, nil, &emptyAccessList)
|
||
|
|
return suite.CreateTestTx(msg, fromPrivKey, 1, false)
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"Ethereum Dynamic Fee Tx (EIP1559)",
|
||
|
|
TestGasLimit,
|
||
|
|
func() sdk.Tx {
|
||
|
|
emptyAccessList := ethtypes.AccessList{}
|
||
|
|
msg := suite.BuildTestEthTx(from, to, nil, make([]byte, 0), big.NewInt(0), big.NewInt(100), big.NewInt(50), &emptyAccessList)
|
||
|
|
return suite.CreateTestTx(msg, fromPrivKey, 1, false)
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"EIP712 message",
|
||
|
|
200000,
|
||
|
|
func() sdk.Tx {
|
||
|
|
amount := sdk.NewCoins(sdk.NewCoin(evmtypes.DefaultEVMDenom, sdkmath.NewInt(20)))
|
||
|
|
gas := uint64(200000)
|
||
|
|
acc := suite.app.AccountKeeper.NewAccountWithAddress(suite.ctx, from.Bytes())
|
||
|
|
suite.Require().NoError(acc.SetSequence(1))
|
||
|
|
suite.app.AccountKeeper.SetAccount(suite.ctx, acc)
|
||
|
|
tx := suite.CreateTestEIP712TxBuilderMsgSend(acc.GetAddress(), fromPrivKey, suite.ctx.ChainID(), gas, amount)
|
||
|
|
return tx.GetTx()
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
// cumulative gas wanted from all test transactions in the same block
|
||
|
|
var expectedGasWanted uint64
|
||
|
|
|
||
|
|
for _, tc := range testCases {
|
||
|
|
suite.Run(tc.name, func() {
|
||
|
|
_, err := dec.AnteHandle(suite.ctx, tc.malleate(), false, NextFn)
|
||
|
|
suite.Require().NoError(err)
|
||
|
|
|
||
|
|
gasWanted := suite.app.FeeMarketKeeper.GetTransientGasWanted(suite.ctx)
|
||
|
|
expectedGasWanted += tc.expectedGasWanted
|
||
|
|
suite.Require().Equal(expectedGasWanted, gasWanted)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|