fix test

crypto-org-chain · Aug 16, 2024 · 8f89bb1 · 8f89bb1
1 parent 48849fe
commit 8f89bb1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/app/ante/eth_test.go b/app/ante/eth_test.go
@@ -93,7 +93,8 @@ func (suite *AnteTestSuite) TestNewEthAccountVerificationDecorator() {
 			suite.Require().NoError(vmdb.Commit())
 
 			accountGetter := ante.NewCachedAccountGetter(suite.ctx, suite.app.AccountKeeper)
-			err := ante.VerifyEthAccount(suite.ctx.WithIsCheckTx(tc.checkTx), tc.tx, suite.app.EvmKeeper, evmtypes.DefaultEVMDenom, accountGetter)
+			balanceGetter := ante.NewCachedBalanceGetter(suite.ctx, suite.app.EvmKeeper)
+			err := ante.VerifyEthAccount(suite.ctx.WithIsCheckTx(tc.checkTx), tc.tx, evmtypes.DefaultEVMDenom, accountGetter, balanceGetter)
 
 			if tc.expPass {
 				suite.Require().NoError(err)
@@ -460,9 +461,9 @@ func (suite *AnteTestSuite) TestCanTransferDecorator() {
 			tc.malleate()
 			suite.Require().NoError(vmdb.Commit())
 
+			balanceGetter := ante.NewCachedBalanceGetter(suite.ctx, suite.app.EvmKeeper)
 			err := ante.CheckEthCanTransfer(
-				suite.ctx.WithIsCheckTx(true), tc.tx,
-				baseFee, rules, suite.app.EvmKeeper, &evmParams,
+				tc.tx, baseFee, rules, &evmParams, balanceGetter,
 			)
 
 			if tc.expPass {