package keeper_test import ( "math/big" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/ethereum/go-ethereum/common" ethtypes "github.com/ethereum/go-ethereum/core/types" ethparams "github.com/ethereum/go-ethereum/params" evmkeeper "github.com/tharsis/ethermint/x/evm/keeper" evmtypes "github.com/tharsis/ethermint/x/evm/types" ) func (suite *KeeperTestSuite) TestCheckSenderBalance() { hundredInt := sdk.NewInt(100) zeroInt := sdk.ZeroInt() oneInt := sdk.OneInt() fiveInt := sdk.NewInt(5) fiftyInt := sdk.NewInt(50) negInt := sdk.NewInt(-10) testCases := []struct { name string to string gasLimit uint64 gasPrice *sdk.Int gasFeeCap *big.Int gasTipCap *big.Int cost *sdk.Int from string accessList *ethtypes.AccessList expectPass bool enableFeemarket bool }{ { name: "Enough balance", to: suite.address.String(), gasLimit: 10, gasPrice: &oneInt, cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: true, }, { name: "Equal balance", to: suite.address.String(), gasLimit: 99, gasPrice: &oneInt, cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: true, }, { name: "negative cost", to: suite.address.String(), gasLimit: 1, gasPrice: &oneInt, cost: &negInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: false, }, { name: "Higher gas limit, not enough balance", to: suite.address.String(), gasLimit: 100, gasPrice: &oneInt, cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: false, }, { name: "Higher gas price, enough balance", to: suite.address.String(), gasLimit: 10, gasPrice: &fiveInt, cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: true, }, { name: "Higher gas price, not enough balance", to: suite.address.String(), gasLimit: 20, gasPrice: &fiveInt, cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: false, }, { name: "Higher cost, enough balance", to: suite.address.String(), gasLimit: 10, gasPrice: &fiveInt, cost: &fiftyInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: true, }, { name: "Higher cost, not enough balance", to: suite.address.String(), gasLimit: 10, gasPrice: &fiveInt, cost: &hundredInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: false, }, { name: "Enough balance w/ enableFeemarket", to: suite.address.String(), gasLimit: 10, gasFeeCap: big.NewInt(1), cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: true, enableFeemarket: true, }, { name: "Equal balance w/ enableFeemarket", to: suite.address.String(), gasLimit: 99, gasFeeCap: big.NewInt(1), cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: true, enableFeemarket: true, }, { name: "negative cost w/ enableFeemarket", to: suite.address.String(), gasLimit: 1, gasFeeCap: big.NewInt(1), cost: &negInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: false, enableFeemarket: true, }, { name: "Higher gas limit, not enough balance w/ enableFeemarket", to: suite.address.String(), gasLimit: 100, gasFeeCap: big.NewInt(1), cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: false, enableFeemarket: true, }, { name: "Higher gas price, enough balance w/ enableFeemarket", to: suite.address.String(), gasLimit: 10, gasFeeCap: big.NewInt(5), cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: true, enableFeemarket: true, }, { name: "Higher gas price, not enough balance w/ enableFeemarket", to: suite.address.String(), gasLimit: 20, gasFeeCap: big.NewInt(5), cost: &oneInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: false, enableFeemarket: true, }, { name: "Higher cost, enough balance w/ enableFeemarket", to: suite.address.String(), gasLimit: 10, gasFeeCap: big.NewInt(5), cost: &fiftyInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: true, enableFeemarket: true, }, { name: "Higher cost, not enough balance w/ enableFeemarket", to: suite.address.String(), gasLimit: 10, gasFeeCap: big.NewInt(5), cost: &hundredInt, from: suite.address.String(), accessList: ðtypes.AccessList{}, expectPass: false, enableFeemarket: true, }, } vmdb := suite.StateDB() vmdb.AddBalance(suite.address, hundredInt.BigInt()) balance := vmdb.GetBalance(suite.address) suite.Require().Equal(balance, hundredInt.BigInt()) vmdb.Commit() for i, tc := range testCases { suite.Run(tc.name, func() { to := common.HexToAddress(tc.from) var amount, gasPrice, gasFeeCap, gasTipCap *big.Int if tc.cost != nil { amount = tc.cost.BigInt() } if tc.enableFeemarket { gasFeeCap = tc.gasFeeCap if tc.gasTipCap == nil { gasTipCap = oneInt.BigInt() } else { gasTipCap = tc.gasTipCap } } else { if tc.gasPrice != nil { gasPrice = tc.gasPrice.BigInt() } } tx := evmtypes.NewTx(zeroInt.BigInt(), 1, &to, amount, tc.gasLimit, gasPrice, gasFeeCap, gasTipCap, nil, tc.accessList) tx.From = tc.from txData, _ := evmtypes.UnpackTxData(tx.Data) acct := suite.app.EvmKeeper.GetAccountOrEmpty(suite.ctx, suite.address) err := evmkeeper.CheckSenderBalance( sdk.NewIntFromBigInt(acct.Balance), txData, ) if tc.expectPass { suite.Require().NoError(err, "valid test %d failed", i) } else { suite.Require().Error(err, "invalid test %d passed", i) } }) } } func (suite *KeeperTestSuite) TestDeductTxCostsFromUserBalance() { hundredInt := sdk.NewInt(100) zeroInt := sdk.ZeroInt() oneInt := sdk.NewInt(1) fiveInt := sdk.NewInt(5) fiftyInt := sdk.NewInt(50) // should be enough to cover all test cases initBalance := sdk.NewInt((ethparams.InitialBaseFee + 10) * 105) testCases := []struct { name string gasLimit uint64 gasPrice *sdk.Int gasFeeCap *big.Int gasTipCap *big.Int cost *sdk.Int accessList *ethtypes.AccessList expectPass bool enableFeemarket bool }{ { name: "Enough balance", gasLimit: 10, gasPrice: &oneInt, cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: true, }, { name: "Equal balance", gasLimit: 100, gasPrice: &oneInt, cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: true, }, { name: "Higher gas limit, not enough balance", gasLimit: 105, gasPrice: &oneInt, cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: false, }, { name: "Higher gas price, enough balance", gasLimit: 20, gasPrice: &fiveInt, cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: true, }, { name: "Higher gas price, not enough balance", gasLimit: 20, gasPrice: &fiftyInt, cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: false, }, // This case is expected to be true because the fees can be deducted, but the tx // execution is going to fail because there is no more balance to pay the cost { name: "Higher cost, enough balance", gasLimit: 100, gasPrice: &oneInt, cost: &fiftyInt, accessList: ðtypes.AccessList{}, expectPass: true, }, // testcases with enableFeemarket enabled. { name: "Invalid gasFeeCap w/ enableFeemarket", gasLimit: 10, gasFeeCap: big.NewInt(1), gasTipCap: big.NewInt(1), cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: false, enableFeemarket: true, }, { name: "empty tip fee is valid to deduct", gasLimit: 10, gasFeeCap: big.NewInt(ethparams.InitialBaseFee), gasTipCap: big.NewInt(1), cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: true, enableFeemarket: true, }, { name: "effectiveTip equal to gasTipCap", gasLimit: 100, gasFeeCap: big.NewInt(ethparams.InitialBaseFee + 2), cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: true, enableFeemarket: true, }, { name: "effectiveTip equal to (gasFeeCap - baseFee)", gasLimit: 105, gasFeeCap: big.NewInt(ethparams.InitialBaseFee + 1), gasTipCap: big.NewInt(2), cost: &oneInt, accessList: ðtypes.AccessList{}, expectPass: true, enableFeemarket: true, }, } for i, tc := range testCases { suite.Run(tc.name, func() { suite.enableFeemarket = tc.enableFeemarket suite.SetupTest() vmdb := suite.StateDB() var amount, gasPrice, gasFeeCap, gasTipCap *big.Int if tc.cost != nil { amount = tc.cost.BigInt() } if suite.enableFeemarket { if tc.gasFeeCap != nil { gasFeeCap = tc.gasFeeCap } if tc.gasTipCap == nil { gasTipCap = oneInt.BigInt() } else { gasTipCap = tc.gasTipCap } vmdb.AddBalance(suite.address, initBalance.BigInt()) balance := vmdb.GetBalance(suite.address) suite.Require().Equal(balance, initBalance.BigInt()) } else { if tc.gasPrice != nil { gasPrice = tc.gasPrice.BigInt() } vmdb.AddBalance(suite.address, hundredInt.BigInt()) balance := vmdb.GetBalance(suite.address) suite.Require().Equal(balance, hundredInt.BigInt()) } vmdb.Commit() tx := evmtypes.NewTx(zeroInt.BigInt(), 1, &suite.address, amount, tc.gasLimit, gasPrice, gasFeeCap, gasTipCap, nil, tc.accessList) tx.From = suite.address.String() txData, _ := evmtypes.UnpackTxData(tx.Data) fees, err := suite.app.EvmKeeper.DeductTxCostsFromUserBalance( suite.ctx, *tx, txData, evmtypes.DefaultEVMDenom, false, false, suite.enableFeemarket, // london ) if tc.expectPass { suite.Require().NoError(err, "valid test %d failed", i) if tc.enableFeemarket { baseFee := suite.app.FeeMarketKeeper.GetBaseFee(suite.ctx) suite.Require().Equal( fees, sdk.NewCoins( sdk.NewCoin(evmtypes.DefaultEVMDenom, sdk.NewIntFromBigInt(txData.EffectiveFee(baseFee))), ), "valid test %d failed, fee value is wrong ", i, ) } else { suite.Require().Equal( fees, sdk.NewCoins( sdk.NewCoin(evmtypes.DefaultEVMDenom, tc.gasPrice.Mul(sdk.NewIntFromUint64(tc.gasLimit))), ), "valid test %d failed, fee value is wrong ", i, ) } } else { suite.Require().Error(err, "invalid test %d passed", i) suite.Require().Nil(fees, "invalid test %d passed. fees value must be nil", i) } }) } suite.enableFeemarket = false // reset flag }